blob: eeeb8511b2131689ab72b11426f47f0dfae7f1bf [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;
Stephen Hines6bcf27b2014-05-29 04:14:42 -0700107 DeclOrVector = ((NamedDecl *)nullptr);
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000108 }
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,
Stephen Hines6bcf27b2014-05-29 04:14:42 -0700175 LookupFilter Filter = nullptr)
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),
Stephen Hines6bcf27b2014-05-29 04:14:42 -0700180 ObjCImplementation(nullptr)
Douglas Gregorca45da02010-11-02 20:36:02 +0000181 {
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 }
Stephen Hines6bcf27b2014-05-29 04:14:42 -0700215
216 Result *data() { return Results.empty()? nullptr : &Results.front(); }
Douglas Gregor86d9a522009-09-21 16:56:56 +0000217 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.
Stephen Hines6bcf27b2014-05-29 04:14:42 -0700292 void MaybeAddResult(Result R, DeclContext *CurContext = nullptr);
293
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000294 /// \brief Add a new result to this result set, where we already know
Stephen Hines176edba2014-12-01 14:53:08 -0800295 /// the hiding declaration (if any).
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000296 ///
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 };
Stephen Hines6bcf27b2014-05-29 04:14:42 -0700367
368 iterator() : DeclOrIterator((NamedDecl *)nullptr), SingleDeclIndex(0) {}
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000369
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 *>()) {
Stephen Hines6bcf27b2014-05-29 04:14:42 -0700378 DeclOrIterator = (NamedDecl *)nullptr;
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000379 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 }
Stephen Hines6bcf27b2014-05-29 04:14:42 -0700464
465 NestedNameSpecifier *Result = nullptr;
Douglas Gregor456c4a12009-09-21 20:12:40 +0000466 while (!TargetParents.empty()) {
Robert Wilhelm344472e2013-08-23 16:11:15 +0000467 const DeclContext *Parent = TargetParents.pop_back_val();
468
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000469 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Parent)) {
Douglas Gregorfb629412010-08-23 21:17:50 +0000470 if (!Namespace->getIdentifier())
471 continue;
472
Douglas Gregor456c4a12009-09-21 20:12:40 +0000473 Result = NestedNameSpecifier::Create(Context, Result, Namespace);
Douglas Gregorfb629412010-08-23 21:17:50 +0000474 }
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000475 else if (const TagDecl *TD = dyn_cast<TagDecl>(Parent))
Douglas Gregor456c4a12009-09-21 20:12:40 +0000476 Result = NestedNameSpecifier::Create(Context, Result,
477 false,
478 Context.getTypeDeclType(TD).getTypePtr());
Douglas Gregor0c8296d2009-11-07 00:00:49 +0000479 }
Douglas Gregor456c4a12009-09-21 20:12:40 +0000480 return Result;
481}
482
Stephen Hinesc568f1e2014-07-21 00:47:37 -0700483/// Determine whether \p Id is a name reserved for the implementation (C99
484/// 7.1.3, C++ [lib.global.names]).
485static bool isReservedName(const IdentifierInfo *Id) {
486 if (Id->getLength() < 2)
487 return false;
488 const char *Name = Id->getNameStart();
489 return Name[0] == '_' &&
490 (Name[1] == '_' || (Name[1] >= 'A' && Name[1] <= 'Z'));
491}
492
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000493bool ResultBuilder::isInterestingDecl(const NamedDecl *ND,
Douglas Gregor45bcd432010-01-14 03:21:49 +0000494 bool &AsNestedNameSpecifier) const {
495 AsNestedNameSpecifier = false;
496
Douglas Gregore495b7f2010-01-14 00:20:49 +0000497 ND = ND->getUnderlyingDecl();
Douglas Gregorf52cede2009-10-09 22:16:47 +0000498
499 // Skip unnamed entities.
Douglas Gregore495b7f2010-01-14 00:20:49 +0000500 if (!ND->getDeclName())
501 return false;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000502
503 // Friend declarations and declarations introduced due to friends are never
504 // added as results.
Stephen Hines0e2c34f2015-03-23 12:09:02 -0700505 if (ND->getFriendObjectKind() == Decl::FOK_Undeclared)
Douglas Gregore495b7f2010-01-14 00:20:49 +0000506 return false;
507
Douglas Gregor76282942009-12-11 17:31:05 +0000508 // Class template (partial) specializations are never added as results.
Douglas Gregore495b7f2010-01-14 00:20:49 +0000509 if (isa<ClassTemplateSpecializationDecl>(ND) ||
510 isa<ClassTemplatePartialSpecializationDecl>(ND))
511 return false;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000512
Douglas Gregor76282942009-12-11 17:31:05 +0000513 // Using declarations themselves are never added as results.
Douglas Gregore495b7f2010-01-14 00:20:49 +0000514 if (isa<UsingDecl>(ND))
515 return false;
516
517 // Some declarations have reserved names that we don't want to ever show.
Stephen Hinesc568f1e2014-07-21 00:47:37 -0700518 // Filter out names reserved for the implementation if they come from a
519 // system header.
520 // TODO: Add a predicate for this.
521 if (const IdentifierInfo *Id = ND->getIdentifier())
522 if (isReservedName(Id) &&
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 Gregor9b0ba872010-11-09 03:59:40 +0000527
Douglas Gregora5fb7c32010-08-16 23:05:20 +0000528 if (Filter == &ResultBuilder::IsNestedNameSpecifier ||
529 ((isa<NamespaceDecl>(ND) || isa<NamespaceAliasDecl>(ND)) &&
530 Filter != &ResultBuilder::IsNamespace &&
Douglas Gregor52779fb2010-09-23 23:01:17 +0000531 Filter != &ResultBuilder::IsNamespaceOrAlias &&
Stephen Hines6bcf27b2014-05-29 04:14:42 -0700532 Filter != nullptr))
Douglas Gregora5fb7c32010-08-16 23:05:20 +0000533 AsNestedNameSpecifier = true;
534
Douglas Gregor86d9a522009-09-21 16:56:56 +0000535 // Filter out any unwanted results.
Douglas Gregor45bcd432010-01-14 03:21:49 +0000536 if (Filter && !(this->*Filter)(ND)) {
537 // Check whether it is interesting as a nested-name-specifier.
David Blaikie4e4d0842012-03-11 07:00:24 +0000538 if (AllowNestedNameSpecifiers && SemaRef.getLangOpts().CPlusPlus &&
Douglas Gregor45bcd432010-01-14 03:21:49 +0000539 IsNestedNameSpecifier(ND) &&
540 (Filter != &ResultBuilder::IsMember ||
541 (isa<CXXRecordDecl>(ND) &&
542 cast<CXXRecordDecl>(ND)->isInjectedClassName()))) {
543 AsNestedNameSpecifier = true;
544 return true;
545 }
546
Douglas Gregore495b7f2010-01-14 00:20:49 +0000547 return false;
Douglas Gregora5fb7c32010-08-16 23:05:20 +0000548 }
Douglas Gregore495b7f2010-01-14 00:20:49 +0000549 // ... then it must be interesting!
550 return true;
551}
552
Douglas Gregor6660d842010-01-14 00:41:07 +0000553bool ResultBuilder::CheckHiddenResult(Result &R, DeclContext *CurContext,
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000554 const NamedDecl *Hiding) {
Douglas Gregor6660d842010-01-14 00:41:07 +0000555 // In C, there is no way to refer to a hidden name.
556 // FIXME: This isn't true; we can find a tag name hidden by an ordinary
557 // name if we introduce the tag type.
David Blaikie4e4d0842012-03-11 07:00:24 +0000558 if (!SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor6660d842010-01-14 00:41:07 +0000559 return true;
560
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000561 const DeclContext *HiddenCtx =
562 R.Declaration->getDeclContext()->getRedeclContext();
Douglas Gregor6660d842010-01-14 00:41:07 +0000563
564 // There is no way to qualify a name declared in a function or method.
565 if (HiddenCtx->isFunctionOrMethod())
566 return true;
567
Sebastian Redl7a126a42010-08-31 00:36:30 +0000568 if (HiddenCtx == Hiding->getDeclContext()->getRedeclContext())
Douglas Gregor6660d842010-01-14 00:41:07 +0000569 return true;
570
571 // We can refer to the result with the appropriate qualification. Do it.
572 R.Hidden = true;
573 R.QualifierIsInformative = false;
574
575 if (!R.Qualifier)
576 R.Qualifier = getRequiredQualification(SemaRef.Context,
577 CurContext,
578 R.Declaration->getDeclContext());
579 return false;
580}
581
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000582/// \brief A simplified classification of types used to determine whether two
583/// types are "similar enough" when adjusting priorities.
Douglas Gregor1827e102010-08-16 16:18:59 +0000584SimplifiedTypeClass clang::getSimplifiedTypeClass(CanQualType T) {
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000585 switch (T->getTypeClass()) {
586 case Type::Builtin:
587 switch (cast<BuiltinType>(T)->getKind()) {
588 case BuiltinType::Void:
589 return STC_Void;
590
591 case BuiltinType::NullPtr:
592 return STC_Pointer;
593
594 case BuiltinType::Overload:
595 case BuiltinType::Dependent:
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000596 return STC_Other;
597
598 case BuiltinType::ObjCId:
599 case BuiltinType::ObjCClass:
600 case BuiltinType::ObjCSel:
601 return STC_ObjectiveC;
602
603 default:
604 return STC_Arithmetic;
605 }
David Blaikie7530c032012-01-17 06:56:22 +0000606
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000607 case Type::Complex:
608 return STC_Arithmetic;
609
610 case Type::Pointer:
611 return STC_Pointer;
612
613 case Type::BlockPointer:
614 return STC_Block;
615
616 case Type::LValueReference:
617 case Type::RValueReference:
618 return getSimplifiedTypeClass(T->getAs<ReferenceType>()->getPointeeType());
619
620 case Type::ConstantArray:
621 case Type::IncompleteArray:
622 case Type::VariableArray:
623 case Type::DependentSizedArray:
624 return STC_Array;
625
626 case Type::DependentSizedExtVector:
627 case Type::Vector:
628 case Type::ExtVector:
629 return STC_Arithmetic;
630
631 case Type::FunctionProto:
632 case Type::FunctionNoProto:
633 return STC_Function;
634
635 case Type::Record:
636 return STC_Record;
637
638 case Type::Enum:
639 return STC_Arithmetic;
640
641 case Type::ObjCObject:
642 case Type::ObjCInterface:
643 case Type::ObjCObjectPointer:
644 return STC_ObjectiveC;
645
646 default:
647 return STC_Other;
648 }
649}
650
651/// \brief Get the type that a given expression will have if this declaration
652/// is used as an expression in its "typical" code-completion form.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000653QualType clang::getDeclUsageType(ASTContext &C, const NamedDecl *ND) {
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000654 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
655
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000656 if (const TypeDecl *Type = dyn_cast<TypeDecl>(ND))
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000657 return C.getTypeDeclType(Type);
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000658 if (const ObjCInterfaceDecl *Iface = dyn_cast<ObjCInterfaceDecl>(ND))
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000659 return C.getObjCInterfaceType(Iface);
660
661 QualType T;
Stephen Hines651f13c2014-04-23 16:59:28 -0700662 if (const FunctionDecl *Function = ND->getAsFunction())
Douglas Gregor5291c3c2010-07-13 08:18:22 +0000663 T = Function->getCallResultType();
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000664 else if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND))
Douglas Gregor5291c3c2010-07-13 08:18:22 +0000665 T = Method->getSendResultType();
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000666 else if (const EnumConstantDecl *Enumerator = dyn_cast<EnumConstantDecl>(ND))
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000667 T = C.getTypeDeclType(cast<EnumDecl>(Enumerator->getDeclContext()));
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000668 else if (const ObjCPropertyDecl *Property = dyn_cast<ObjCPropertyDecl>(ND))
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000669 T = Property->getType();
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000670 else if (const ValueDecl *Value = dyn_cast<ValueDecl>(ND))
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000671 T = Value->getType();
672 else
673 return QualType();
Douglas Gregor3e64d562011-04-14 20:33:34 +0000674
675 // Dig through references, function pointers, and block pointers to
676 // get down to the likely type of an expression when the entity is
677 // used.
678 do {
679 if (const ReferenceType *Ref = T->getAs<ReferenceType>()) {
680 T = Ref->getPointeeType();
681 continue;
682 }
683
684 if (const PointerType *Pointer = T->getAs<PointerType>()) {
685 if (Pointer->getPointeeType()->isFunctionType()) {
686 T = Pointer->getPointeeType();
687 continue;
688 }
689
690 break;
691 }
692
693 if (const BlockPointerType *Block = T->getAs<BlockPointerType>()) {
694 T = Block->getPointeeType();
695 continue;
696 }
697
698 if (const FunctionType *Function = T->getAs<FunctionType>()) {
Stephen Hines651f13c2014-04-23 16:59:28 -0700699 T = Function->getReturnType();
Douglas Gregor3e64d562011-04-14 20:33:34 +0000700 continue;
701 }
702
703 break;
704 } while (true);
705
706 return T;
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000707}
708
Douglas Gregord1f09b42013-01-31 04:52:16 +0000709unsigned ResultBuilder::getBasePriority(const NamedDecl *ND) {
710 if (!ND)
711 return CCP_Unlikely;
712
713 // Context-based decisions.
Richard Smitha41c97a2013-09-20 01:15:31 +0000714 const DeclContext *LexicalDC = ND->getLexicalDeclContext();
715 if (LexicalDC->isFunctionOrMethod()) {
Douglas Gregord1f09b42013-01-31 04:52:16 +0000716 // _cmd is relatively rare
717 if (const ImplicitParamDecl *ImplicitParam =
718 dyn_cast<ImplicitParamDecl>(ND))
719 if (ImplicitParam->getIdentifier() &&
720 ImplicitParam->getIdentifier()->isStr("_cmd"))
721 return CCP_ObjC_cmd;
722
723 return CCP_LocalDeclaration;
724 }
Richard Smitha41c97a2013-09-20 01:15:31 +0000725
726 const DeclContext *DC = ND->getDeclContext()->getRedeclContext();
Douglas Gregord1f09b42013-01-31 04:52:16 +0000727 if (DC->isRecord() || isa<ObjCContainerDecl>(DC))
728 return CCP_MemberDeclaration;
729
730 // Content-based decisions.
731 if (isa<EnumConstantDecl>(ND))
732 return CCP_Constant;
733
Douglas Gregor626799b2013-01-31 05:03:46 +0000734 // Use CCP_Type for type declarations unless we're in a statement, Objective-C
735 // message receiver, or parenthesized expression context. There, it's as
736 // likely that the user will want to write a type as other declarations.
737 if ((isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND)) &&
738 !(CompletionContext.getKind() == CodeCompletionContext::CCC_Statement ||
739 CompletionContext.getKind()
740 == CodeCompletionContext::CCC_ObjCMessageReceiver ||
741 CompletionContext.getKind()
742 == CodeCompletionContext::CCC_ParenthesizedExpression))
Douglas Gregord1f09b42013-01-31 04:52:16 +0000743 return CCP_Type;
744
745 return CCP_Declaration;
746}
747
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000748void ResultBuilder::AdjustResultPriorityForDecl(Result &R) {
749 // If this is an Objective-C method declaration whose selector matches our
750 // preferred selector, give it a priority boost.
751 if (!PreferredSelector.isNull())
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000752 if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(R.Declaration))
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000753 if (PreferredSelector == Method->getSelector())
754 R.Priority += CCD_SelectorMatch;
Douglas Gregor08f43cd2010-09-20 23:11:55 +0000755
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000756 // If we have a preferred type, adjust the priority for results with exactly-
757 // matching or nearly-matching types.
758 if (!PreferredType.isNull()) {
759 QualType T = getDeclUsageType(SemaRef.Context, R.Declaration);
760 if (!T.isNull()) {
761 CanQualType TC = SemaRef.Context.getCanonicalType(T);
762 // Check for exactly-matching types (modulo qualifiers).
763 if (SemaRef.Context.hasSameUnqualifiedType(PreferredType, TC))
764 R.Priority /= CCF_ExactTypeMatch;
765 // Check for nearly-matching types, based on classification of each.
766 else if ((getSimplifiedTypeClass(PreferredType)
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000767 == getSimplifiedTypeClass(TC)) &&
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000768 !(PreferredType->isEnumeralType() && TC->isEnumeralType()))
769 R.Priority /= CCF_SimilarTypeMatch;
770 }
771 }
Douglas Gregor1f5537a2010-07-08 23:20:03 +0000772}
773
Douglas Gregor6f942b22010-09-21 16:06:22 +0000774void ResultBuilder::MaybeAddConstructorResults(Result R) {
David Blaikie4e4d0842012-03-11 07:00:24 +0000775 if (!SemaRef.getLangOpts().CPlusPlus || !R.Declaration ||
Douglas Gregor6f942b22010-09-21 16:06:22 +0000776 !CompletionContext.wantConstructorResults())
777 return;
778
779 ASTContext &Context = SemaRef.Context;
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000780 const NamedDecl *D = R.Declaration;
Stephen Hines6bcf27b2014-05-29 04:14:42 -0700781 const CXXRecordDecl *Record = nullptr;
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000782 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D))
Douglas Gregor6f942b22010-09-21 16:06:22 +0000783 Record = ClassTemplate->getTemplatedDecl();
784 else if ((Record = dyn_cast<CXXRecordDecl>(D))) {
785 // Skip specializations and partial specializations.
786 if (isa<ClassTemplateSpecializationDecl>(Record))
787 return;
788 } else {
789 // There are no constructors here.
790 return;
791 }
792
793 Record = Record->getDefinition();
794 if (!Record)
795 return;
796
797
798 QualType RecordTy = Context.getTypeDeclType(Record);
799 DeclarationName ConstructorName
800 = Context.DeclarationNames.getCXXConstructorName(
801 Context.getCanonicalType(RecordTy));
Stephen Hines0e2c34f2015-03-23 12:09:02 -0700802 DeclContext::lookup_result Ctors = Record->lookup(ConstructorName);
803 for (DeclContext::lookup_iterator I = Ctors.begin(),
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000804 E = Ctors.end();
805 I != E; ++I) {
David Blaikie3bc93e32012-12-19 00:45:41 +0000806 R.Declaration = *I;
Douglas Gregor6f942b22010-09-21 16:06:22 +0000807 R.CursorKind = getCursorKindForDecl(R.Declaration);
808 Results.push_back(R);
809 }
810}
811
Douglas Gregore495b7f2010-01-14 00:20:49 +0000812void ResultBuilder::MaybeAddResult(Result R, DeclContext *CurContext) {
813 assert(!ShadowMaps.empty() && "Must enter into a results scope");
814
815 if (R.Kind != Result::RK_Declaration) {
816 // For non-declaration results, just add the result.
817 Results.push_back(R);
818 return;
819 }
820
821 // Look through using declarations.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000822 if (const UsingShadowDecl *Using =
823 dyn_cast<UsingShadowDecl>(R.Declaration)) {
Douglas Gregord1f09b42013-01-31 04:52:16 +0000824 MaybeAddResult(Result(Using->getTargetDecl(),
825 getBasePriority(Using->getTargetDecl()),
826 R.Qualifier),
827 CurContext);
Douglas Gregore495b7f2010-01-14 00:20:49 +0000828 return;
829 }
830
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000831 const Decl *CanonDecl = R.Declaration->getCanonicalDecl();
Douglas Gregore495b7f2010-01-14 00:20:49 +0000832 unsigned IDNS = CanonDecl->getIdentifierNamespace();
833
Douglas Gregor45bcd432010-01-14 03:21:49 +0000834 bool AsNestedNameSpecifier = false;
835 if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier))
Douglas Gregore495b7f2010-01-14 00:20:49 +0000836 return;
837
Douglas Gregor6f942b22010-09-21 16:06:22 +0000838 // C++ constructors are never found by name lookup.
839 if (isa<CXXConstructorDecl>(R.Declaration))
840 return;
841
Douglas Gregor86d9a522009-09-21 16:56:56 +0000842 ShadowMap &SMap = ShadowMaps.back();
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000843 ShadowMapEntry::iterator I, IEnd;
844 ShadowMap::iterator NamePos = SMap.find(R.Declaration->getDeclName());
845 if (NamePos != SMap.end()) {
846 I = NamePos->second.begin();
847 IEnd = NamePos->second.end();
848 }
849
850 for (; I != IEnd; ++I) {
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000851 const NamedDecl *ND = I->first;
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000852 unsigned Index = I->second;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000853 if (ND->getCanonicalDecl() == CanonDecl) {
854 // This is a redeclaration. Always pick the newer declaration.
Douglas Gregor86d9a522009-09-21 16:56:56 +0000855 Results[Index].Declaration = R.Declaration;
856
Douglas Gregor86d9a522009-09-21 16:56:56 +0000857 // We're done.
858 return;
859 }
860 }
861
862 // This is a new declaration in this scope. However, check whether this
863 // declaration name is hidden by a similarly-named declaration in an outer
864 // scope.
865 std::list<ShadowMap>::iterator SM, SMEnd = ShadowMaps.end();
866 --SMEnd;
867 for (SM = ShadowMaps.begin(); SM != SMEnd; ++SM) {
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000868 ShadowMapEntry::iterator I, IEnd;
869 ShadowMap::iterator NamePos = SM->find(R.Declaration->getDeclName());
870 if (NamePos != SM->end()) {
871 I = NamePos->second.begin();
872 IEnd = NamePos->second.end();
873 }
874 for (; I != IEnd; ++I) {
Douglas Gregor86d9a522009-09-21 16:56:56 +0000875 // A tag declaration does not hide a non-tag declaration.
John McCall0d6b1642010-04-23 18:46:30 +0000876 if (I->first->hasTagIdentifierNamespace() &&
Richard Smitha41c97a2013-09-20 01:15:31 +0000877 (IDNS & (Decl::IDNS_Member | Decl::IDNS_Ordinary |
878 Decl::IDNS_LocalExtern | Decl::IDNS_ObjCProtocol)))
Douglas Gregor86d9a522009-09-21 16:56:56 +0000879 continue;
880
881 // Protocols are in distinct namespaces from everything else.
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000882 if (((I->first->getIdentifierNamespace() & Decl::IDNS_ObjCProtocol)
Douglas Gregor86d9a522009-09-21 16:56:56 +0000883 || (IDNS & Decl::IDNS_ObjCProtocol)) &&
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000884 I->first->getIdentifierNamespace() != IDNS)
Douglas Gregor86d9a522009-09-21 16:56:56 +0000885 continue;
886
887 // The newly-added result is hidden by an entry in the shadow map.
Douglas Gregor6660d842010-01-14 00:41:07 +0000888 if (CheckHiddenResult(R, CurContext, I->first))
Douglas Gregor86d9a522009-09-21 16:56:56 +0000889 return;
Douglas Gregor86d9a522009-09-21 16:56:56 +0000890
891 break;
892 }
893 }
894
895 // Make sure that any given declaration only shows up in the result set once.
Stephen Hines176edba2014-12-01 14:53:08 -0800896 if (!AllDeclsFound.insert(CanonDecl).second)
Douglas Gregor86d9a522009-09-21 16:56:56 +0000897 return;
Douglas Gregor265f7492010-08-27 15:29:55 +0000898
Douglas Gregoreb5758b2009-09-23 22:26:46 +0000899 // If the filter is for nested-name-specifiers, then this result starts a
900 // nested-name-specifier.
Douglas Gregor12e13132010-05-26 22:00:08 +0000901 if (AsNestedNameSpecifier) {
Douglas Gregoreb5758b2009-09-23 22:26:46 +0000902 R.StartsNestedNameSpecifier = true;
Douglas Gregor12e13132010-05-26 22:00:08 +0000903 R.Priority = CCP_NestedNameSpecifier;
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000904 } else
905 AdjustResultPriorityForDecl(R);
Douglas Gregor265f7492010-08-27 15:29:55 +0000906
Douglas Gregor0563c262009-09-22 23:15:58 +0000907 // If this result is supposed to have an informative qualifier, add one.
Douglas Gregoreb5758b2009-09-23 22:26:46 +0000908 if (R.QualifierIsInformative && !R.Qualifier &&
909 !R.StartsNestedNameSpecifier) {
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000910 const DeclContext *Ctx = R.Declaration->getDeclContext();
911 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx))
Stephen Hines6bcf27b2014-05-29 04:14:42 -0700912 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, nullptr,
913 Namespace);
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000914 else if (const TagDecl *Tag = dyn_cast<TagDecl>(Ctx))
Stephen Hines6bcf27b2014-05-29 04:14:42 -0700915 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, nullptr,
916 false, SemaRef.Context.getTypeDeclType(Tag).getTypePtr());
Douglas Gregor0563c262009-09-22 23:15:58 +0000917 else
918 R.QualifierIsInformative = false;
919 }
Douglas Gregoreb5758b2009-09-23 22:26:46 +0000920
Douglas Gregor86d9a522009-09-21 16:56:56 +0000921 // Insert this result into the set of results and into the current shadow
922 // map.
Douglas Gregorfbcb5d62009-12-06 20:23:50 +0000923 SMap[R.Declaration->getDeclName()].Add(R.Declaration, Results.size());
Douglas Gregor86d9a522009-09-21 16:56:56 +0000924 Results.push_back(R);
Douglas Gregor6f942b22010-09-21 16:06:22 +0000925
926 if (!AsNestedNameSpecifier)
927 MaybeAddConstructorResults(R);
Douglas Gregor86d9a522009-09-21 16:56:56 +0000928}
929
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000930void ResultBuilder::AddResult(Result R, DeclContext *CurContext,
Douglas Gregor0cc84042010-01-14 15:47:35 +0000931 NamedDecl *Hiding, bool InBaseClass = false) {
Douglas Gregora4477812010-01-14 16:01:26 +0000932 if (R.Kind != Result::RK_Declaration) {
933 // For non-declaration results, just add the result.
934 Results.push_back(R);
935 return;
936 }
937
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000938 // Look through using declarations.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000939 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(R.Declaration)) {
Douglas Gregord1f09b42013-01-31 04:52:16 +0000940 AddResult(Result(Using->getTargetDecl(),
941 getBasePriority(Using->getTargetDecl()),
942 R.Qualifier),
943 CurContext, Hiding);
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000944 return;
945 }
946
Douglas Gregor45bcd432010-01-14 03:21:49 +0000947 bool AsNestedNameSpecifier = false;
948 if (!isInterestingDecl(R.Declaration, AsNestedNameSpecifier))
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000949 return;
950
Douglas Gregor6f942b22010-09-21 16:06:22 +0000951 // C++ constructors are never found by name lookup.
952 if (isa<CXXConstructorDecl>(R.Declaration))
953 return;
954
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000955 if (Hiding && CheckHiddenResult(R, CurContext, Hiding))
956 return;
Nick Lewycky173a37a2012-04-03 21:44:08 +0000957
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000958 // Make sure that any given declaration only shows up in the result set once.
Stephen Hines176edba2014-12-01 14:53:08 -0800959 if (!AllDeclsFound.insert(R.Declaration->getCanonicalDecl()).second)
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000960 return;
961
962 // If the filter is for nested-name-specifiers, then this result starts a
963 // nested-name-specifier.
Douglas Gregor12e13132010-05-26 22:00:08 +0000964 if (AsNestedNameSpecifier) {
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000965 R.StartsNestedNameSpecifier = true;
Douglas Gregor12e13132010-05-26 22:00:08 +0000966 R.Priority = CCP_NestedNameSpecifier;
967 }
Douglas Gregor0cc84042010-01-14 15:47:35 +0000968 else if (Filter == &ResultBuilder::IsMember && !R.Qualifier && InBaseClass &&
969 isa<CXXRecordDecl>(R.Declaration->getDeclContext()
Sebastian Redl7a126a42010-08-31 00:36:30 +0000970 ->getRedeclContext()))
Douglas Gregor0cc84042010-01-14 15:47:35 +0000971 R.QualifierIsInformative = true;
972
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000973 // If this result is supposed to have an informative qualifier, add one.
974 if (R.QualifierIsInformative && !R.Qualifier &&
975 !R.StartsNestedNameSpecifier) {
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000976 const DeclContext *Ctx = R.Declaration->getDeclContext();
977 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(Ctx))
Stephen Hines6bcf27b2014-05-29 04:14:42 -0700978 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, nullptr,
979 Namespace);
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000980 else if (const TagDecl *Tag = dyn_cast<TagDecl>(Ctx))
Stephen Hines6bcf27b2014-05-29 04:14:42 -0700981 R.Qualifier = NestedNameSpecifier::Create(SemaRef.Context, nullptr, false,
Douglas Gregor45bcd432010-01-14 03:21:49 +0000982 SemaRef.Context.getTypeDeclType(Tag).getTypePtr());
Douglas Gregor1ca6ae82010-01-14 01:09:38 +0000983 else
984 R.QualifierIsInformative = false;
985 }
986
Douglas Gregor12e13132010-05-26 22:00:08 +0000987 // Adjust the priority if this result comes from a base class.
988 if (InBaseClass)
989 R.Priority += CCD_InBaseClass;
990
Douglas Gregorcee9ff12010-09-20 22:39:41 +0000991 AdjustResultPriorityForDecl(R);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +0000992
Douglas Gregor3cdee122010-08-26 16:36:48 +0000993 if (HasObjectTypeQualifiers)
Dmitri Gribenko89cf4252013-01-23 17:21:11 +0000994 if (const CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(R.Declaration))
Douglas Gregor3cdee122010-08-26 16:36:48 +0000995 if (Method->isInstance()) {
996 Qualifiers MethodQuals
997 = Qualifiers::fromCVRMask(Method->getTypeQualifiers());
998 if (ObjectTypeQualifiers == MethodQuals)
999 R.Priority += CCD_ObjectQualifierMatch;
1000 else if (ObjectTypeQualifiers - MethodQuals) {
1001 // The method cannot be invoked, because doing so would drop
1002 // qualifiers.
1003 return;
1004 }
1005 }
1006
Douglas Gregor1ca6ae82010-01-14 01:09:38 +00001007 // Insert this result into the set of results.
1008 Results.push_back(R);
Douglas Gregor6f942b22010-09-21 16:06:22 +00001009
1010 if (!AsNestedNameSpecifier)
1011 MaybeAddConstructorResults(R);
Douglas Gregor1ca6ae82010-01-14 01:09:38 +00001012}
1013
Douglas Gregora4477812010-01-14 16:01:26 +00001014void ResultBuilder::AddResult(Result R) {
1015 assert(R.Kind != Result::RK_Declaration &&
1016 "Declaration results need more context");
1017 Results.push_back(R);
1018}
1019
Douglas Gregor86d9a522009-09-21 16:56:56 +00001020/// \brief Enter into a new scope.
1021void ResultBuilder::EnterNewScope() {
1022 ShadowMaps.push_back(ShadowMap());
1023}
1024
1025/// \brief Exit from the current scope.
1026void ResultBuilder::ExitScope() {
Douglas Gregorfbcb5d62009-12-06 20:23:50 +00001027 for (ShadowMap::iterator E = ShadowMaps.back().begin(),
1028 EEnd = ShadowMaps.back().end();
1029 E != EEnd;
1030 ++E)
1031 E->second.Destroy();
1032
Douglas Gregor86d9a522009-09-21 16:56:56 +00001033 ShadowMaps.pop_back();
1034}
1035
Douglas Gregor791215b2009-09-21 20:51:25 +00001036/// \brief Determines whether this given declaration will be found by
1037/// ordinary name lookup.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001038bool ResultBuilder::IsOrdinaryName(const NamedDecl *ND) const {
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001039 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1040
Richard Smitha41c97a2013-09-20 01:15:31 +00001041 // If name lookup finds a local extern declaration, then we are in a
1042 // context where it behaves like an ordinary name.
1043 unsigned IDNS = Decl::IDNS_Ordinary | Decl::IDNS_LocalExtern;
David Blaikie4e4d0842012-03-11 07:00:24 +00001044 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor9b30b262010-06-15 20:26:51 +00001045 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace | Decl::IDNS_Member;
David Blaikie4e4d0842012-03-11 07:00:24 +00001046 else if (SemaRef.getLangOpts().ObjC1) {
Douglas Gregorca45da02010-11-02 20:36:02 +00001047 if (isa<ObjCIvarDecl>(ND))
1048 return true;
Douglas Gregorca45da02010-11-02 20:36:02 +00001049 }
1050
Douglas Gregor791215b2009-09-21 20:51:25 +00001051 return ND->getIdentifierNamespace() & IDNS;
1052}
1053
Douglas Gregor01dfea02010-01-10 23:08:15 +00001054/// \brief Determines whether this given declaration will be found by
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001055/// ordinary name lookup but is not a type name.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001056bool ResultBuilder::IsOrdinaryNonTypeName(const NamedDecl *ND) const {
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001057 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1058 if (isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND))
1059 return false;
1060
Richard Smitha41c97a2013-09-20 01:15:31 +00001061 unsigned IDNS = Decl::IDNS_Ordinary | Decl::IDNS_LocalExtern;
David Blaikie4e4d0842012-03-11 07:00:24 +00001062 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor9b30b262010-06-15 20:26:51 +00001063 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace | Decl::IDNS_Member;
David Blaikie4e4d0842012-03-11 07:00:24 +00001064 else if (SemaRef.getLangOpts().ObjC1) {
Douglas Gregorca45da02010-11-02 20:36:02 +00001065 if (isa<ObjCIvarDecl>(ND))
1066 return true;
Douglas Gregorca45da02010-11-02 20:36:02 +00001067 }
1068
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001069 return ND->getIdentifierNamespace() & IDNS;
1070}
1071
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001072bool ResultBuilder::IsIntegralConstantValue(const NamedDecl *ND) const {
Douglas Gregorf9578432010-07-28 21:50:18 +00001073 if (!IsOrdinaryNonTypeName(ND))
1074 return 0;
1075
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001076 if (const ValueDecl *VD = dyn_cast<ValueDecl>(ND->getUnderlyingDecl()))
Douglas Gregorf9578432010-07-28 21:50:18 +00001077 if (VD->getType()->isIntegralOrEnumerationType())
1078 return true;
1079
1080 return false;
1081}
1082
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001083/// \brief Determines whether this given declaration will be found by
Douglas Gregor01dfea02010-01-10 23:08:15 +00001084/// ordinary name lookup.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001085bool ResultBuilder::IsOrdinaryNonValueName(const NamedDecl *ND) const {
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001086 ND = cast<NamedDecl>(ND->getUnderlyingDecl());
1087
Richard Smitha41c97a2013-09-20 01:15:31 +00001088 unsigned IDNS = Decl::IDNS_Ordinary | Decl::IDNS_LocalExtern;
David Blaikie4e4d0842012-03-11 07:00:24 +00001089 if (SemaRef.getLangOpts().CPlusPlus)
John McCall0d6b1642010-04-23 18:46:30 +00001090 IDNS |= Decl::IDNS_Tag | Decl::IDNS_Namespace;
Douglas Gregor01dfea02010-01-10 23:08:15 +00001091
1092 return (ND->getIdentifierNamespace() & IDNS) &&
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001093 !isa<ValueDecl>(ND) && !isa<FunctionTemplateDecl>(ND) &&
1094 !isa<ObjCPropertyDecl>(ND);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001095}
1096
Douglas Gregor86d9a522009-09-21 16:56:56 +00001097/// \brief Determines whether the given declaration is suitable as the
1098/// start of a C++ nested-name-specifier, e.g., a class or namespace.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001099bool ResultBuilder::IsNestedNameSpecifier(const NamedDecl *ND) const {
Douglas Gregor86d9a522009-09-21 16:56:56 +00001100 // Allow us to find class templates, too.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001101 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
Douglas Gregor86d9a522009-09-21 16:56:56 +00001102 ND = ClassTemplate->getTemplatedDecl();
1103
1104 return SemaRef.isAcceptableNestedNameSpecifier(ND);
1105}
1106
1107/// \brief Determines whether the given declaration is an enumeration.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001108bool ResultBuilder::IsEnum(const NamedDecl *ND) const {
Douglas Gregor86d9a522009-09-21 16:56:56 +00001109 return isa<EnumDecl>(ND);
1110}
1111
1112/// \brief Determines whether the given declaration is a class or struct.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001113bool ResultBuilder::IsClassOrStruct(const NamedDecl *ND) const {
Douglas Gregor86d9a522009-09-21 16:56:56 +00001114 // Allow us to find class templates, too.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001115 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
Douglas Gregor86d9a522009-09-21 16:56:56 +00001116 ND = ClassTemplate->getTemplatedDecl();
Joao Matos6666ed42012-08-31 18:45:21 +00001117
1118 // For purposes of this check, interfaces match too.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001119 if (const RecordDecl *RD = dyn_cast<RecordDecl>(ND))
Abramo Bagnara465d41b2010-05-11 21:36:43 +00001120 return RD->getTagKind() == TTK_Class ||
Joao Matos6666ed42012-08-31 18:45:21 +00001121 RD->getTagKind() == TTK_Struct ||
1122 RD->getTagKind() == TTK_Interface;
Douglas Gregor86d9a522009-09-21 16:56:56 +00001123
1124 return false;
1125}
1126
1127/// \brief Determines whether the given declaration is a union.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001128bool ResultBuilder::IsUnion(const NamedDecl *ND) const {
Douglas Gregor86d9a522009-09-21 16:56:56 +00001129 // Allow us to find class templates, too.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001130 if (const ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(ND))
Douglas Gregor86d9a522009-09-21 16:56:56 +00001131 ND = ClassTemplate->getTemplatedDecl();
1132
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001133 if (const RecordDecl *RD = dyn_cast<RecordDecl>(ND))
Abramo Bagnara465d41b2010-05-11 21:36:43 +00001134 return RD->getTagKind() == TTK_Union;
Douglas Gregor86d9a522009-09-21 16:56:56 +00001135
1136 return false;
1137}
1138
1139/// \brief Determines whether the given declaration is a namespace.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001140bool ResultBuilder::IsNamespace(const NamedDecl *ND) const {
Douglas Gregor86d9a522009-09-21 16:56:56 +00001141 return isa<NamespaceDecl>(ND);
1142}
1143
1144/// \brief Determines whether the given declaration is a namespace or
1145/// namespace alias.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001146bool ResultBuilder::IsNamespaceOrAlias(const NamedDecl *ND) const {
Douglas Gregor86d9a522009-09-21 16:56:56 +00001147 return isa<NamespaceDecl>(ND) || isa<NamespaceAliasDecl>(ND);
1148}
1149
Douglas Gregor76282942009-12-11 17:31:05 +00001150/// \brief Determines whether the given declaration is a type.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001151bool ResultBuilder::IsType(const NamedDecl *ND) const {
1152 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(ND))
Douglas Gregord32b0222010-08-24 01:06:58 +00001153 ND = Using->getTargetDecl();
1154
1155 return isa<TypeDecl>(ND) || isa<ObjCInterfaceDecl>(ND);
Douglas Gregor86d9a522009-09-21 16:56:56 +00001156}
1157
Douglas Gregor76282942009-12-11 17:31:05 +00001158/// \brief Determines which members of a class should be visible via
1159/// "." or "->". Only value declarations, nested name specifiers, and
1160/// using declarations thereof should show up.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001161bool ResultBuilder::IsMember(const NamedDecl *ND) const {
1162 if (const UsingShadowDecl *Using = dyn_cast<UsingShadowDecl>(ND))
Douglas Gregor76282942009-12-11 17:31:05 +00001163 ND = Using->getTargetDecl();
1164
Douglas Gregorce821962009-12-11 18:14:22 +00001165 return isa<ValueDecl>(ND) || isa<FunctionTemplateDecl>(ND) ||
1166 isa<ObjCPropertyDecl>(ND);
Douglas Gregoreb5758b2009-09-23 22:26:46 +00001167}
1168
Douglas Gregor8e254cf2010-05-27 23:06:34 +00001169static bool isObjCReceiverType(ASTContext &C, QualType T) {
1170 T = C.getCanonicalType(T);
1171 switch (T->getTypeClass()) {
1172 case Type::ObjCObject:
1173 case Type::ObjCInterface:
1174 case Type::ObjCObjectPointer:
1175 return true;
1176
1177 case Type::Builtin:
1178 switch (cast<BuiltinType>(T)->getKind()) {
1179 case BuiltinType::ObjCId:
1180 case BuiltinType::ObjCClass:
1181 case BuiltinType::ObjCSel:
1182 return true;
1183
1184 default:
1185 break;
1186 }
1187 return false;
1188
1189 default:
1190 break;
1191 }
1192
David Blaikie4e4d0842012-03-11 07:00:24 +00001193 if (!C.getLangOpts().CPlusPlus)
Douglas Gregor8e254cf2010-05-27 23:06:34 +00001194 return false;
1195
1196 // FIXME: We could perform more analysis here to determine whether a
1197 // particular class type has any conversions to Objective-C types. For now,
1198 // just accept all class types.
1199 return T->isDependentType() || T->isRecordType();
1200}
1201
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001202bool ResultBuilder::IsObjCMessageReceiver(const NamedDecl *ND) const {
Douglas Gregor8e254cf2010-05-27 23:06:34 +00001203 QualType T = getDeclUsageType(SemaRef.Context, ND);
1204 if (T.isNull())
1205 return false;
1206
1207 T = SemaRef.Context.getBaseElementType(T);
1208 return isObjCReceiverType(SemaRef.Context, T);
1209}
1210
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001211bool ResultBuilder::IsObjCMessageReceiverOrLambdaCapture(const NamedDecl *ND) const {
Douglas Gregor81f3bff2012-02-15 15:34:24 +00001212 if (IsObjCMessageReceiver(ND))
1213 return true;
1214
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001215 const VarDecl *Var = dyn_cast<VarDecl>(ND);
Douglas Gregor81f3bff2012-02-15 15:34:24 +00001216 if (!Var)
1217 return false;
1218
1219 return Var->hasLocalStorage() && !Var->hasAttr<BlocksAttr>();
1220}
1221
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001222bool ResultBuilder::IsObjCCollection(const NamedDecl *ND) const {
David Blaikie4e4d0842012-03-11 07:00:24 +00001223 if ((SemaRef.getLangOpts().CPlusPlus && !IsOrdinaryName(ND)) ||
1224 (!SemaRef.getLangOpts().CPlusPlus && !IsOrdinaryNonTypeName(ND)))
Douglas Gregorfb629412010-08-23 21:17:50 +00001225 return false;
1226
1227 QualType T = getDeclUsageType(SemaRef.Context, ND);
1228 if (T.isNull())
1229 return false;
1230
1231 T = SemaRef.Context.getBaseElementType(T);
1232 return T->isObjCObjectType() || T->isObjCObjectPointerType() ||
1233 T->isObjCIdType() ||
David Blaikie4e4d0842012-03-11 07:00:24 +00001234 (SemaRef.getLangOpts().CPlusPlus && T->isRecordType());
Douglas Gregorfb629412010-08-23 21:17:50 +00001235}
Douglas Gregor8e254cf2010-05-27 23:06:34 +00001236
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001237bool ResultBuilder::IsImpossibleToSatisfy(const NamedDecl *ND) const {
Douglas Gregor52779fb2010-09-23 23:01:17 +00001238 return false;
1239}
1240
James Dennettde23c7e2012-06-17 05:33:25 +00001241/// \brief Determines whether the given declaration is an Objective-C
Douglas Gregor80f4f4c2010-01-14 16:08:12 +00001242/// instance variable.
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00001243bool ResultBuilder::IsObjCIvar(const NamedDecl *ND) const {
Douglas Gregor80f4f4c2010-01-14 16:08:12 +00001244 return isa<ObjCIvarDecl>(ND);
1245}
1246
Douglas Gregor1ca6ae82010-01-14 01:09:38 +00001247namespace {
1248 /// \brief Visible declaration consumer that adds a code-completion result
1249 /// for each visible declaration.
1250 class CodeCompletionDeclConsumer : public VisibleDeclConsumer {
1251 ResultBuilder &Results;
1252 DeclContext *CurContext;
1253
1254 public:
1255 CodeCompletionDeclConsumer(ResultBuilder &Results, DeclContext *CurContext)
1256 : Results(Results), CurContext(CurContext) { }
Stephen Hines651f13c2014-04-23 16:59:28 -07001257
1258 void FoundDecl(NamedDecl *ND, NamedDecl *Hiding, DeclContext *Ctx,
1259 bool InBaseClass) override {
Erik Verbruggend1205962011-10-06 07:27:49 +00001260 bool Accessible = true;
Douglas Gregor17015ef2011-11-03 16:51:37 +00001261 if (Ctx)
1262 Accessible = Results.getSema().IsSimplyAccessible(ND, Ctx);
Stephen Hines6bcf27b2014-05-29 04:14:42 -07001263
1264 ResultBuilder::Result Result(ND, Results.getBasePriority(ND), nullptr,
1265 false, Accessible);
Erik Verbruggend1205962011-10-06 07:27:49 +00001266 Results.AddResult(Result, CurContext, Hiding, InBaseClass);
Douglas Gregor1ca6ae82010-01-14 01:09:38 +00001267 }
1268 };
1269}
1270
Douglas Gregor86d9a522009-09-21 16:56:56 +00001271/// \brief Add type specifiers for the current language as keyword results.
Douglas Gregorbca403c2010-01-13 23:51:12 +00001272static void AddTypeSpecifierResults(const LangOptions &LangOpts,
Douglas Gregor86d9a522009-09-21 16:56:56 +00001273 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00001274 typedef CodeCompletionResult Result;
Douglas Gregor12e13132010-05-26 22:00:08 +00001275 Results.AddResult(Result("short", CCP_Type));
1276 Results.AddResult(Result("long", CCP_Type));
1277 Results.AddResult(Result("signed", CCP_Type));
1278 Results.AddResult(Result("unsigned", CCP_Type));
1279 Results.AddResult(Result("void", CCP_Type));
1280 Results.AddResult(Result("char", CCP_Type));
1281 Results.AddResult(Result("int", CCP_Type));
1282 Results.AddResult(Result("float", CCP_Type));
1283 Results.AddResult(Result("double", CCP_Type));
1284 Results.AddResult(Result("enum", CCP_Type));
1285 Results.AddResult(Result("struct", CCP_Type));
1286 Results.AddResult(Result("union", CCP_Type));
1287 Results.AddResult(Result("const", CCP_Type));
1288 Results.AddResult(Result("volatile", CCP_Type));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001289
Douglas Gregor86d9a522009-09-21 16:56:56 +00001290 if (LangOpts.C99) {
1291 // C99-specific
Douglas Gregor12e13132010-05-26 22:00:08 +00001292 Results.AddResult(Result("_Complex", CCP_Type));
1293 Results.AddResult(Result("_Imaginary", CCP_Type));
1294 Results.AddResult(Result("_Bool", CCP_Type));
1295 Results.AddResult(Result("restrict", CCP_Type));
Douglas Gregor86d9a522009-09-21 16:56:56 +00001296 }
1297
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00001298 CodeCompletionBuilder Builder(Results.getAllocator(),
1299 Results.getCodeCompletionTUInfo());
Douglas Gregor86d9a522009-09-21 16:56:56 +00001300 if (LangOpts.CPlusPlus) {
1301 // C++-specific
Douglas Gregorb05496d2010-09-20 21:11:48 +00001302 Results.AddResult(Result("bool", CCP_Type +
1303 (LangOpts.ObjC1? CCD_bool_in_ObjC : 0)));
Douglas Gregor12e13132010-05-26 22:00:08 +00001304 Results.AddResult(Result("class", CCP_Type));
1305 Results.AddResult(Result("wchar_t", CCP_Type));
Douglas Gregor86d9a522009-09-21 16:56:56 +00001306
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001307 // typename qualified-id
Douglas Gregor218937c2011-02-01 19:23:04 +00001308 Builder.AddTypedTextChunk("typename");
1309 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1310 Builder.AddPlaceholderChunk("qualifier");
1311 Builder.AddTextChunk("::");
1312 Builder.AddPlaceholderChunk("name");
1313 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregord8e8a582010-05-25 21:41:55 +00001314
Richard Smith80ad52f2013-01-02 11:42:31 +00001315 if (LangOpts.CPlusPlus11) {
Douglas Gregor12e13132010-05-26 22:00:08 +00001316 Results.AddResult(Result("auto", CCP_Type));
1317 Results.AddResult(Result("char16_t", CCP_Type));
1318 Results.AddResult(Result("char32_t", CCP_Type));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001319
Douglas Gregor218937c2011-02-01 19:23:04 +00001320 Builder.AddTypedTextChunk("decltype");
1321 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1322 Builder.AddPlaceholderChunk("expression");
1323 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1324 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor86d9a522009-09-21 16:56:56 +00001325 }
1326 }
1327
1328 // GNU extensions
1329 if (LangOpts.GNUMode) {
1330 // FIXME: Enable when we actually support decimal floating point.
Douglas Gregora4477812010-01-14 16:01:26 +00001331 // Results.AddResult(Result("_Decimal32"));
1332 // Results.AddResult(Result("_Decimal64"));
1333 // Results.AddResult(Result("_Decimal128"));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001334
Douglas Gregor218937c2011-02-01 19:23:04 +00001335 Builder.AddTypedTextChunk("typeof");
1336 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1337 Builder.AddPlaceholderChunk("expression");
1338 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001339
Douglas Gregor218937c2011-02-01 19:23:04 +00001340 Builder.AddTypedTextChunk("typeof");
1341 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1342 Builder.AddPlaceholderChunk("type");
1343 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1344 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor86d9a522009-09-21 16:56:56 +00001345 }
1346}
1347
John McCallf312b1e2010-08-26 23:41:50 +00001348static void AddStorageSpecifiers(Sema::ParserCompletionContext CCC,
Douglas Gregor01dfea02010-01-10 23:08:15 +00001349 const LangOptions &LangOpts,
Douglas Gregor01dfea02010-01-10 23:08:15 +00001350 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00001351 typedef CodeCompletionResult Result;
Douglas Gregor01dfea02010-01-10 23:08:15 +00001352 // Note: we don't suggest either "auto" or "register", because both
1353 // are pointless as storage specifiers. Elsewhere, we suggest "auto"
1354 // in C++0x as a type specifier.
Douglas Gregora4477812010-01-14 16:01:26 +00001355 Results.AddResult(Result("extern"));
1356 Results.AddResult(Result("static"));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001357}
1358
John McCallf312b1e2010-08-26 23:41:50 +00001359static void AddFunctionSpecifiers(Sema::ParserCompletionContext CCC,
Douglas Gregor01dfea02010-01-10 23:08:15 +00001360 const LangOptions &LangOpts,
Douglas Gregor01dfea02010-01-10 23:08:15 +00001361 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00001362 typedef CodeCompletionResult Result;
Douglas Gregor01dfea02010-01-10 23:08:15 +00001363 switch (CCC) {
John McCallf312b1e2010-08-26 23:41:50 +00001364 case Sema::PCC_Class:
1365 case Sema::PCC_MemberTemplate:
Douglas Gregor01dfea02010-01-10 23:08:15 +00001366 if (LangOpts.CPlusPlus) {
Douglas Gregora4477812010-01-14 16:01:26 +00001367 Results.AddResult(Result("explicit"));
1368 Results.AddResult(Result("friend"));
1369 Results.AddResult(Result("mutable"));
1370 Results.AddResult(Result("virtual"));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001371 }
1372 // Fall through
1373
John McCallf312b1e2010-08-26 23:41:50 +00001374 case Sema::PCC_ObjCInterface:
1375 case Sema::PCC_ObjCImplementation:
1376 case Sema::PCC_Namespace:
1377 case Sema::PCC_Template:
Douglas Gregor01dfea02010-01-10 23:08:15 +00001378 if (LangOpts.CPlusPlus || LangOpts.C99)
Douglas Gregora4477812010-01-14 16:01:26 +00001379 Results.AddResult(Result("inline"));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001380 break;
1381
John McCallf312b1e2010-08-26 23:41:50 +00001382 case Sema::PCC_ObjCInstanceVariableList:
1383 case Sema::PCC_Expression:
1384 case Sema::PCC_Statement:
1385 case Sema::PCC_ForInit:
1386 case Sema::PCC_Condition:
1387 case Sema::PCC_RecoveryInFunction:
1388 case Sema::PCC_Type:
Douglas Gregor02688102010-09-14 23:59:36 +00001389 case Sema::PCC_ParenthesizedExpression:
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00001390 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregor01dfea02010-01-10 23:08:15 +00001391 break;
1392 }
1393}
1394
Douglas Gregorbca403c2010-01-13 23:51:12 +00001395static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt);
1396static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt);
1397static void AddObjCVisibilityResults(const LangOptions &LangOpts,
Douglas Gregorc38c3e12010-01-13 21:54:15 +00001398 ResultBuilder &Results,
1399 bool NeedAt);
Douglas Gregorbca403c2010-01-13 23:51:12 +00001400static void AddObjCImplementationResults(const LangOptions &LangOpts,
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001401 ResultBuilder &Results,
1402 bool NeedAt);
Douglas Gregorbca403c2010-01-13 23:51:12 +00001403static void AddObjCInterfaceResults(const LangOptions &LangOpts,
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001404 ResultBuilder &Results,
1405 bool NeedAt);
Douglas Gregorbca403c2010-01-13 23:51:12 +00001406static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001407
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001408static void AddTypedefResult(ResultBuilder &Results) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00001409 CodeCompletionBuilder Builder(Results.getAllocator(),
1410 Results.getCodeCompletionTUInfo());
Douglas Gregor218937c2011-02-01 19:23:04 +00001411 Builder.AddTypedTextChunk("typedef");
1412 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1413 Builder.AddPlaceholderChunk("type");
1414 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1415 Builder.AddPlaceholderChunk("name");
1416 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001417}
1418
John McCallf312b1e2010-08-26 23:41:50 +00001419static bool WantTypesInContext(Sema::ParserCompletionContext CCC,
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001420 const LangOptions &LangOpts) {
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001421 switch (CCC) {
John McCallf312b1e2010-08-26 23:41:50 +00001422 case Sema::PCC_Namespace:
1423 case Sema::PCC_Class:
1424 case Sema::PCC_ObjCInstanceVariableList:
1425 case Sema::PCC_Template:
1426 case Sema::PCC_MemberTemplate:
1427 case Sema::PCC_Statement:
1428 case Sema::PCC_RecoveryInFunction:
1429 case Sema::PCC_Type:
Douglas Gregor02688102010-09-14 23:59:36 +00001430 case Sema::PCC_ParenthesizedExpression:
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00001431 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001432 return true;
1433
John McCallf312b1e2010-08-26 23:41:50 +00001434 case Sema::PCC_Expression:
1435 case Sema::PCC_Condition:
Douglas Gregor02688102010-09-14 23:59:36 +00001436 return LangOpts.CPlusPlus;
1437
1438 case Sema::PCC_ObjCInterface:
1439 case Sema::PCC_ObjCImplementation:
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001440 return false;
1441
John McCallf312b1e2010-08-26 23:41:50 +00001442 case Sema::PCC_ForInit:
Douglas Gregor02688102010-09-14 23:59:36 +00001443 return LangOpts.CPlusPlus || LangOpts.ObjC1 || LangOpts.C99;
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001444 }
David Blaikie7530c032012-01-17 06:56:22 +00001445
1446 llvm_unreachable("Invalid ParserCompletionContext!");
Douglas Gregor4710e5b2010-05-28 00:49:12 +00001447}
1448
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00001449static PrintingPolicy getCompletionPrintingPolicy(const ASTContext &Context,
1450 const Preprocessor &PP) {
1451 PrintingPolicy Policy = Sema::getPrintingPolicy(Context, PP);
Douglas Gregor8ca72082011-10-18 21:20:17 +00001452 Policy.AnonymousTagLocations = false;
1453 Policy.SuppressStrongLifetime = true;
Douglas Gregor25270b62011-11-03 00:16:13 +00001454 Policy.SuppressUnwrittenScope = true;
Douglas Gregor8ca72082011-10-18 21:20:17 +00001455 return Policy;
1456}
1457
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00001458/// \brief Retrieve a printing policy suitable for code completion.
1459static PrintingPolicy getCompletionPrintingPolicy(Sema &S) {
1460 return getCompletionPrintingPolicy(S.Context, S.PP);
1461}
1462
Douglas Gregor8ca72082011-10-18 21:20:17 +00001463/// \brief Retrieve the string representation of the given type as a string
1464/// that has the appropriate lifetime for code completion.
1465///
1466/// This routine provides a fast path where we provide constant strings for
1467/// common type names.
1468static const char *GetCompletionTypeString(QualType T,
1469 ASTContext &Context,
1470 const PrintingPolicy &Policy,
1471 CodeCompletionAllocator &Allocator) {
1472 if (!T.getLocalQualifiers()) {
1473 // Built-in type names are constant strings.
1474 if (const BuiltinType *BT = dyn_cast<BuiltinType>(T))
Argyrios Kyrtzidis27a00972012-05-05 04:20:28 +00001475 return BT->getNameAsCString(Policy);
Douglas Gregor8ca72082011-10-18 21:20:17 +00001476
1477 // Anonymous tag types are constant strings.
1478 if (const TagType *TagT = dyn_cast<TagType>(T))
1479 if (TagDecl *Tag = TagT->getDecl())
John McCall83972f12013-03-09 00:54:27 +00001480 if (!Tag->hasNameForLinkage()) {
Douglas Gregor8ca72082011-10-18 21:20:17 +00001481 switch (Tag->getTagKind()) {
1482 case TTK_Struct: return "struct <anonymous>";
Joao Matos6666ed42012-08-31 18:45:21 +00001483 case TTK_Interface: return "__interface <anonymous>";
1484 case TTK_Class: return "class <anonymous>";
Douglas Gregor8ca72082011-10-18 21:20:17 +00001485 case TTK_Union: return "union <anonymous>";
1486 case TTK_Enum: return "enum <anonymous>";
1487 }
1488 }
1489 }
1490
1491 // Slow path: format the type as a string.
1492 std::string Result;
1493 T.getAsStringInternal(Result, Policy);
1494 return Allocator.CopyString(Result);
1495}
1496
Douglas Gregor81f3bff2012-02-15 15:34:24 +00001497/// \brief Add a completion for "this", if we're in a member function.
1498static void addThisCompletion(Sema &S, ResultBuilder &Results) {
1499 QualType ThisTy = S.getCurrentThisType();
1500 if (ThisTy.isNull())
1501 return;
1502
1503 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00001504 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor81f3bff2012-02-15 15:34:24 +00001505 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
1506 Builder.AddResultTypeChunk(GetCompletionTypeString(ThisTy,
1507 S.Context,
1508 Policy,
1509 Allocator));
1510 Builder.AddTypedTextChunk("this");
Joao Matos6666ed42012-08-31 18:45:21 +00001511 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
Douglas Gregor81f3bff2012-02-15 15:34:24 +00001512}
1513
Douglas Gregor01dfea02010-01-10 23:08:15 +00001514/// \brief Add language constructs that show up for "ordinary" names.
John McCallf312b1e2010-08-26 23:41:50 +00001515static void AddOrdinaryNameResults(Sema::ParserCompletionContext CCC,
Douglas Gregor01dfea02010-01-10 23:08:15 +00001516 Scope *S,
1517 Sema &SemaRef,
Douglas Gregor01dfea02010-01-10 23:08:15 +00001518 ResultBuilder &Results) {
Douglas Gregor8ca72082011-10-18 21:20:17 +00001519 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00001520 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor8ca72082011-10-18 21:20:17 +00001521 PrintingPolicy Policy = getCompletionPrintingPolicy(SemaRef);
Douglas Gregor218937c2011-02-01 19:23:04 +00001522
John McCall0a2c5e22010-08-25 06:19:51 +00001523 typedef CodeCompletionResult Result;
Douglas Gregor01dfea02010-01-10 23:08:15 +00001524 switch (CCC) {
John McCallf312b1e2010-08-26 23:41:50 +00001525 case Sema::PCC_Namespace:
David Blaikie4e4d0842012-03-11 07:00:24 +00001526 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001527 if (Results.includeCodePatterns()) {
1528 // namespace <identifier> { declarations }
Douglas Gregor218937c2011-02-01 19:23:04 +00001529 Builder.AddTypedTextChunk("namespace");
1530 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1531 Builder.AddPlaceholderChunk("identifier");
1532 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1533 Builder.AddPlaceholderChunk("declarations");
1534 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1535 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1536 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001537 }
1538
Douglas Gregor01dfea02010-01-10 23:08:15 +00001539 // namespace identifier = identifier ;
Douglas Gregor218937c2011-02-01 19:23:04 +00001540 Builder.AddTypedTextChunk("namespace");
1541 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1542 Builder.AddPlaceholderChunk("name");
1543 Builder.AddChunk(CodeCompletionString::CK_Equal);
1544 Builder.AddPlaceholderChunk("namespace");
1545 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001546
1547 // Using directives
Douglas Gregor218937c2011-02-01 19:23:04 +00001548 Builder.AddTypedTextChunk("using");
1549 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1550 Builder.AddTextChunk("namespace");
1551 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1552 Builder.AddPlaceholderChunk("identifier");
1553 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001554
1555 // asm(string-literal)
Douglas Gregor218937c2011-02-01 19:23:04 +00001556 Builder.AddTypedTextChunk("asm");
1557 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1558 Builder.AddPlaceholderChunk("string-literal");
1559 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1560 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001561
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001562 if (Results.includeCodePatterns()) {
1563 // Explicit template instantiation
Douglas Gregor218937c2011-02-01 19:23:04 +00001564 Builder.AddTypedTextChunk("template");
1565 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1566 Builder.AddPlaceholderChunk("declaration");
1567 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001568 }
Douglas Gregor01dfea02010-01-10 23:08:15 +00001569 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001570
David Blaikie4e4d0842012-03-11 07:00:24 +00001571 if (SemaRef.getLangOpts().ObjC1)
Douglas Gregorbca403c2010-01-13 23:51:12 +00001572 AddObjCTopLevelResults(Results, true);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001573
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001574 AddTypedefResult(Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001575 // Fall through
1576
John McCallf312b1e2010-08-26 23:41:50 +00001577 case Sema::PCC_Class:
David Blaikie4e4d0842012-03-11 07:00:24 +00001578 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregor01dfea02010-01-10 23:08:15 +00001579 // Using declaration
Douglas Gregor218937c2011-02-01 19:23:04 +00001580 Builder.AddTypedTextChunk("using");
1581 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1582 Builder.AddPlaceholderChunk("qualifier");
1583 Builder.AddTextChunk("::");
1584 Builder.AddPlaceholderChunk("name");
1585 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001586
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001587 // using typename qualifier::name (only in a dependent context)
Douglas Gregor01dfea02010-01-10 23:08:15 +00001588 if (SemaRef.CurContext->isDependentContext()) {
Douglas Gregor218937c2011-02-01 19:23:04 +00001589 Builder.AddTypedTextChunk("using");
1590 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1591 Builder.AddTextChunk("typename");
1592 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1593 Builder.AddPlaceholderChunk("qualifier");
1594 Builder.AddTextChunk("::");
1595 Builder.AddPlaceholderChunk("name");
1596 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001597 }
1598
John McCallf312b1e2010-08-26 23:41:50 +00001599 if (CCC == Sema::PCC_Class) {
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001600 AddTypedefResult(Results);
1601
Douglas Gregor01dfea02010-01-10 23:08:15 +00001602 // public:
Douglas Gregor218937c2011-02-01 19:23:04 +00001603 Builder.AddTypedTextChunk("public");
Douglas Gregor10ccf122012-04-10 17:56:28 +00001604 if (Results.includeCodePatterns())
1605 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregor218937c2011-02-01 19:23:04 +00001606 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001607
1608 // protected:
Douglas Gregor218937c2011-02-01 19:23:04 +00001609 Builder.AddTypedTextChunk("protected");
Douglas Gregor10ccf122012-04-10 17:56:28 +00001610 if (Results.includeCodePatterns())
1611 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregor218937c2011-02-01 19:23:04 +00001612 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001613
1614 // private:
Douglas Gregor218937c2011-02-01 19:23:04 +00001615 Builder.AddTypedTextChunk("private");
Douglas Gregor10ccf122012-04-10 17:56:28 +00001616 if (Results.includeCodePatterns())
1617 Builder.AddChunk(CodeCompletionString::CK_Colon);
Douglas Gregor218937c2011-02-01 19:23:04 +00001618 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001619 }
1620 }
1621 // Fall through
1622
John McCallf312b1e2010-08-26 23:41:50 +00001623 case Sema::PCC_Template:
1624 case Sema::PCC_MemberTemplate:
David Blaikie4e4d0842012-03-11 07:00:24 +00001625 if (SemaRef.getLangOpts().CPlusPlus && Results.includeCodePatterns()) {
Douglas Gregor01dfea02010-01-10 23:08:15 +00001626 // template < parameters >
Douglas Gregor218937c2011-02-01 19:23:04 +00001627 Builder.AddTypedTextChunk("template");
1628 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1629 Builder.AddPlaceholderChunk("parameters");
1630 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1631 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001632 }
1633
David Blaikie4e4d0842012-03-11 07:00:24 +00001634 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1635 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001636 break;
1637
John McCallf312b1e2010-08-26 23:41:50 +00001638 case Sema::PCC_ObjCInterface:
David Blaikie4e4d0842012-03-11 07:00:24 +00001639 AddObjCInterfaceResults(SemaRef.getLangOpts(), Results, true);
1640 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1641 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001642 break;
1643
John McCallf312b1e2010-08-26 23:41:50 +00001644 case Sema::PCC_ObjCImplementation:
David Blaikie4e4d0842012-03-11 07:00:24 +00001645 AddObjCImplementationResults(SemaRef.getLangOpts(), Results, true);
1646 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
1647 AddFunctionSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001648 break;
1649
John McCallf312b1e2010-08-26 23:41:50 +00001650 case Sema::PCC_ObjCInstanceVariableList:
David Blaikie4e4d0842012-03-11 07:00:24 +00001651 AddObjCVisibilityResults(SemaRef.getLangOpts(), Results, true);
Douglas Gregorc38c3e12010-01-13 21:54:15 +00001652 break;
1653
John McCallf312b1e2010-08-26 23:41:50 +00001654 case Sema::PCC_RecoveryInFunction:
1655 case Sema::PCC_Statement: {
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001656 AddTypedefResult(Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001657
David Blaikie4e4d0842012-03-11 07:00:24 +00001658 if (SemaRef.getLangOpts().CPlusPlus && Results.includeCodePatterns() &&
1659 SemaRef.getLangOpts().CXXExceptions) {
Douglas Gregor218937c2011-02-01 19:23:04 +00001660 Builder.AddTypedTextChunk("try");
1661 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1662 Builder.AddPlaceholderChunk("statements");
1663 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1664 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1665 Builder.AddTextChunk("catch");
1666 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1667 Builder.AddPlaceholderChunk("declaration");
1668 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1669 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1670 Builder.AddPlaceholderChunk("statements");
1671 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1672 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1673 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001674 }
David Blaikie4e4d0842012-03-11 07:00:24 +00001675 if (SemaRef.getLangOpts().ObjC1)
Douglas Gregorbca403c2010-01-13 23:51:12 +00001676 AddObjCStatementResults(Results, true);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00001677
Douglas Gregord8e8a582010-05-25 21:41:55 +00001678 if (Results.includeCodePatterns()) {
1679 // if (condition) { statements }
Douglas Gregor218937c2011-02-01 19:23:04 +00001680 Builder.AddTypedTextChunk("if");
1681 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikie4e4d0842012-03-11 07:00:24 +00001682 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor218937c2011-02-01 19:23:04 +00001683 Builder.AddPlaceholderChunk("condition");
Douglas Gregord8e8a582010-05-25 21:41:55 +00001684 else
Douglas Gregor218937c2011-02-01 19:23:04 +00001685 Builder.AddPlaceholderChunk("expression");
1686 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1687 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1688 Builder.AddPlaceholderChunk("statements");
1689 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1690 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1691 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001692
Douglas Gregord8e8a582010-05-25 21:41:55 +00001693 // switch (condition) { }
Douglas Gregor218937c2011-02-01 19:23:04 +00001694 Builder.AddTypedTextChunk("switch");
1695 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikie4e4d0842012-03-11 07:00:24 +00001696 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor218937c2011-02-01 19:23:04 +00001697 Builder.AddPlaceholderChunk("condition");
Douglas Gregord8e8a582010-05-25 21:41:55 +00001698 else
Douglas Gregor218937c2011-02-01 19:23:04 +00001699 Builder.AddPlaceholderChunk("expression");
1700 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1701 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1702 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1703 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1704 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregord8e8a582010-05-25 21:41:55 +00001705 }
1706
Douglas Gregor01dfea02010-01-10 23:08:15 +00001707 // Switch-specific statements.
John McCall781472f2010-08-25 08:40:02 +00001708 if (!SemaRef.getCurFunction()->SwitchStack.empty()) {
Douglas Gregor01dfea02010-01-10 23:08:15 +00001709 // case expression:
Douglas Gregor218937c2011-02-01 19:23:04 +00001710 Builder.AddTypedTextChunk("case");
1711 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1712 Builder.AddPlaceholderChunk("expression");
1713 Builder.AddChunk(CodeCompletionString::CK_Colon);
1714 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001715
1716 // default:
Douglas Gregor218937c2011-02-01 19:23:04 +00001717 Builder.AddTypedTextChunk("default");
1718 Builder.AddChunk(CodeCompletionString::CK_Colon);
1719 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001720 }
1721
Douglas Gregord8e8a582010-05-25 21:41:55 +00001722 if (Results.includeCodePatterns()) {
1723 /// while (condition) { statements }
Douglas Gregor218937c2011-02-01 19:23:04 +00001724 Builder.AddTypedTextChunk("while");
1725 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikie4e4d0842012-03-11 07:00:24 +00001726 if (SemaRef.getLangOpts().CPlusPlus)
Douglas Gregor218937c2011-02-01 19:23:04 +00001727 Builder.AddPlaceholderChunk("condition");
Douglas Gregord8e8a582010-05-25 21:41:55 +00001728 else
Douglas Gregor218937c2011-02-01 19:23:04 +00001729 Builder.AddPlaceholderChunk("expression");
1730 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1731 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1732 Builder.AddPlaceholderChunk("statements");
1733 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1734 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1735 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregord8e8a582010-05-25 21:41:55 +00001736
1737 // do { statements } while ( expression );
Douglas Gregor218937c2011-02-01 19:23:04 +00001738 Builder.AddTypedTextChunk("do");
1739 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1740 Builder.AddPlaceholderChunk("statements");
1741 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1742 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1743 Builder.AddTextChunk("while");
1744 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1745 Builder.AddPlaceholderChunk("expression");
1746 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1747 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001748
Douglas Gregord8e8a582010-05-25 21:41:55 +00001749 // for ( for-init-statement ; condition ; expression ) { statements }
Douglas Gregor218937c2011-02-01 19:23:04 +00001750 Builder.AddTypedTextChunk("for");
1751 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikie4e4d0842012-03-11 07:00:24 +00001752 if (SemaRef.getLangOpts().CPlusPlus || SemaRef.getLangOpts().C99)
Douglas Gregor218937c2011-02-01 19:23:04 +00001753 Builder.AddPlaceholderChunk("init-statement");
Douglas Gregord8e8a582010-05-25 21:41:55 +00001754 else
Douglas Gregor218937c2011-02-01 19:23:04 +00001755 Builder.AddPlaceholderChunk("init-expression");
1756 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
1757 Builder.AddPlaceholderChunk("condition");
1758 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
1759 Builder.AddPlaceholderChunk("inc-expression");
1760 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1761 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
1762 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1763 Builder.AddPlaceholderChunk("statements");
1764 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
1765 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
1766 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregord8e8a582010-05-25 21:41:55 +00001767 }
Douglas Gregor01dfea02010-01-10 23:08:15 +00001768
1769 if (S->getContinueParent()) {
1770 // continue ;
Douglas Gregor218937c2011-02-01 19:23:04 +00001771 Builder.AddTypedTextChunk("continue");
1772 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001773 }
1774
1775 if (S->getBreakParent()) {
1776 // break ;
Douglas Gregor218937c2011-02-01 19:23:04 +00001777 Builder.AddTypedTextChunk("break");
1778 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001779 }
1780
1781 // "return expression ;" or "return ;", depending on whether we
1782 // know the function is void or not.
1783 bool isVoid = false;
1784 if (FunctionDecl *Function = dyn_cast<FunctionDecl>(SemaRef.CurContext))
Stephen Hines651f13c2014-04-23 16:59:28 -07001785 isVoid = Function->getReturnType()->isVoidType();
Douglas Gregor01dfea02010-01-10 23:08:15 +00001786 else if (ObjCMethodDecl *Method
1787 = dyn_cast<ObjCMethodDecl>(SemaRef.CurContext))
Stephen Hines651f13c2014-04-23 16:59:28 -07001788 isVoid = Method->getReturnType()->isVoidType();
Douglas Gregor9ea9bdb2010-03-01 23:15:13 +00001789 else if (SemaRef.getCurBlock() &&
1790 !SemaRef.getCurBlock()->ReturnType.isNull())
1791 isVoid = SemaRef.getCurBlock()->ReturnType->isVoidType();
Douglas Gregor218937c2011-02-01 19:23:04 +00001792 Builder.AddTypedTextChunk("return");
Douglas Gregor93298002010-02-18 04:06:48 +00001793 if (!isVoid) {
Douglas Gregor218937c2011-02-01 19:23:04 +00001794 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1795 Builder.AddPlaceholderChunk("expression");
Douglas Gregor93298002010-02-18 04:06:48 +00001796 }
Douglas Gregor218937c2011-02-01 19:23:04 +00001797 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001798
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001799 // goto identifier ;
Douglas Gregor218937c2011-02-01 19:23:04 +00001800 Builder.AddTypedTextChunk("goto");
1801 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1802 Builder.AddPlaceholderChunk("label");
1803 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001804
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001805 // Using directives
Douglas Gregor218937c2011-02-01 19:23:04 +00001806 Builder.AddTypedTextChunk("using");
1807 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1808 Builder.AddTextChunk("namespace");
1809 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1810 Builder.AddPlaceholderChunk("identifier");
1811 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001812 }
1813
1814 // Fall through (for statement expressions).
John McCallf312b1e2010-08-26 23:41:50 +00001815 case Sema::PCC_ForInit:
1816 case Sema::PCC_Condition:
David Blaikie4e4d0842012-03-11 07:00:24 +00001817 AddStorageSpecifiers(CCC, SemaRef.getLangOpts(), Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001818 // Fall through: conditions and statements can have expressions.
1819
Douglas Gregor02688102010-09-14 23:59:36 +00001820 case Sema::PCC_ParenthesizedExpression:
David Blaikie4e4d0842012-03-11 07:00:24 +00001821 if (SemaRef.getLangOpts().ObjCAutoRefCount &&
John McCallf85e1932011-06-15 23:02:42 +00001822 CCC == Sema::PCC_ParenthesizedExpression) {
1823 // (__bridge <type>)<expression>
1824 Builder.AddTypedTextChunk("__bridge");
1825 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1826 Builder.AddPlaceholderChunk("type");
1827 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1828 Builder.AddPlaceholderChunk("expression");
1829 Results.AddResult(Result(Builder.TakeString()));
1830
1831 // (__bridge_transfer <Objective-C type>)<expression>
1832 Builder.AddTypedTextChunk("__bridge_transfer");
1833 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1834 Builder.AddPlaceholderChunk("Objective-C type");
1835 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1836 Builder.AddPlaceholderChunk("expression");
1837 Results.AddResult(Result(Builder.TakeString()));
1838
1839 // (__bridge_retained <CF type>)<expression>
1840 Builder.AddTypedTextChunk("__bridge_retained");
1841 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1842 Builder.AddPlaceholderChunk("CF type");
1843 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1844 Builder.AddPlaceholderChunk("expression");
1845 Results.AddResult(Result(Builder.TakeString()));
1846 }
1847 // Fall through
1848
John McCallf312b1e2010-08-26 23:41:50 +00001849 case Sema::PCC_Expression: {
David Blaikie4e4d0842012-03-11 07:00:24 +00001850 if (SemaRef.getLangOpts().CPlusPlus) {
Douglas Gregor01dfea02010-01-10 23:08:15 +00001851 // 'this', if we're in a non-static member function.
Douglas Gregor81f3bff2012-02-15 15:34:24 +00001852 addThisCompletion(SemaRef, Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00001853
Douglas Gregor8ca72082011-10-18 21:20:17 +00001854 // true
1855 Builder.AddResultTypeChunk("bool");
1856 Builder.AddTypedTextChunk("true");
1857 Results.AddResult(Result(Builder.TakeString()));
1858
1859 // false
1860 Builder.AddResultTypeChunk("bool");
1861 Builder.AddTypedTextChunk("false");
1862 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001863
David Blaikie4e4d0842012-03-11 07:00:24 +00001864 if (SemaRef.getLangOpts().RTTI) {
Douglas Gregorec3310a2011-04-12 02:47:21 +00001865 // dynamic_cast < type-id > ( expression )
1866 Builder.AddTypedTextChunk("dynamic_cast");
1867 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1868 Builder.AddPlaceholderChunk("type");
1869 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1870 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1871 Builder.AddPlaceholderChunk("expression");
1872 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1873 Results.AddResult(Result(Builder.TakeString()));
1874 }
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001875
1876 // static_cast < type-id > ( expression )
Douglas Gregor218937c2011-02-01 19:23:04 +00001877 Builder.AddTypedTextChunk("static_cast");
1878 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1879 Builder.AddPlaceholderChunk("type");
1880 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1881 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1882 Builder.AddPlaceholderChunk("expression");
1883 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1884 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001885
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001886 // reinterpret_cast < type-id > ( expression )
Douglas Gregor218937c2011-02-01 19:23:04 +00001887 Builder.AddTypedTextChunk("reinterpret_cast");
1888 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1889 Builder.AddPlaceholderChunk("type");
1890 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1891 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1892 Builder.AddPlaceholderChunk("expression");
1893 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1894 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001895
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001896 // const_cast < type-id > ( expression )
Douglas Gregor218937c2011-02-01 19:23:04 +00001897 Builder.AddTypedTextChunk("const_cast");
1898 Builder.AddChunk(CodeCompletionString::CK_LeftAngle);
1899 Builder.AddPlaceholderChunk("type");
1900 Builder.AddChunk(CodeCompletionString::CK_RightAngle);
1901 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1902 Builder.AddPlaceholderChunk("expression");
1903 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1904 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001905
David Blaikie4e4d0842012-03-11 07:00:24 +00001906 if (SemaRef.getLangOpts().RTTI) {
Douglas Gregorec3310a2011-04-12 02:47:21 +00001907 // typeid ( expression-or-type )
Douglas Gregor8ca72082011-10-18 21:20:17 +00001908 Builder.AddResultTypeChunk("std::type_info");
Douglas Gregorec3310a2011-04-12 02:47:21 +00001909 Builder.AddTypedTextChunk("typeid");
1910 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1911 Builder.AddPlaceholderChunk("expression-or-type");
1912 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1913 Results.AddResult(Result(Builder.TakeString()));
1914 }
1915
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001916 // new T ( ... )
Douglas Gregor218937c2011-02-01 19:23:04 +00001917 Builder.AddTypedTextChunk("new");
1918 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1919 Builder.AddPlaceholderChunk("type");
1920 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1921 Builder.AddPlaceholderChunk("expressions");
1922 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1923 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001924
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001925 // new T [ ] ( ... )
Douglas Gregor218937c2011-02-01 19:23:04 +00001926 Builder.AddTypedTextChunk("new");
1927 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1928 Builder.AddPlaceholderChunk("type");
1929 Builder.AddChunk(CodeCompletionString::CK_LeftBracket);
1930 Builder.AddPlaceholderChunk("size");
1931 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
1932 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1933 Builder.AddPlaceholderChunk("expressions");
1934 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1935 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001936
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001937 // delete expression
Douglas Gregor8ca72082011-10-18 21:20:17 +00001938 Builder.AddResultTypeChunk("void");
Douglas Gregor218937c2011-02-01 19:23:04 +00001939 Builder.AddTypedTextChunk("delete");
1940 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1941 Builder.AddPlaceholderChunk("expression");
1942 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001943
Douglas Gregorc8bddde2010-05-28 00:22:41 +00001944 // delete [] expression
Douglas Gregor8ca72082011-10-18 21:20:17 +00001945 Builder.AddResultTypeChunk("void");
Douglas Gregor218937c2011-02-01 19:23:04 +00001946 Builder.AddTypedTextChunk("delete");
1947 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1948 Builder.AddChunk(CodeCompletionString::CK_LeftBracket);
1949 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
1950 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1951 Builder.AddPlaceholderChunk("expression");
1952 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00001953
David Blaikie4e4d0842012-03-11 07:00:24 +00001954 if (SemaRef.getLangOpts().CXXExceptions) {
Douglas Gregorec3310a2011-04-12 02:47:21 +00001955 // throw expression
Douglas Gregor8ca72082011-10-18 21:20:17 +00001956 Builder.AddResultTypeChunk("void");
Douglas Gregorec3310a2011-04-12 02:47:21 +00001957 Builder.AddTypedTextChunk("throw");
1958 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
1959 Builder.AddPlaceholderChunk("expression");
1960 Results.AddResult(Result(Builder.TakeString()));
1961 }
Douglas Gregora50216c2011-10-18 16:29:03 +00001962
Douglas Gregor12e13132010-05-26 22:00:08 +00001963 // FIXME: Rethrow?
Douglas Gregora50216c2011-10-18 16:29:03 +00001964
Richard Smith80ad52f2013-01-02 11:42:31 +00001965 if (SemaRef.getLangOpts().CPlusPlus11) {
Douglas Gregora50216c2011-10-18 16:29:03 +00001966 // nullptr
Douglas Gregor8ca72082011-10-18 21:20:17 +00001967 Builder.AddResultTypeChunk("std::nullptr_t");
Douglas Gregora50216c2011-10-18 16:29:03 +00001968 Builder.AddTypedTextChunk("nullptr");
1969 Results.AddResult(Result(Builder.TakeString()));
1970
1971 // alignof
Douglas Gregor8ca72082011-10-18 21:20:17 +00001972 Builder.AddResultTypeChunk("size_t");
Douglas Gregora50216c2011-10-18 16:29:03 +00001973 Builder.AddTypedTextChunk("alignof");
1974 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1975 Builder.AddPlaceholderChunk("type");
1976 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1977 Results.AddResult(Result(Builder.TakeString()));
1978
1979 // noexcept
Douglas Gregor8ca72082011-10-18 21:20:17 +00001980 Builder.AddResultTypeChunk("bool");
Douglas Gregora50216c2011-10-18 16:29:03 +00001981 Builder.AddTypedTextChunk("noexcept");
1982 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1983 Builder.AddPlaceholderChunk("expression");
1984 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1985 Results.AddResult(Result(Builder.TakeString()));
1986
1987 // sizeof... expression
Douglas Gregor8ca72082011-10-18 21:20:17 +00001988 Builder.AddResultTypeChunk("size_t");
Douglas Gregora50216c2011-10-18 16:29:03 +00001989 Builder.AddTypedTextChunk("sizeof...");
1990 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
1991 Builder.AddPlaceholderChunk("parameter-pack");
1992 Builder.AddChunk(CodeCompletionString::CK_RightParen);
1993 Results.AddResult(Result(Builder.TakeString()));
1994 }
Douglas Gregor01dfea02010-01-10 23:08:15 +00001995 }
1996
David Blaikie4e4d0842012-03-11 07:00:24 +00001997 if (SemaRef.getLangOpts().ObjC1) {
Douglas Gregor01dfea02010-01-10 23:08:15 +00001998 // Add "super", if we're in an Objective-C class with a superclass.
Ted Kremenek681e2562010-05-31 21:43:10 +00001999 if (ObjCMethodDecl *Method = SemaRef.getCurMethodDecl()) {
2000 // The interface can be NULL.
2001 if (ObjCInterfaceDecl *ID = Method->getClassInterface())
Douglas Gregor8ca72082011-10-18 21:20:17 +00002002 if (ID->getSuperClass()) {
2003 std::string SuperType;
2004 SuperType = ID->getSuperClass()->getNameAsString();
2005 if (Method->isInstanceMethod())
2006 SuperType += " *";
2007
2008 Builder.AddResultTypeChunk(Allocator.CopyString(SuperType));
2009 Builder.AddTypedTextChunk("super");
2010 Results.AddResult(Result(Builder.TakeString()));
2011 }
Ted Kremenek681e2562010-05-31 21:43:10 +00002012 }
2013
Douglas Gregorbca403c2010-01-13 23:51:12 +00002014 AddObjCExpressionResults(Results, true);
Douglas Gregor01dfea02010-01-10 23:08:15 +00002015 }
2016
Jordan Rosef70a8862012-06-30 21:33:57 +00002017 if (SemaRef.getLangOpts().C11) {
2018 // _Alignof
2019 Builder.AddResultTypeChunk("size_t");
2020 if (SemaRef.getASTContext().Idents.get("alignof").hasMacroDefinition())
2021 Builder.AddTypedTextChunk("alignof");
2022 else
2023 Builder.AddTypedTextChunk("_Alignof");
2024 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2025 Builder.AddPlaceholderChunk("type");
2026 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2027 Results.AddResult(Result(Builder.TakeString()));
2028 }
2029
Douglas Gregorc8bddde2010-05-28 00:22:41 +00002030 // sizeof expression
Douglas Gregor8ca72082011-10-18 21:20:17 +00002031 Builder.AddResultTypeChunk("size_t");
Douglas Gregor218937c2011-02-01 19:23:04 +00002032 Builder.AddTypedTextChunk("sizeof");
2033 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
2034 Builder.AddPlaceholderChunk("expression-or-type");
2035 Builder.AddChunk(CodeCompletionString::CK_RightParen);
2036 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor01dfea02010-01-10 23:08:15 +00002037 break;
2038 }
Douglas Gregord32b0222010-08-24 01:06:58 +00002039
John McCallf312b1e2010-08-26 23:41:50 +00002040 case Sema::PCC_Type:
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00002041 case Sema::PCC_LocalDeclarationSpecifiers:
Douglas Gregord32b0222010-08-24 01:06:58 +00002042 break;
Douglas Gregor01dfea02010-01-10 23:08:15 +00002043 }
2044
David Blaikie4e4d0842012-03-11 07:00:24 +00002045 if (WantTypesInContext(CCC, SemaRef.getLangOpts()))
2046 AddTypeSpecifierResults(SemaRef.getLangOpts(), Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00002047
David Blaikie4e4d0842012-03-11 07:00:24 +00002048 if (SemaRef.getLangOpts().CPlusPlus && CCC != Sema::PCC_Type)
Douglas Gregora4477812010-01-14 16:01:26 +00002049 Results.AddResult(Result("operator"));
Douglas Gregor01dfea02010-01-10 23:08:15 +00002050}
2051
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002052/// \brief If the given declaration has an associated type, add it as a result
2053/// type chunk.
2054static void AddResultTypeChunk(ASTContext &Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00002055 const PrintingPolicy &Policy,
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002056 const NamedDecl *ND,
Douglas Gregor218937c2011-02-01 19:23:04 +00002057 CodeCompletionBuilder &Result) {
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002058 if (!ND)
2059 return;
Douglas Gregor6f942b22010-09-21 16:06:22 +00002060
2061 // Skip constructors and conversion functions, which have their return types
2062 // built into their names.
2063 if (isa<CXXConstructorDecl>(ND) || isa<CXXConversionDecl>(ND))
2064 return;
2065
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002066 // Determine the type of the declaration (if it has a type).
Stephen Hines651f13c2014-04-23 16:59:28 -07002067 QualType T;
2068 if (const FunctionDecl *Function = ND->getAsFunction())
2069 T = Function->getReturnType();
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002070 else if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND))
Stephen Hines651f13c2014-04-23 16:59:28 -07002071 T = Method->getReturnType();
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002072 else if (const EnumConstantDecl *Enumerator = dyn_cast<EnumConstantDecl>(ND))
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002073 T = Context.getTypeDeclType(cast<TypeDecl>(Enumerator->getDeclContext()));
2074 else if (isa<UnresolvedUsingValueDecl>(ND)) {
2075 /* Do nothing: ignore unresolved using declarations*/
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002076 } else if (const ValueDecl *Value = dyn_cast<ValueDecl>(ND)) {
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002077 T = Value->getType();
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002078 } else if (const ObjCPropertyDecl *Property = dyn_cast<ObjCPropertyDecl>(ND))
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002079 T = Property->getType();
2080
2081 if (T.isNull() || Context.hasSameType(T, Context.DependentTy))
2082 return;
2083
Douglas Gregor8987b232011-09-27 23:30:47 +00002084 Result.AddResultTypeChunk(GetCompletionTypeString(T, Context, Policy,
Douglas Gregora63f6de2011-02-01 21:15:40 +00002085 Result.getAllocator()));
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002086}
2087
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002088static void MaybeAddSentinel(ASTContext &Context,
2089 const NamedDecl *FunctionOrMethod,
Douglas Gregor218937c2011-02-01 19:23:04 +00002090 CodeCompletionBuilder &Result) {
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002091 if (SentinelAttr *Sentinel = FunctionOrMethod->getAttr<SentinelAttr>())
2092 if (Sentinel->getSentinel() == 0) {
David Blaikie4e4d0842012-03-11 07:00:24 +00002093 if (Context.getLangOpts().ObjC1 &&
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002094 Context.Idents.get("nil").hasMacroDefinition())
Douglas Gregor218937c2011-02-01 19:23:04 +00002095 Result.AddTextChunk(", nil");
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002096 else if (Context.Idents.get("NULL").hasMacroDefinition())
Douglas Gregor218937c2011-02-01 19:23:04 +00002097 Result.AddTextChunk(", NULL");
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002098 else
Douglas Gregor218937c2011-02-01 19:23:04 +00002099 Result.AddTextChunk(", (void*)0");
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002100 }
2101}
2102
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002103static std::string formatObjCParamQualifiers(unsigned ObjCQuals) {
2104 std::string Result;
2105 if (ObjCQuals & Decl::OBJC_TQ_In)
Douglas Gregor6ef92092011-11-09 02:13:45 +00002106 Result += "in ";
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002107 else if (ObjCQuals & Decl::OBJC_TQ_Inout)
Douglas Gregor6ef92092011-11-09 02:13:45 +00002108 Result += "inout ";
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002109 else if (ObjCQuals & Decl::OBJC_TQ_Out)
Douglas Gregor6ef92092011-11-09 02:13:45 +00002110 Result += "out ";
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002111 if (ObjCQuals & Decl::OBJC_TQ_Bycopy)
Douglas Gregor6ef92092011-11-09 02:13:45 +00002112 Result += "bycopy ";
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002113 else if (ObjCQuals & Decl::OBJC_TQ_Byref)
Douglas Gregor6ef92092011-11-09 02:13:45 +00002114 Result += "byref ";
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002115 if (ObjCQuals & Decl::OBJC_TQ_Oneway)
Douglas Gregor6ef92092011-11-09 02:13:45 +00002116 Result += "oneway ";
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002117 return Result;
2118}
2119
Douglas Gregor83482d12010-08-24 16:15:59 +00002120static std::string FormatFunctionParameter(ASTContext &Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00002121 const PrintingPolicy &Policy,
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002122 const ParmVarDecl *Param,
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002123 bool SuppressName = false,
2124 bool SuppressBlock = false) {
Douglas Gregor83482d12010-08-24 16:15:59 +00002125 bool ObjCMethodParam = isa<ObjCMethodDecl>(Param->getDeclContext());
2126 if (Param->getType()->isDependentType() ||
2127 !Param->getType()->isBlockPointerType()) {
2128 // The argument for a dependent or non-block parameter is a placeholder
2129 // containing that parameter's type.
2130 std::string Result;
2131
Douglas Gregoraba48082010-08-29 19:47:46 +00002132 if (Param->getIdentifier() && !ObjCMethodParam && !SuppressName)
Douglas Gregor83482d12010-08-24 16:15:59 +00002133 Result = Param->getIdentifier()->getName();
2134
John McCallf85e1932011-06-15 23:02:42 +00002135 Param->getType().getAsStringInternal(Result, Policy);
Douglas Gregor83482d12010-08-24 16:15:59 +00002136
2137 if (ObjCMethodParam) {
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002138 Result = "(" + formatObjCParamQualifiers(Param->getObjCDeclQualifier())
2139 + Result + ")";
Douglas Gregoraba48082010-08-29 19:47:46 +00002140 if (Param->getIdentifier() && !SuppressName)
Douglas Gregor83482d12010-08-24 16:15:59 +00002141 Result += Param->getIdentifier()->getName();
2142 }
2143 return Result;
2144 }
2145
2146 // The argument for a block pointer parameter is a block literal with
2147 // the appropriate type.
David Blaikie39e6ab42013-02-18 22:06:02 +00002148 FunctionTypeLoc Block;
2149 FunctionProtoTypeLoc BlockProto;
Douglas Gregor83482d12010-08-24 16:15:59 +00002150 TypeLoc TL;
2151 if (TypeSourceInfo *TSInfo = Param->getTypeSourceInfo()) {
2152 TL = TSInfo->getTypeLoc().getUnqualifiedLoc();
2153 while (true) {
2154 // Look through typedefs.
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002155 if (!SuppressBlock) {
David Blaikie39e6ab42013-02-18 22:06:02 +00002156 if (TypedefTypeLoc TypedefTL = TL.getAs<TypedefTypeLoc>()) {
2157 if (TypeSourceInfo *InnerTSInfo =
2158 TypedefTL.getTypedefNameDecl()->getTypeSourceInfo()) {
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002159 TL = InnerTSInfo->getTypeLoc().getUnqualifiedLoc();
2160 continue;
2161 }
2162 }
2163
2164 // Look through qualified types
David Blaikie39e6ab42013-02-18 22:06:02 +00002165 if (QualifiedTypeLoc QualifiedTL = TL.getAs<QualifiedTypeLoc>()) {
2166 TL = QualifiedTL.getUnqualifiedLoc();
Douglas Gregor83482d12010-08-24 16:15:59 +00002167 continue;
2168 }
2169 }
2170
Douglas Gregor83482d12010-08-24 16:15:59 +00002171 // Try to get the function prototype behind the block pointer type,
2172 // then we're done.
David Blaikie39e6ab42013-02-18 22:06:02 +00002173 if (BlockPointerTypeLoc BlockPtr = TL.getAs<BlockPointerTypeLoc>()) {
2174 TL = BlockPtr.getPointeeLoc().IgnoreParens();
2175 Block = TL.getAs<FunctionTypeLoc>();
2176 BlockProto = TL.getAs<FunctionProtoTypeLoc>();
Douglas Gregor83482d12010-08-24 16:15:59 +00002177 }
2178 break;
2179 }
2180 }
2181
2182 if (!Block) {
2183 // We were unable to find a FunctionProtoTypeLoc with parameter names
2184 // for the block; just use the parameter type as a placeholder.
2185 std::string Result;
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002186 if (!ObjCMethodParam && Param->getIdentifier())
2187 Result = Param->getIdentifier()->getName();
2188
John McCallf85e1932011-06-15 23:02:42 +00002189 Param->getType().getUnqualifiedType().getAsStringInternal(Result, Policy);
Douglas Gregor83482d12010-08-24 16:15:59 +00002190
2191 if (ObjCMethodParam) {
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002192 Result = "(" + formatObjCParamQualifiers(Param->getObjCDeclQualifier())
2193 + Result + ")";
Douglas Gregor83482d12010-08-24 16:15:59 +00002194 if (Param->getIdentifier())
2195 Result += Param->getIdentifier()->getName();
2196 }
2197
2198 return Result;
2199 }
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002200
Douglas Gregor83482d12010-08-24 16:15:59 +00002201 // We have the function prototype behind the block pointer type, as it was
2202 // written in the source.
Douglas Gregor38276252010-09-08 22:47:51 +00002203 std::string Result;
Stephen Hines651f13c2014-04-23 16:59:28 -07002204 QualType ResultType = Block.getTypePtr()->getReturnType();
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002205 if (!ResultType->isVoidType() || SuppressBlock)
John McCallf85e1932011-06-15 23:02:42 +00002206 ResultType.getAsStringInternal(Result, Policy);
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002207
2208 // Format the parameter list.
2209 std::string Params;
Stephen Hines651f13c2014-04-23 16:59:28 -07002210 if (!BlockProto || Block.getNumParams() == 0) {
David Blaikie39e6ab42013-02-18 22:06:02 +00002211 if (BlockProto && BlockProto.getTypePtr()->isVariadic())
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002212 Params = "(...)";
Douglas Gregorc2760bc2010-10-02 23:49:58 +00002213 else
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002214 Params = "(void)";
Douglas Gregor38276252010-09-08 22:47:51 +00002215 } else {
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002216 Params += "(";
Stephen Hines651f13c2014-04-23 16:59:28 -07002217 for (unsigned I = 0, N = Block.getNumParams(); I != N; ++I) {
Douglas Gregor38276252010-09-08 22:47:51 +00002218 if (I)
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002219 Params += ", ";
Stephen Hines651f13c2014-04-23 16:59:28 -07002220 Params += FormatFunctionParameter(Context, Policy, Block.getParam(I),
2221 /*SuppressName=*/false,
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002222 /*SuppressBlock=*/true);
Stephen Hines651f13c2014-04-23 16:59:28 -07002223
David Blaikie39e6ab42013-02-18 22:06:02 +00002224 if (I == N - 1 && BlockProto.getTypePtr()->isVariadic())
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002225 Params += ", ...";
Douglas Gregor38276252010-09-08 22:47:51 +00002226 }
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002227 Params += ")";
Douglas Gregore17794f2010-08-31 05:13:43 +00002228 }
Douglas Gregor38276252010-09-08 22:47:51 +00002229
Douglas Gregoree1c68a2011-10-18 04:23:19 +00002230 if (SuppressBlock) {
2231 // Format as a parameter.
2232 Result = Result + " (^";
2233 if (Param->getIdentifier())
2234 Result += Param->getIdentifier()->getName();
2235 Result += ")";
2236 Result += Params;
2237 } else {
2238 // Format as a block literal argument.
2239 Result = '^' + Result;
2240 Result += Params;
2241
2242 if (Param->getIdentifier())
2243 Result += Param->getIdentifier()->getName();
2244 }
2245
Douglas Gregor83482d12010-08-24 16:15:59 +00002246 return Result;
2247}
2248
Douglas Gregor86d9a522009-09-21 16:56:56 +00002249/// \brief Add function parameter chunks to the given code completion string.
2250static void AddFunctionParameterChunks(ASTContext &Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00002251 const PrintingPolicy &Policy,
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002252 const FunctionDecl *Function,
Douglas Gregor218937c2011-02-01 19:23:04 +00002253 CodeCompletionBuilder &Result,
2254 unsigned Start = 0,
2255 bool InOptional = false) {
Douglas Gregor218937c2011-02-01 19:23:04 +00002256 bool FirstParameter = true;
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002257
Douglas Gregor218937c2011-02-01 19:23:04 +00002258 for (unsigned P = Start, N = Function->getNumParams(); P != N; ++P) {
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002259 const ParmVarDecl *Param = Function->getParamDecl(P);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002260
Douglas Gregor218937c2011-02-01 19:23:04 +00002261 if (Param->hasDefaultArg() && !InOptional) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00002262 // When we see an optional default argument, put that argument and
2263 // the remaining default arguments into a new, optional string.
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00002264 CodeCompletionBuilder Opt(Result.getAllocator(),
2265 Result.getCodeCompletionTUInfo());
Douglas Gregor218937c2011-02-01 19:23:04 +00002266 if (!FirstParameter)
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002267 Opt.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor8987b232011-09-27 23:30:47 +00002268 AddFunctionParameterChunks(Context, Policy, Function, Opt, P, true);
Douglas Gregor218937c2011-02-01 19:23:04 +00002269 Result.AddOptionalChunk(Opt.TakeString());
2270 break;
Douglas Gregor86d9a522009-09-21 16:56:56 +00002271 }
2272
Douglas Gregor218937c2011-02-01 19:23:04 +00002273 if (FirstParameter)
2274 FirstParameter = false;
2275 else
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002276 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor218937c2011-02-01 19:23:04 +00002277
2278 InOptional = false;
Douglas Gregor86d9a522009-09-21 16:56:56 +00002279
2280 // Format the placeholder string.
Douglas Gregor8987b232011-09-27 23:30:47 +00002281 std::string PlaceholderStr = FormatFunctionParameter(Context, Policy,
2282 Param);
Douglas Gregor83482d12010-08-24 16:15:59 +00002283
Douglas Gregore17794f2010-08-31 05:13:43 +00002284 if (Function->isVariadic() && P == N - 1)
2285 PlaceholderStr += ", ...";
2286
Douglas Gregor86d9a522009-09-21 16:56:56 +00002287 // Add the placeholder string.
Douglas Gregordae68752011-02-01 22:57:45 +00002288 Result.AddPlaceholderChunk(
2289 Result.getAllocator().CopyString(PlaceholderStr));
Douglas Gregor86d9a522009-09-21 16:56:56 +00002290 }
Douglas Gregorb3d45252009-09-22 21:42:17 +00002291
2292 if (const FunctionProtoType *Proto
2293 = Function->getType()->getAs<FunctionProtoType>())
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002294 if (Proto->isVariadic()) {
Stephen Hines651f13c2014-04-23 16:59:28 -07002295 if (Proto->getNumParams() == 0)
Douglas Gregor218937c2011-02-01 19:23:04 +00002296 Result.AddPlaceholderChunk("...");
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002297
Douglas Gregor218937c2011-02-01 19:23:04 +00002298 MaybeAddSentinel(Context, Function, Result);
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002299 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00002300}
2301
2302/// \brief Add template parameter chunks to the given code completion string.
2303static void AddTemplateParameterChunks(ASTContext &Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00002304 const PrintingPolicy &Policy,
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002305 const TemplateDecl *Template,
Douglas Gregor218937c2011-02-01 19:23:04 +00002306 CodeCompletionBuilder &Result,
2307 unsigned MaxParameters = 0,
2308 unsigned Start = 0,
2309 bool InDefaultArg = false) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00002310 bool FirstParameter = true;
Stephen Hines0e2c34f2015-03-23 12:09:02 -07002311
2312 // Prefer to take the template parameter names from the first declaration of
2313 // the template.
2314 Template = cast<TemplateDecl>(Template->getCanonicalDecl());
2315
Douglas Gregor86d9a522009-09-21 16:56:56 +00002316 TemplateParameterList *Params = Template->getTemplateParameters();
2317 TemplateParameterList::iterator PEnd = Params->end();
2318 if (MaxParameters)
2319 PEnd = Params->begin() + MaxParameters;
Douglas Gregor218937c2011-02-01 19:23:04 +00002320 for (TemplateParameterList::iterator P = Params->begin() + Start;
2321 P != PEnd; ++P) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00002322 bool HasDefaultArg = false;
2323 std::string PlaceholderStr;
2324 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(*P)) {
2325 if (TTP->wasDeclaredWithTypename())
2326 PlaceholderStr = "typename";
2327 else
2328 PlaceholderStr = "class";
2329
2330 if (TTP->getIdentifier()) {
2331 PlaceholderStr += ' ';
2332 PlaceholderStr += TTP->getIdentifier()->getName();
2333 }
2334
2335 HasDefaultArg = TTP->hasDefaultArgument();
2336 } else if (NonTypeTemplateParmDecl *NTTP
Douglas Gregor218937c2011-02-01 19:23:04 +00002337 = dyn_cast<NonTypeTemplateParmDecl>(*P)) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00002338 if (NTTP->getIdentifier())
2339 PlaceholderStr = NTTP->getIdentifier()->getName();
John McCallf85e1932011-06-15 23:02:42 +00002340 NTTP->getType().getAsStringInternal(PlaceholderStr, Policy);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002341 HasDefaultArg = NTTP->hasDefaultArgument();
2342 } else {
2343 assert(isa<TemplateTemplateParmDecl>(*P));
2344 TemplateTemplateParmDecl *TTP = cast<TemplateTemplateParmDecl>(*P);
2345
2346 // Since putting the template argument list into the placeholder would
2347 // be very, very long, we just use an abbreviation.
2348 PlaceholderStr = "template<...> class";
2349 if (TTP->getIdentifier()) {
2350 PlaceholderStr += ' ';
2351 PlaceholderStr += TTP->getIdentifier()->getName();
2352 }
2353
2354 HasDefaultArg = TTP->hasDefaultArgument();
2355 }
2356
Douglas Gregor218937c2011-02-01 19:23:04 +00002357 if (HasDefaultArg && !InDefaultArg) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00002358 // When we see an optional default argument, put that argument and
2359 // the remaining default arguments into a new, optional string.
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00002360 CodeCompletionBuilder Opt(Result.getAllocator(),
2361 Result.getCodeCompletionTUInfo());
Douglas Gregor218937c2011-02-01 19:23:04 +00002362 if (!FirstParameter)
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002363 Opt.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor8987b232011-09-27 23:30:47 +00002364 AddTemplateParameterChunks(Context, Policy, Template, Opt, MaxParameters,
Douglas Gregor218937c2011-02-01 19:23:04 +00002365 P - Params->begin(), true);
2366 Result.AddOptionalChunk(Opt.TakeString());
2367 break;
Douglas Gregor86d9a522009-09-21 16:56:56 +00002368 }
2369
Douglas Gregor218937c2011-02-01 19:23:04 +00002370 InDefaultArg = false;
2371
Douglas Gregor86d9a522009-09-21 16:56:56 +00002372 if (FirstParameter)
2373 FirstParameter = false;
2374 else
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002375 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002376
2377 // Add the placeholder string.
Douglas Gregordae68752011-02-01 22:57:45 +00002378 Result.AddPlaceholderChunk(
2379 Result.getAllocator().CopyString(PlaceholderStr));
Douglas Gregor86d9a522009-09-21 16:56:56 +00002380 }
2381}
2382
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00002383/// \brief Add a qualifier to the given code-completion string, if the
2384/// provided nested-name-specifier is non-NULL.
Douglas Gregora61a8792009-12-11 18:44:16 +00002385static void
Douglas Gregor218937c2011-02-01 19:23:04 +00002386AddQualifierToCompletionString(CodeCompletionBuilder &Result,
Douglas Gregora61a8792009-12-11 18:44:16 +00002387 NestedNameSpecifier *Qualifier,
2388 bool QualifierIsInformative,
Douglas Gregor8987b232011-09-27 23:30:47 +00002389 ASTContext &Context,
2390 const PrintingPolicy &Policy) {
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00002391 if (!Qualifier)
2392 return;
2393
2394 std::string PrintedNNS;
2395 {
2396 llvm::raw_string_ostream OS(PrintedNNS);
Douglas Gregor8987b232011-09-27 23:30:47 +00002397 Qualifier->print(OS, Policy);
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00002398 }
Douglas Gregor0563c262009-09-22 23:15:58 +00002399 if (QualifierIsInformative)
Douglas Gregordae68752011-02-01 22:57:45 +00002400 Result.AddInformativeChunk(Result.getAllocator().CopyString(PrintedNNS));
Douglas Gregor0563c262009-09-22 23:15:58 +00002401 else
Douglas Gregordae68752011-02-01 22:57:45 +00002402 Result.AddTextChunk(Result.getAllocator().CopyString(PrintedNNS));
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00002403}
2404
Douglas Gregor218937c2011-02-01 19:23:04 +00002405static void
2406AddFunctionTypeQualsToCompletionString(CodeCompletionBuilder &Result,
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002407 const FunctionDecl *Function) {
Douglas Gregora61a8792009-12-11 18:44:16 +00002408 const FunctionProtoType *Proto
2409 = Function->getType()->getAs<FunctionProtoType>();
2410 if (!Proto || !Proto->getTypeQuals())
2411 return;
2412
Douglas Gregora63f6de2011-02-01 21:15:40 +00002413 // FIXME: Add ref-qualifier!
2414
2415 // Handle single qualifiers without copying
2416 if (Proto->getTypeQuals() == Qualifiers::Const) {
2417 Result.AddInformativeChunk(" const");
2418 return;
2419 }
2420
2421 if (Proto->getTypeQuals() == Qualifiers::Volatile) {
2422 Result.AddInformativeChunk(" volatile");
2423 return;
2424 }
2425
2426 if (Proto->getTypeQuals() == Qualifiers::Restrict) {
2427 Result.AddInformativeChunk(" restrict");
2428 return;
2429 }
2430
2431 // Handle multiple qualifiers.
Douglas Gregora61a8792009-12-11 18:44:16 +00002432 std::string QualsStr;
David Blaikie4ef832f2012-08-10 00:55:35 +00002433 if (Proto->isConst())
Douglas Gregora61a8792009-12-11 18:44:16 +00002434 QualsStr += " const";
David Blaikie4ef832f2012-08-10 00:55:35 +00002435 if (Proto->isVolatile())
Douglas Gregora61a8792009-12-11 18:44:16 +00002436 QualsStr += " volatile";
David Blaikie4ef832f2012-08-10 00:55:35 +00002437 if (Proto->isRestrict())
Douglas Gregora61a8792009-12-11 18:44:16 +00002438 QualsStr += " restrict";
Douglas Gregordae68752011-02-01 22:57:45 +00002439 Result.AddInformativeChunk(Result.getAllocator().CopyString(QualsStr));
Douglas Gregora61a8792009-12-11 18:44:16 +00002440}
2441
Douglas Gregor6f942b22010-09-21 16:06:22 +00002442/// \brief Add the name of the given declaration
Douglas Gregor8987b232011-09-27 23:30:47 +00002443static void AddTypedNameChunk(ASTContext &Context, const PrintingPolicy &Policy,
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002444 const NamedDecl *ND,
2445 CodeCompletionBuilder &Result) {
Douglas Gregor6f942b22010-09-21 16:06:22 +00002446 DeclarationName Name = ND->getDeclName();
2447 if (!Name)
2448 return;
2449
2450 switch (Name.getNameKind()) {
Douglas Gregora63f6de2011-02-01 21:15:40 +00002451 case DeclarationName::CXXOperatorName: {
Stephen Hines6bcf27b2014-05-29 04:14:42 -07002452 const char *OperatorName = nullptr;
Douglas Gregora63f6de2011-02-01 21:15:40 +00002453 switch (Name.getCXXOverloadedOperator()) {
2454 case OO_None:
2455 case OO_Conditional:
2456 case NUM_OVERLOADED_OPERATORS:
2457 OperatorName = "operator";
2458 break;
2459
2460#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
2461 case OO_##Name: OperatorName = "operator" Spelling; break;
2462#define OVERLOADED_OPERATOR_MULTI(Name,Spelling,Unary,Binary,MemberOnly)
2463#include "clang/Basic/OperatorKinds.def"
2464
2465 case OO_New: OperatorName = "operator new"; break;
2466 case OO_Delete: OperatorName = "operator delete"; break;
2467 case OO_Array_New: OperatorName = "operator new[]"; break;
2468 case OO_Array_Delete: OperatorName = "operator delete[]"; break;
2469 case OO_Call: OperatorName = "operator()"; break;
2470 case OO_Subscript: OperatorName = "operator[]"; break;
2471 }
2472 Result.AddTypedTextChunk(OperatorName);
2473 break;
2474 }
2475
Douglas Gregor6f942b22010-09-21 16:06:22 +00002476 case DeclarationName::Identifier:
2477 case DeclarationName::CXXConversionFunctionName:
Douglas Gregor6f942b22010-09-21 16:06:22 +00002478 case DeclarationName::CXXDestructorName:
2479 case DeclarationName::CXXLiteralOperatorName:
Douglas Gregordae68752011-02-01 22:57:45 +00002480 Result.AddTypedTextChunk(
2481 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor6f942b22010-09-21 16:06:22 +00002482 break;
2483
2484 case DeclarationName::CXXUsingDirective:
2485 case DeclarationName::ObjCZeroArgSelector:
2486 case DeclarationName::ObjCOneArgSelector:
2487 case DeclarationName::ObjCMultiArgSelector:
2488 break;
2489
2490 case DeclarationName::CXXConstructorName: {
Stephen Hines6bcf27b2014-05-29 04:14:42 -07002491 CXXRecordDecl *Record = nullptr;
Douglas Gregor6f942b22010-09-21 16:06:22 +00002492 QualType Ty = Name.getCXXNameType();
2493 if (const RecordType *RecordTy = Ty->getAs<RecordType>())
2494 Record = cast<CXXRecordDecl>(RecordTy->getDecl());
2495 else if (const InjectedClassNameType *InjectedTy
2496 = Ty->getAs<InjectedClassNameType>())
2497 Record = InjectedTy->getDecl();
2498 else {
Douglas Gregordae68752011-02-01 22:57:45 +00002499 Result.AddTypedTextChunk(
2500 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor6f942b22010-09-21 16:06:22 +00002501 break;
2502 }
2503
Douglas Gregordae68752011-02-01 22:57:45 +00002504 Result.AddTypedTextChunk(
2505 Result.getAllocator().CopyString(Record->getNameAsString()));
Douglas Gregor6f942b22010-09-21 16:06:22 +00002506 if (ClassTemplateDecl *Template = Record->getDescribedClassTemplate()) {
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002507 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Douglas Gregor8987b232011-09-27 23:30:47 +00002508 AddTemplateParameterChunks(Context, Policy, Template, Result);
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002509 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor6f942b22010-09-21 16:06:22 +00002510 }
2511 break;
2512 }
2513 }
2514}
2515
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002516CodeCompletionString *CodeCompletionResult::CreateCodeCompletionString(Sema &S,
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00002517 CodeCompletionAllocator &Allocator,
Dmitri Gribenkod99ef532012-07-02 17:35:10 +00002518 CodeCompletionTUInfo &CCTUInfo,
2519 bool IncludeBriefComments) {
2520 return CreateCodeCompletionString(S.Context, S.PP, Allocator, CCTUInfo,
2521 IncludeBriefComments);
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002522}
2523
Douglas Gregor86d9a522009-09-21 16:56:56 +00002524/// \brief If possible, create a new code completion string for the given
2525/// result.
2526///
2527/// \returns Either a new, heap-allocated code completion string describing
2528/// how to use this result, or NULL to indicate that the string or name of the
2529/// result is all that is needed.
2530CodeCompletionString *
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002531CodeCompletionResult::CreateCodeCompletionString(ASTContext &Ctx,
2532 Preprocessor &PP,
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00002533 CodeCompletionAllocator &Allocator,
Dmitri Gribenkod99ef532012-07-02 17:35:10 +00002534 CodeCompletionTUInfo &CCTUInfo,
2535 bool IncludeBriefComments) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00002536 CodeCompletionBuilder Result(Allocator, CCTUInfo, Priority, Availability);
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002537
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002538 PrintingPolicy Policy = getCompletionPrintingPolicy(Ctx, PP);
Douglas Gregor218937c2011-02-01 19:23:04 +00002539 if (Kind == RK_Pattern) {
2540 Pattern->Priority = Priority;
2541 Pattern->Availability = Availability;
Douglas Gregorba103062012-03-27 23:34:16 +00002542
2543 if (Declaration) {
2544 Result.addParentContext(Declaration->getDeclContext());
Douglas Gregorba103062012-03-27 23:34:16 +00002545 Pattern->ParentName = Result.getParentName();
Fariborz Jahanianc02ddb22013-03-22 17:55:27 +00002546 // Provide code completion comment for self.GetterName where
2547 // GetterName is the getter method for a property with name
2548 // different from the property name (declared via a property
2549 // getter attribute.
2550 const NamedDecl *ND = Declaration;
2551 if (const ObjCMethodDecl *M = dyn_cast<ObjCMethodDecl>(ND))
2552 if (M->isPropertyAccessor())
2553 if (const ObjCPropertyDecl *PDecl = M->findPropertyDecl())
2554 if (PDecl->getGetterName() == M->getSelector() &&
Fariborz Jahanian16861372013-03-23 01:10:45 +00002555 PDecl->getIdentifier() != M->getIdentifier()) {
2556 if (const RawComment *RC =
2557 Ctx.getRawCommentForAnyRedecl(M)) {
Fariborz Jahanianc02ddb22013-03-22 17:55:27 +00002558 Result.addBriefComment(RC->getBriefText(Ctx));
2559 Pattern->BriefComment = Result.getBriefComment();
2560 }
Fariborz Jahanian16861372013-03-23 01:10:45 +00002561 else if (const RawComment *RC =
2562 Ctx.getRawCommentForAnyRedecl(PDecl)) {
2563 Result.addBriefComment(RC->getBriefText(Ctx));
2564 Pattern->BriefComment = Result.getBriefComment();
2565 }
2566 }
Douglas Gregorba103062012-03-27 23:34:16 +00002567 }
2568
Douglas Gregor218937c2011-02-01 19:23:04 +00002569 return Pattern;
2570 }
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002571
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002572 if (Kind == RK_Keyword) {
Douglas Gregor218937c2011-02-01 19:23:04 +00002573 Result.AddTypedTextChunk(Keyword);
2574 return Result.TakeString();
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002575 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00002576
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002577 if (Kind == RK_Macro) {
Argyrios Kyrtzidis9818a1d2013-02-20 00:54:57 +00002578 const MacroDirective *MD = PP.getMacroDirectiveHistory(Macro);
2579 assert(MD && "Not a macro?");
Argyrios Kyrtzidisc56fff72013-03-26 17:17:01 +00002580 const MacroInfo *MI = MD->getMacroInfo();
Stephen Hines176edba2014-12-01 14:53:08 -08002581 assert((!MD->isDefined() || MI) && "missing MacroInfo for define");
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002582
Douglas Gregordae68752011-02-01 22:57:45 +00002583 Result.AddTypedTextChunk(
2584 Result.getAllocator().CopyString(Macro->getName()));
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002585
Stephen Hines176edba2014-12-01 14:53:08 -08002586 if (!MI || !MI->isFunctionLike())
Douglas Gregor218937c2011-02-01 19:23:04 +00002587 return Result.TakeString();
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002588
2589 // Format a function-like macro with placeholders for the arguments.
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002590 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregore4244702011-07-30 08:17:44 +00002591 MacroInfo::arg_iterator A = MI->arg_begin(), AEnd = MI->arg_end();
Douglas Gregorc8dc1352012-01-21 00:43:38 +00002592
2593 // C99 variadic macros add __VA_ARGS__ at the end. Skip it.
2594 if (MI->isC99Varargs()) {
2595 --AEnd;
2596
2597 if (A == AEnd) {
2598 Result.AddPlaceholderChunk("...");
2599 }
Douglas Gregore4244702011-07-30 08:17:44 +00002600 }
Douglas Gregorc8dc1352012-01-21 00:43:38 +00002601
Douglas Gregore4244702011-07-30 08:17:44 +00002602 for (MacroInfo::arg_iterator A = MI->arg_begin(); A != AEnd; ++A) {
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002603 if (A != MI->arg_begin())
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002604 Result.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorc8dc1352012-01-21 00:43:38 +00002605
2606 if (MI->isVariadic() && (A+1) == AEnd) {
Dylan Noblesmithf7ccbad2012-02-05 02:13:05 +00002607 SmallString<32> Arg = (*A)->getName();
Douglas Gregorc8dc1352012-01-21 00:43:38 +00002608 if (MI->isC99Varargs())
2609 Arg += ", ...";
2610 else
2611 Arg += "...";
Douglas Gregordae68752011-02-01 22:57:45 +00002612 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregorc8dc1352012-01-21 00:43:38 +00002613 break;
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002614 }
Douglas Gregorc8dc1352012-01-21 00:43:38 +00002615
2616 // Non-variadic macros are simple.
2617 Result.AddPlaceholderChunk(
2618 Result.getAllocator().CopyString((*A)->getName()));
Douglas Gregore4244702011-07-30 08:17:44 +00002619 }
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002620 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor218937c2011-02-01 19:23:04 +00002621 return Result.TakeString();
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00002622 }
2623
Douglas Gregord8e8a582010-05-25 21:41:55 +00002624 assert(Kind == RK_Declaration && "Missed a result kind?");
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002625 const NamedDecl *ND = Declaration;
Douglas Gregorba103062012-03-27 23:34:16 +00002626 Result.addParentContext(ND->getDeclContext());
Dmitri Gribenkod99ef532012-07-02 17:35:10 +00002627
2628 if (IncludeBriefComments) {
2629 // Add documentation comment, if it exists.
Dmitri Gribenkof50555e2012-08-11 00:51:43 +00002630 if (const RawComment *RC = Ctx.getRawCommentForAnyRedecl(ND)) {
Dmitri Gribenkod99ef532012-07-02 17:35:10 +00002631 Result.addBriefComment(RC->getBriefText(Ctx));
Fariborz Jahanianb98f7af2013-02-28 17:47:14 +00002632 }
2633 else if (const ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
2634 if (OMD->isPropertyAccessor())
2635 if (const ObjCPropertyDecl *PDecl = OMD->findPropertyDecl())
2636 if (const RawComment *RC = Ctx.getRawCommentForAnyRedecl(PDecl))
2637 Result.addBriefComment(RC->getBriefText(Ctx));
Dmitri Gribenkod99ef532012-07-02 17:35:10 +00002638 }
2639
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002640 if (StartsNestedNameSpecifier) {
Douglas Gregordae68752011-02-01 22:57:45 +00002641 Result.AddTypedTextChunk(
2642 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor218937c2011-02-01 19:23:04 +00002643 Result.AddTextChunk("::");
2644 return Result.TakeString();
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002645 }
Erik Verbruggen6164ea12011-10-14 15:31:08 +00002646
Stephen Hines651f13c2014-04-23 16:59:28 -07002647 for (const auto *I : ND->specific_attrs<AnnotateAttr>())
2648 Result.AddAnnotation(Result.getAllocator().CopyString(I->getAnnotation()));
2649
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002650 AddResultTypeChunk(Ctx, Policy, ND, Result);
Douglas Gregorff5ce6e2009-12-18 18:53:37 +00002651
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002652 if (const FunctionDecl *Function = dyn_cast<FunctionDecl>(ND)) {
Douglas Gregor0563c262009-09-22 23:15:58 +00002653 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002654 Ctx, Policy);
2655 AddTypedNameChunk(Ctx, Policy, ND, Result);
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002656 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002657 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002658 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregora61a8792009-12-11 18:44:16 +00002659 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregor218937c2011-02-01 19:23:04 +00002660 return Result.TakeString();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002661 }
2662
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002663 if (const FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(ND)) {
Douglas Gregor0563c262009-09-22 23:15:58 +00002664 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002665 Ctx, Policy);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002666 FunctionDecl *Function = FunTmpl->getTemplatedDecl();
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002667 AddTypedNameChunk(Ctx, Policy, Function, Result);
Douglas Gregor6f942b22010-09-21 16:06:22 +00002668
Douglas Gregor86d9a522009-09-21 16:56:56 +00002669 // Figure out which template parameters are deduced (or have default
2670 // arguments).
Benjamin Kramer013b3662012-01-30 16:17:39 +00002671 llvm::SmallBitVector Deduced;
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002672 Sema::MarkDeducedTemplateParameters(Ctx, FunTmpl, Deduced);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002673 unsigned LastDeducibleArgument;
2674 for (LastDeducibleArgument = Deduced.size(); LastDeducibleArgument > 0;
2675 --LastDeducibleArgument) {
2676 if (!Deduced[LastDeducibleArgument - 1]) {
2677 // C++0x: Figure out if the template argument has a default. If so,
2678 // the user doesn't need to type this argument.
2679 // FIXME: We need to abstract template parameters better!
2680 bool HasDefaultArg = false;
2681 NamedDecl *Param = FunTmpl->getTemplateParameters()->getParam(
Douglas Gregor218937c2011-02-01 19:23:04 +00002682 LastDeducibleArgument - 1);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002683 if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
2684 HasDefaultArg = TTP->hasDefaultArgument();
2685 else if (NonTypeTemplateParmDecl *NTTP
2686 = dyn_cast<NonTypeTemplateParmDecl>(Param))
2687 HasDefaultArg = NTTP->hasDefaultArgument();
2688 else {
2689 assert(isa<TemplateTemplateParmDecl>(Param));
2690 HasDefaultArg
Douglas Gregor0c8296d2009-11-07 00:00:49 +00002691 = cast<TemplateTemplateParmDecl>(Param)->hasDefaultArgument();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002692 }
2693
2694 if (!HasDefaultArg)
2695 break;
2696 }
2697 }
2698
2699 if (LastDeducibleArgument) {
2700 // Some of the function template arguments cannot be deduced from a
2701 // function call, so we introduce an explicit template argument list
2702 // containing all of the arguments up to the first deducible argument.
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002703 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002704 AddTemplateParameterChunks(Ctx, Policy, FunTmpl, Result,
Douglas Gregor86d9a522009-09-21 16:56:56 +00002705 LastDeducibleArgument);
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002706 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor86d9a522009-09-21 16:56:56 +00002707 }
2708
2709 // Add the function parameters
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002710 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002711 AddFunctionParameterChunks(Ctx, Policy, Function, Result);
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002712 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregora61a8792009-12-11 18:44:16 +00002713 AddFunctionTypeQualsToCompletionString(Result, Function);
Douglas Gregor218937c2011-02-01 19:23:04 +00002714 return Result.TakeString();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002715 }
2716
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002717 if (const TemplateDecl *Template = dyn_cast<TemplateDecl>(ND)) {
Douglas Gregor0563c262009-09-22 23:15:58 +00002718 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002719 Ctx, Policy);
Douglas Gregordae68752011-02-01 22:57:45 +00002720 Result.AddTypedTextChunk(
2721 Result.getAllocator().CopyString(Template->getNameAsString()));
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002722 Result.AddChunk(CodeCompletionString::CK_LeftAngle);
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002723 AddTemplateParameterChunks(Ctx, Policy, Template, Result);
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002724 Result.AddChunk(CodeCompletionString::CK_RightAngle);
Douglas Gregor218937c2011-02-01 19:23:04 +00002725 return Result.TakeString();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002726 }
2727
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002728 if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(ND)) {
Douglas Gregor9630eb62009-11-17 16:44:22 +00002729 Selector Sel = Method->getSelector();
2730 if (Sel.isUnarySelector()) {
Douglas Gregordae68752011-02-01 22:57:45 +00002731 Result.AddTypedTextChunk(Result.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00002732 Sel.getNameForSlot(0)));
Douglas Gregor218937c2011-02-01 19:23:04 +00002733 return Result.TakeString();
Douglas Gregor9630eb62009-11-17 16:44:22 +00002734 }
2735
Douglas Gregor813d8342011-02-18 22:29:55 +00002736 std::string SelName = Sel.getNameForSlot(0).str();
Douglas Gregord3c68542009-11-19 01:08:35 +00002737 SelName += ':';
2738 if (StartParameter == 0)
Douglas Gregordae68752011-02-01 22:57:45 +00002739 Result.AddTypedTextChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregord3c68542009-11-19 01:08:35 +00002740 else {
Douglas Gregordae68752011-02-01 22:57:45 +00002741 Result.AddInformativeChunk(Result.getAllocator().CopyString(SelName));
Douglas Gregord3c68542009-11-19 01:08:35 +00002742
2743 // If there is only one parameter, and we're past it, add an empty
2744 // typed-text chunk since there is nothing to type.
2745 if (Method->param_size() == 1)
Douglas Gregor218937c2011-02-01 19:23:04 +00002746 Result.AddTypedTextChunk("");
Douglas Gregord3c68542009-11-19 01:08:35 +00002747 }
Douglas Gregor9630eb62009-11-17 16:44:22 +00002748 unsigned Idx = 0;
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00002749 for (ObjCMethodDecl::param_const_iterator P = Method->param_begin(),
2750 PEnd = Method->param_end();
Douglas Gregor9630eb62009-11-17 16:44:22 +00002751 P != PEnd; (void)++P, ++Idx) {
2752 if (Idx > 0) {
Douglas Gregord3c68542009-11-19 01:08:35 +00002753 std::string Keyword;
2754 if (Idx > StartParameter)
Douglas Gregor218937c2011-02-01 19:23:04 +00002755 Result.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor9630eb62009-11-17 16:44:22 +00002756 if (IdentifierInfo *II = Sel.getIdentifierInfoForSlot(Idx))
Benjamin Kramera0651c52011-07-26 16:59:25 +00002757 Keyword += II->getName();
Douglas Gregor9630eb62009-11-17 16:44:22 +00002758 Keyword += ":";
Douglas Gregor1f5537a2010-07-08 23:20:03 +00002759 if (Idx < StartParameter || AllParametersAreInformative)
Douglas Gregordae68752011-02-01 22:57:45 +00002760 Result.AddInformativeChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00002761 else
Douglas Gregordae68752011-02-01 22:57:45 +00002762 Result.AddTypedTextChunk(Result.getAllocator().CopyString(Keyword));
Douglas Gregor9630eb62009-11-17 16:44:22 +00002763 }
Douglas Gregord3c68542009-11-19 01:08:35 +00002764
2765 // If we're before the starting parameter, skip the placeholder.
2766 if (Idx < StartParameter)
2767 continue;
Douglas Gregor9630eb62009-11-17 16:44:22 +00002768
2769 std::string Arg;
Douglas Gregor83482d12010-08-24 16:15:59 +00002770
2771 if ((*P)->getType()->isBlockPointerType() && !DeclaringEntity)
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002772 Arg = FormatFunctionParameter(Ctx, Policy, *P, true);
Douglas Gregor83482d12010-08-24 16:15:59 +00002773 else {
John McCallf85e1932011-06-15 23:02:42 +00002774 (*P)->getType().getAsStringInternal(Arg, Policy);
Douglas Gregor6fa14dd2011-07-30 07:55:26 +00002775 Arg = "(" + formatObjCParamQualifiers((*P)->getObjCDeclQualifier())
2776 + Arg + ")";
Douglas Gregor83482d12010-08-24 16:15:59 +00002777 if (IdentifierInfo *II = (*P)->getIdentifier())
Douglas Gregoraba48082010-08-29 19:47:46 +00002778 if (DeclaringEntity || AllParametersAreInformative)
Benjamin Kramera0651c52011-07-26 16:59:25 +00002779 Arg += II->getName();
Douglas Gregor83482d12010-08-24 16:15:59 +00002780 }
2781
Douglas Gregore17794f2010-08-31 05:13:43 +00002782 if (Method->isVariadic() && (P + 1) == PEnd)
2783 Arg += ", ...";
2784
Douglas Gregor1f5537a2010-07-08 23:20:03 +00002785 if (DeclaringEntity)
Douglas Gregordae68752011-02-01 22:57:45 +00002786 Result.AddTextChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor1f5537a2010-07-08 23:20:03 +00002787 else if (AllParametersAreInformative)
Douglas Gregordae68752011-02-01 22:57:45 +00002788 Result.AddInformativeChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor4ad96852009-11-19 07:41:15 +00002789 else
Douglas Gregordae68752011-02-01 22:57:45 +00002790 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Arg));
Douglas Gregor9630eb62009-11-17 16:44:22 +00002791 }
2792
Douglas Gregor2a17af02009-12-23 00:21:46 +00002793 if (Method->isVariadic()) {
Douglas Gregore17794f2010-08-31 05:13:43 +00002794 if (Method->param_size() == 0) {
2795 if (DeclaringEntity)
Douglas Gregor218937c2011-02-01 19:23:04 +00002796 Result.AddTextChunk(", ...");
Douglas Gregore17794f2010-08-31 05:13:43 +00002797 else if (AllParametersAreInformative)
Douglas Gregor218937c2011-02-01 19:23:04 +00002798 Result.AddInformativeChunk(", ...");
Douglas Gregore17794f2010-08-31 05:13:43 +00002799 else
Douglas Gregor218937c2011-02-01 19:23:04 +00002800 Result.AddPlaceholderChunk(", ...");
Douglas Gregore17794f2010-08-31 05:13:43 +00002801 }
Douglas Gregoraaa107a2010-08-23 23:51:41 +00002802
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002803 MaybeAddSentinel(Ctx, Method, Result);
Douglas Gregor2a17af02009-12-23 00:21:46 +00002804 }
2805
Douglas Gregor218937c2011-02-01 19:23:04 +00002806 return Result.TakeString();
Douglas Gregor9630eb62009-11-17 16:44:22 +00002807 }
2808
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002809 if (Qualifier)
Douglas Gregor0563c262009-09-22 23:15:58 +00002810 AddQualifierToCompletionString(Result, Qualifier, QualifierIsInformative,
Argyrios Kyrtzidisea8c59a2012-01-17 02:15:51 +00002811 Ctx, Policy);
Douglas Gregor2b4074f2009-12-01 05:55:20 +00002812
Douglas Gregordae68752011-02-01 22:57:45 +00002813 Result.AddTypedTextChunk(
2814 Result.getAllocator().CopyString(ND->getNameAsString()));
Douglas Gregor218937c2011-02-01 19:23:04 +00002815 return Result.TakeString();
Douglas Gregor86d9a522009-09-21 16:56:56 +00002816}
2817
Stephen Hines0e2c34f2015-03-23 12:09:02 -07002818/// \brief Add function overload parameter chunks to the given code completion
2819/// string.
2820static void AddOverloadParameterChunks(ASTContext &Context,
2821 const PrintingPolicy &Policy,
2822 const FunctionDecl *Function,
2823 const FunctionProtoType *Prototype,
2824 CodeCompletionBuilder &Result,
2825 unsigned CurrentArg,
2826 unsigned Start = 0,
2827 bool InOptional = false) {
2828 bool FirstParameter = true;
2829 unsigned NumParams = Function ? Function->getNumParams()
2830 : Prototype->getNumParams();
2831
2832 for (unsigned P = Start; P != NumParams; ++P) {
2833 if (Function && Function->getParamDecl(P)->hasDefaultArg() && !InOptional) {
2834 // When we see an optional default argument, put that argument and
2835 // the remaining default arguments into a new, optional string.
2836 CodeCompletionBuilder Opt(Result.getAllocator(),
2837 Result.getCodeCompletionTUInfo());
2838 if (!FirstParameter)
2839 Opt.AddChunk(CodeCompletionString::CK_Comma);
2840 // Optional sections are nested.
2841 AddOverloadParameterChunks(Context, Policy, Function, Prototype, Opt,
2842 CurrentArg, P, /*InOptional=*/true);
2843 Result.AddOptionalChunk(Opt.TakeString());
2844 return;
2845 }
2846
2847 if (FirstParameter)
2848 FirstParameter = false;
2849 else
2850 Result.AddChunk(CodeCompletionString::CK_Comma);
2851
2852 InOptional = false;
2853
2854 // Format the placeholder string.
2855 std::string Placeholder;
2856 if (Function)
2857 Placeholder = FormatFunctionParameter(Context, Policy,
2858 Function->getParamDecl(P));
2859 else
2860 Placeholder = Prototype->getParamType(P).getAsString(Policy);
2861
2862 if (P == CurrentArg)
2863 Result.AddCurrentParameterChunk(
2864 Result.getAllocator().CopyString(Placeholder));
2865 else
2866 Result.AddPlaceholderChunk(Result.getAllocator().CopyString(Placeholder));
2867 }
2868
2869 if (Prototype && Prototype->isVariadic()) {
2870 CodeCompletionBuilder Opt(Result.getAllocator(),
2871 Result.getCodeCompletionTUInfo());
2872 if (!FirstParameter)
2873 Opt.AddChunk(CodeCompletionString::CK_Comma);
2874
2875 if (CurrentArg < NumParams)
2876 Opt.AddPlaceholderChunk("...");
2877 else
2878 Opt.AddCurrentParameterChunk("...");
2879
2880 Result.AddOptionalChunk(Opt.TakeString());
2881 }
2882}
2883
Douglas Gregor86d802e2009-09-23 00:34:09 +00002884CodeCompletionString *
2885CodeCompleteConsumer::OverloadCandidate::CreateSignatureString(
Stephen Hines0e2c34f2015-03-23 12:09:02 -07002886 unsigned CurrentArg, Sema &S,
2887 CodeCompletionAllocator &Allocator,
2888 CodeCompletionTUInfo &CCTUInfo,
2889 bool IncludeBriefComments) const {
Douglas Gregor8987b232011-09-27 23:30:47 +00002890 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
John McCallf85e1932011-06-15 23:02:42 +00002891
Douglas Gregor218937c2011-02-01 19:23:04 +00002892 // FIXME: Set priority, availability appropriately.
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00002893 CodeCompletionBuilder Result(Allocator,CCTUInfo, 1, CXAvailability_Available);
Douglas Gregor86d802e2009-09-23 00:34:09 +00002894 FunctionDecl *FDecl = getFunction();
Stephen Hines0e2c34f2015-03-23 12:09:02 -07002895 const FunctionProtoType *Proto
Douglas Gregor86d802e2009-09-23 00:34:09 +00002896 = dyn_cast<FunctionProtoType>(getFunctionType());
2897 if (!FDecl && !Proto) {
2898 // Function without a prototype. Just give the return type and a
2899 // highlighted ellipsis.
2900 const FunctionType *FT = getFunctionType();
Stephen Hines0e2c34f2015-03-23 12:09:02 -07002901 Result.AddResultTypeChunk(Result.getAllocator().CopyString(
2902 FT->getReturnType().getAsString(Policy)));
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002903 Result.AddChunk(CodeCompletionString::CK_LeftParen);
2904 Result.AddChunk(CodeCompletionString::CK_CurrentParameter, "...");
2905 Result.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregor218937c2011-02-01 19:23:04 +00002906 return Result.TakeString();
Douglas Gregor86d802e2009-09-23 00:34:09 +00002907 }
Stephen Hines0e2c34f2015-03-23 12:09:02 -07002908
2909 if (FDecl) {
2910 if (IncludeBriefComments && CurrentArg < FDecl->getNumParams())
2911 if (auto RC = S.getASTContext().getRawCommentForAnyRedecl(
2912 FDecl->getParamDecl(CurrentArg)))
2913 Result.addBriefComment(RC->getBriefText(S.getASTContext()));
2914 AddResultTypeChunk(S.Context, Policy, FDecl, Result);
Douglas Gregordae68752011-02-01 22:57:45 +00002915 Result.AddTextChunk(
Stephen Hines0e2c34f2015-03-23 12:09:02 -07002916 Result.getAllocator().CopyString(FDecl->getNameAsString()));
2917 } else {
2918 Result.AddResultTypeChunk(
2919 Result.getAllocator().CopyString(
Stephen Hines651f13c2014-04-23 16:59:28 -07002920 Proto->getReturnType().getAsString(Policy)));
Stephen Hines0e2c34f2015-03-23 12:09:02 -07002921 }
Stephen Hines651f13c2014-04-23 16:59:28 -07002922
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002923 Result.AddChunk(CodeCompletionString::CK_LeftParen);
Stephen Hines0e2c34f2015-03-23 12:09:02 -07002924 AddOverloadParameterChunks(S.getASTContext(), Policy, FDecl, Proto, Result,
2925 CurrentArg);
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00002926 Result.AddChunk(CodeCompletionString::CK_RightParen);
Stephen Hines0e2c34f2015-03-23 12:09:02 -07002927
Douglas Gregor218937c2011-02-01 19:23:04 +00002928 return Result.TakeString();
Douglas Gregor86d802e2009-09-23 00:34:09 +00002929}
2930
Chris Lattner5f9e2722011-07-23 10:55:15 +00002931unsigned clang::getMacroUsagePriority(StringRef MacroName,
Douglas Gregorb05496d2010-09-20 21:11:48 +00002932 const LangOptions &LangOpts,
Douglas Gregor1827e102010-08-16 16:18:59 +00002933 bool PreferredTypeIsPointer) {
2934 unsigned Priority = CCP_Macro;
2935
Douglas Gregorb05496d2010-09-20 21:11:48 +00002936 // Treat the "nil", "Nil" and "NULL" macros as null pointer constants.
2937 if (MacroName.equals("nil") || MacroName.equals("NULL") ||
2938 MacroName.equals("Nil")) {
Douglas Gregor1827e102010-08-16 16:18:59 +00002939 Priority = CCP_Constant;
2940 if (PreferredTypeIsPointer)
2941 Priority = Priority / CCF_SimilarTypeMatch;
Douglas Gregorb05496d2010-09-20 21:11:48 +00002942 }
2943 // Treat "YES", "NO", "true", and "false" as constants.
2944 else if (MacroName.equals("YES") || MacroName.equals("NO") ||
2945 MacroName.equals("true") || MacroName.equals("false"))
2946 Priority = CCP_Constant;
2947 // Treat "bool" as a type.
2948 else if (MacroName.equals("bool"))
2949 Priority = CCP_Type + (LangOpts.ObjC1? CCD_bool_in_ObjC : 0);
2950
Douglas Gregor1827e102010-08-16 16:18:59 +00002951
2952 return Priority;
2953}
2954
Dmitri Gribenko06d8c602013-01-11 20:32:41 +00002955CXCursorKind clang::getCursorKindForDecl(const Decl *D) {
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002956 if (!D)
2957 return CXCursor_UnexposedDecl;
2958
2959 switch (D->getKind()) {
2960 case Decl::Enum: return CXCursor_EnumDecl;
2961 case Decl::EnumConstant: return CXCursor_EnumConstantDecl;
2962 case Decl::Field: return CXCursor_FieldDecl;
2963 case Decl::Function:
2964 return CXCursor_FunctionDecl;
2965 case Decl::ObjCCategory: return CXCursor_ObjCCategoryDecl;
2966 case Decl::ObjCCategoryImpl: return CXCursor_ObjCCategoryImplDecl;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002967 case Decl::ObjCImplementation: return CXCursor_ObjCImplementationDecl;
Douglas Gregor375bb142011-12-27 22:43:10 +00002968
Argyrios Kyrtzidisc15707d2012-01-24 21:39:26 +00002969 case Decl::ObjCInterface: return CXCursor_ObjCInterfaceDecl;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002970 case Decl::ObjCIvar: return CXCursor_ObjCIvarDecl;
2971 case Decl::ObjCMethod:
2972 return cast<ObjCMethodDecl>(D)->isInstanceMethod()
2973 ? CXCursor_ObjCInstanceMethodDecl : CXCursor_ObjCClassMethodDecl;
2974 case Decl::CXXMethod: return CXCursor_CXXMethod;
2975 case Decl::CXXConstructor: return CXCursor_Constructor;
2976 case Decl::CXXDestructor: return CXCursor_Destructor;
2977 case Decl::CXXConversion: return CXCursor_ConversionFunction;
2978 case Decl::ObjCProperty: return CXCursor_ObjCPropertyDecl;
Argyrios Kyrtzidisc15707d2012-01-24 21:39:26 +00002979 case Decl::ObjCProtocol: return CXCursor_ObjCProtocolDecl;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002980 case Decl::ParmVar: return CXCursor_ParmDecl;
2981 case Decl::Typedef: return CXCursor_TypedefDecl;
Richard Smith162e1c12011-04-15 14:24:37 +00002982 case Decl::TypeAlias: return CXCursor_TypeAliasDecl;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002983 case Decl::Var: return CXCursor_VarDecl;
2984 case Decl::Namespace: return CXCursor_Namespace;
2985 case Decl::NamespaceAlias: return CXCursor_NamespaceAlias;
2986 case Decl::TemplateTypeParm: return CXCursor_TemplateTypeParameter;
2987 case Decl::NonTypeTemplateParm:return CXCursor_NonTypeTemplateParameter;
2988 case Decl::TemplateTemplateParm:return CXCursor_TemplateTemplateParameter;
2989 case Decl::FunctionTemplate: return CXCursor_FunctionTemplate;
2990 case Decl::ClassTemplate: return CXCursor_ClassTemplate;
Argyrios Kyrtzidis2dfdb942011-09-30 17:58:23 +00002991 case Decl::AccessSpec: return CXCursor_CXXAccessSpecifier;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002992 case Decl::ClassTemplatePartialSpecialization:
2993 return CXCursor_ClassTemplatePartialSpecialization;
2994 case Decl::UsingDirective: return CXCursor_UsingDirective;
Douglas Gregor8e5900c2012-04-30 23:41:16 +00002995 case Decl::TranslationUnit: return CXCursor_TranslationUnit;
Douglas Gregore8d7beb2010-09-03 23:30:36 +00002996
2997 case Decl::Using:
2998 case Decl::UnresolvedUsingValue:
2999 case Decl::UnresolvedUsingTypename:
3000 return CXCursor_UsingDeclaration;
3001
Douglas Gregor352697a2011-06-03 23:08:58 +00003002 case Decl::ObjCPropertyImpl:
3003 switch (cast<ObjCPropertyImplDecl>(D)->getPropertyImplementation()) {
3004 case ObjCPropertyImplDecl::Dynamic:
3005 return CXCursor_ObjCDynamicDecl;
3006
3007 case ObjCPropertyImplDecl::Synthesize:
3008 return CXCursor_ObjCSynthesizeDecl;
3009 }
Argyrios Kyrtzidis6a010122012-10-05 00:22:24 +00003010
3011 case Decl::Import:
3012 return CXCursor_ModuleImportDecl;
Douglas Gregor352697a2011-06-03 23:08:58 +00003013
Douglas Gregore8d7beb2010-09-03 23:30:36 +00003014 default:
Dmitri Gribenko06d8c602013-01-11 20:32:41 +00003015 if (const TagDecl *TD = dyn_cast<TagDecl>(D)) {
Douglas Gregore8d7beb2010-09-03 23:30:36 +00003016 switch (TD->getTagKind()) {
Joao Matos6666ed42012-08-31 18:45:21 +00003017 case TTK_Interface: // fall through
Douglas Gregore8d7beb2010-09-03 23:30:36 +00003018 case TTK_Struct: return CXCursor_StructDecl;
3019 case TTK_Class: return CXCursor_ClassDecl;
3020 case TTK_Union: return CXCursor_UnionDecl;
3021 case TTK_Enum: return CXCursor_EnumDecl;
3022 }
3023 }
3024 }
3025
3026 return CXCursor_UnexposedDecl;
3027}
3028
Douglas Gregor590c7d52010-07-08 20:55:51 +00003029static void AddMacroResults(Preprocessor &PP, ResultBuilder &Results,
Douglas Gregor3644d972012-10-09 16:01:50 +00003030 bool IncludeUndefined,
Douglas Gregor590c7d52010-07-08 20:55:51 +00003031 bool TargetTypeIsPointer = false) {
John McCall0a2c5e22010-08-25 06:19:51 +00003032 typedef CodeCompletionResult Result;
Douglas Gregor590c7d52010-07-08 20:55:51 +00003033
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00003034 Results.EnterNewScope();
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00003035
Douglas Gregor0c8296d2009-11-07 00:00:49 +00003036 for (Preprocessor::macro_iterator M = PP.macro_begin(),
3037 MEnd = PP.macro_end();
Douglas Gregor590c7d52010-07-08 20:55:51 +00003038 M != MEnd; ++M) {
Stephen Hines6bcf27b2014-05-29 04:14:42 -07003039 if (IncludeUndefined || M->first->hasMacroDefinition()) {
3040 if (MacroInfo *MI = M->second->getMacroInfo())
3041 if (MI->isUsedForHeaderGuard())
3042 continue;
3043
Douglas Gregor3644d972012-10-09 16:01:50 +00003044 Results.AddResult(Result(M->first,
Douglas Gregor1827e102010-08-16 16:18:59 +00003045 getMacroUsagePriority(M->first->getName(),
David Blaikie4e4d0842012-03-11 07:00:24 +00003046 PP.getLangOpts(),
Douglas Gregor1827e102010-08-16 16:18:59 +00003047 TargetTypeIsPointer)));
Stephen Hines6bcf27b2014-05-29 04:14:42 -07003048 }
Douglas Gregor590c7d52010-07-08 20:55:51 +00003049 }
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00003050
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00003051 Results.ExitScope();
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00003052
Douglas Gregor3f7c7f42009-10-30 16:50:04 +00003053}
3054
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003055static void AddPrettyFunctionResults(const LangOptions &LangOpts,
3056 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00003057 typedef CodeCompletionResult Result;
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003058
3059 Results.EnterNewScope();
Douglas Gregorc7b7b7a2010-10-18 21:05:04 +00003060
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003061 Results.AddResult(Result("__PRETTY_FUNCTION__", CCP_Constant));
3062 Results.AddResult(Result("__FUNCTION__", CCP_Constant));
Richard Smith80ad52f2013-01-02 11:42:31 +00003063 if (LangOpts.C99 || LangOpts.CPlusPlus11)
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003064 Results.AddResult(Result("__func__", CCP_Constant));
3065 Results.ExitScope();
3066}
3067
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003068static void HandleCodeCompleteResults(Sema *S,
3069 CodeCompleteConsumer *CodeCompleter,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003070 CodeCompletionContext Context,
John McCall0a2c5e22010-08-25 06:19:51 +00003071 CodeCompletionResult *Results,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003072 unsigned NumResults) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00003073 if (CodeCompleter)
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003074 CodeCompleter->ProcessCodeCompleteResults(*S, Context, Results, NumResults);
Douglas Gregor86d9a522009-09-21 16:56:56 +00003075}
3076
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003077static enum CodeCompletionContext::Kind mapCodeCompletionContext(Sema &S,
3078 Sema::ParserCompletionContext PCC) {
3079 switch (PCC) {
John McCallf312b1e2010-08-26 23:41:50 +00003080 case Sema::PCC_Namespace:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003081 return CodeCompletionContext::CCC_TopLevel;
3082
John McCallf312b1e2010-08-26 23:41:50 +00003083 case Sema::PCC_Class:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003084 return CodeCompletionContext::CCC_ClassStructUnion;
3085
John McCallf312b1e2010-08-26 23:41:50 +00003086 case Sema::PCC_ObjCInterface:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003087 return CodeCompletionContext::CCC_ObjCInterface;
3088
John McCallf312b1e2010-08-26 23:41:50 +00003089 case Sema::PCC_ObjCImplementation:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003090 return CodeCompletionContext::CCC_ObjCImplementation;
3091
John McCallf312b1e2010-08-26 23:41:50 +00003092 case Sema::PCC_ObjCInstanceVariableList:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003093 return CodeCompletionContext::CCC_ObjCIvarList;
3094
John McCallf312b1e2010-08-26 23:41:50 +00003095 case Sema::PCC_Template:
3096 case Sema::PCC_MemberTemplate:
Douglas Gregor52779fb2010-09-23 23:01:17 +00003097 if (S.CurContext->isFileContext())
3098 return CodeCompletionContext::CCC_TopLevel;
David Blaikie7530c032012-01-17 06:56:22 +00003099 if (S.CurContext->isRecord())
Douglas Gregor52779fb2010-09-23 23:01:17 +00003100 return CodeCompletionContext::CCC_ClassStructUnion;
David Blaikie7530c032012-01-17 06:56:22 +00003101 return CodeCompletionContext::CCC_Other;
Douglas Gregor52779fb2010-09-23 23:01:17 +00003102
John McCallf312b1e2010-08-26 23:41:50 +00003103 case Sema::PCC_RecoveryInFunction:
Douglas Gregor52779fb2010-09-23 23:01:17 +00003104 return CodeCompletionContext::CCC_Recovery;
Douglas Gregora5450a02010-10-18 22:01:46 +00003105
John McCallf312b1e2010-08-26 23:41:50 +00003106 case Sema::PCC_ForInit:
David Blaikie4e4d0842012-03-11 07:00:24 +00003107 if (S.getLangOpts().CPlusPlus || S.getLangOpts().C99 ||
3108 S.getLangOpts().ObjC1)
Douglas Gregora5450a02010-10-18 22:01:46 +00003109 return CodeCompletionContext::CCC_ParenthesizedExpression;
3110 else
3111 return CodeCompletionContext::CCC_Expression;
3112
3113 case Sema::PCC_Expression:
John McCallf312b1e2010-08-26 23:41:50 +00003114 case Sema::PCC_Condition:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003115 return CodeCompletionContext::CCC_Expression;
3116
John McCallf312b1e2010-08-26 23:41:50 +00003117 case Sema::PCC_Statement:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003118 return CodeCompletionContext::CCC_Statement;
Douglas Gregor72db1082010-08-24 01:11:00 +00003119
John McCallf312b1e2010-08-26 23:41:50 +00003120 case Sema::PCC_Type:
Douglas Gregor72db1082010-08-24 01:11:00 +00003121 return CodeCompletionContext::CCC_Type;
Douglas Gregor02688102010-09-14 23:59:36 +00003122
3123 case Sema::PCC_ParenthesizedExpression:
3124 return CodeCompletionContext::CCC_ParenthesizedExpression;
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00003125
3126 case Sema::PCC_LocalDeclarationSpecifiers:
3127 return CodeCompletionContext::CCC_Type;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003128 }
David Blaikie7530c032012-01-17 06:56:22 +00003129
3130 llvm_unreachable("Invalid ParserCompletionContext!");
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003131}
3132
Douglas Gregorf6961522010-08-27 21:18:54 +00003133/// \brief If we're in a C++ virtual member function, add completion results
3134/// that invoke the functions we override, since it's common to invoke the
3135/// overridden function as well as adding new functionality.
3136///
3137/// \param S The semantic analysis object for which we are generating results.
3138///
3139/// \param InContext This context in which the nested-name-specifier preceding
3140/// the code-completion point
3141static void MaybeAddOverrideCalls(Sema &S, DeclContext *InContext,
3142 ResultBuilder &Results) {
3143 // Look through blocks.
3144 DeclContext *CurContext = S.CurContext;
3145 while (isa<BlockDecl>(CurContext))
3146 CurContext = CurContext->getParent();
3147
3148
3149 CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(CurContext);
3150 if (!Method || !Method->isVirtual())
3151 return;
3152
3153 // We need to have names for all of the parameters, if we're going to
3154 // generate a forwarding call.
Stephen Hines651f13c2014-04-23 16:59:28 -07003155 for (auto P : Method->params())
3156 if (!P->getDeclName())
Douglas Gregorf6961522010-08-27 21:18:54 +00003157 return;
Douglas Gregorf6961522010-08-27 21:18:54 +00003158
Douglas Gregor8987b232011-09-27 23:30:47 +00003159 PrintingPolicy Policy = getCompletionPrintingPolicy(S);
Douglas Gregorf6961522010-08-27 21:18:54 +00003160 for (CXXMethodDecl::method_iterator M = Method->begin_overridden_methods(),
3161 MEnd = Method->end_overridden_methods();
3162 M != MEnd; ++M) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003163 CodeCompletionBuilder Builder(Results.getAllocator(),
3164 Results.getCodeCompletionTUInfo());
Dmitri Gribenko68a932d2013-02-14 13:53:30 +00003165 const CXXMethodDecl *Overridden = *M;
Douglas Gregorf6961522010-08-27 21:18:54 +00003166 if (Overridden->getCanonicalDecl() == Method->getCanonicalDecl())
3167 continue;
3168
3169 // If we need a nested-name-specifier, add one now.
3170 if (!InContext) {
3171 NestedNameSpecifier *NNS
3172 = getRequiredQualification(S.Context, CurContext,
3173 Overridden->getDeclContext());
3174 if (NNS) {
3175 std::string Str;
3176 llvm::raw_string_ostream OS(Str);
Douglas Gregor8987b232011-09-27 23:30:47 +00003177 NNS->print(OS, Policy);
Douglas Gregordae68752011-02-01 22:57:45 +00003178 Builder.AddTextChunk(Results.getAllocator().CopyString(OS.str()));
Douglas Gregorf6961522010-08-27 21:18:54 +00003179 }
3180 } else if (!InContext->Equals(Overridden->getDeclContext()))
3181 continue;
3182
Douglas Gregordae68752011-02-01 22:57:45 +00003183 Builder.AddTypedTextChunk(Results.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00003184 Overridden->getNameAsString()));
3185 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorf6961522010-08-27 21:18:54 +00003186 bool FirstParam = true;
Stephen Hines651f13c2014-04-23 16:59:28 -07003187 for (auto P : Method->params()) {
Douglas Gregorf6961522010-08-27 21:18:54 +00003188 if (FirstParam)
3189 FirstParam = false;
3190 else
Douglas Gregor218937c2011-02-01 19:23:04 +00003191 Builder.AddChunk(CodeCompletionString::CK_Comma);
Douglas Gregorf6961522010-08-27 21:18:54 +00003192
Stephen Hines651f13c2014-04-23 16:59:28 -07003193 Builder.AddPlaceholderChunk(
3194 Results.getAllocator().CopyString(P->getIdentifier()->getName()));
Douglas Gregorf6961522010-08-27 21:18:54 +00003195 }
Douglas Gregor218937c2011-02-01 19:23:04 +00003196 Builder.AddChunk(CodeCompletionString::CK_RightParen);
3197 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregorf6961522010-08-27 21:18:54 +00003198 CCP_SuperCompletion,
Douglas Gregorba103062012-03-27 23:34:16 +00003199 CXCursor_CXXMethod,
3200 CXAvailability_Available,
3201 Overridden));
Douglas Gregorf6961522010-08-27 21:18:54 +00003202 Results.Ignore(Overridden);
3203 }
3204}
3205
Douglas Gregorc5b2e582012-01-29 18:15:03 +00003206void Sema::CodeCompleteModuleImport(SourceLocation ImportLoc,
3207 ModuleIdPath Path) {
3208 typedef CodeCompletionResult Result;
3209 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003210 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregorc5b2e582012-01-29 18:15:03 +00003211 CodeCompletionContext::CCC_Other);
3212 Results.EnterNewScope();
3213
3214 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003215 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregorc5b2e582012-01-29 18:15:03 +00003216 typedef CodeCompletionResult Result;
3217 if (Path.empty()) {
3218 // Enumerate all top-level modules.
Dmitri Gribenkocfa88f82013-01-12 19:30:44 +00003219 SmallVector<Module *, 8> Modules;
Douglas Gregorc5b2e582012-01-29 18:15:03 +00003220 PP.getHeaderSearchInfo().collectAllModules(Modules);
3221 for (unsigned I = 0, N = Modules.size(); I != N; ++I) {
3222 Builder.AddTypedTextChunk(
3223 Builder.getAllocator().CopyString(Modules[I]->Name));
3224 Results.AddResult(Result(Builder.TakeString(),
3225 CCP_Declaration,
Argyrios Kyrtzidisfe038a32013-05-29 18:50:15 +00003226 CXCursor_ModuleImportDecl,
Douglas Gregorc5b2e582012-01-29 18:15:03 +00003227 Modules[I]->isAvailable()
3228 ? CXAvailability_Available
3229 : CXAvailability_NotAvailable));
3230 }
Daniel Jasper056ec122013-08-05 20:26:17 +00003231 } else if (getLangOpts().Modules) {
Douglas Gregorc5b2e582012-01-29 18:15:03 +00003232 // Load the named module.
3233 Module *Mod = PP.getModuleLoader().loadModule(ImportLoc, Path,
3234 Module::AllVisible,
3235 /*IsInclusionDirective=*/false);
3236 // Enumerate submodules.
3237 if (Mod) {
3238 for (Module::submodule_iterator Sub = Mod->submodule_begin(),
3239 SubEnd = Mod->submodule_end();
3240 Sub != SubEnd; ++Sub) {
3241
3242 Builder.AddTypedTextChunk(
3243 Builder.getAllocator().CopyString((*Sub)->Name));
3244 Results.AddResult(Result(Builder.TakeString(),
3245 CCP_Declaration,
Argyrios Kyrtzidisfe038a32013-05-29 18:50:15 +00003246 CXCursor_ModuleImportDecl,
Douglas Gregorc5b2e582012-01-29 18:15:03 +00003247 (*Sub)->isAvailable()
3248 ? CXAvailability_Available
3249 : CXAvailability_NotAvailable));
3250 }
3251 }
3252 }
3253 Results.ExitScope();
3254 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
3255 Results.data(),Results.size());
3256}
3257
Douglas Gregor01dfea02010-01-10 23:08:15 +00003258void Sema::CodeCompleteOrdinaryName(Scope *S,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003259 ParserCompletionContext CompletionContext) {
Douglas Gregor218937c2011-02-01 19:23:04 +00003260 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003261 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00003262 mapCodeCompletionContext(*this, CompletionContext));
Douglas Gregorf6961522010-08-27 21:18:54 +00003263 Results.EnterNewScope();
Douglas Gregorcee9ff12010-09-20 22:39:41 +00003264
Douglas Gregor01dfea02010-01-10 23:08:15 +00003265 // Determine how to filter results, e.g., so that the names of
3266 // values (functions, enumerators, function templates, etc.) are
3267 // only allowed where we can have an expression.
3268 switch (CompletionContext) {
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003269 case PCC_Namespace:
3270 case PCC_Class:
3271 case PCC_ObjCInterface:
3272 case PCC_ObjCImplementation:
3273 case PCC_ObjCInstanceVariableList:
3274 case PCC_Template:
3275 case PCC_MemberTemplate:
Douglas Gregor72db1082010-08-24 01:11:00 +00003276 case PCC_Type:
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00003277 case PCC_LocalDeclarationSpecifiers:
Douglas Gregor01dfea02010-01-10 23:08:15 +00003278 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
3279 break;
3280
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003281 case PCC_Statement:
Douglas Gregor02688102010-09-14 23:59:36 +00003282 case PCC_ParenthesizedExpression:
Douglas Gregoreb0d0142010-08-24 23:58:17 +00003283 case PCC_Expression:
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003284 case PCC_ForInit:
3285 case PCC_Condition:
David Blaikie4e4d0842012-03-11 07:00:24 +00003286 if (WantTypesInContext(CompletionContext, getLangOpts()))
Douglas Gregor4710e5b2010-05-28 00:49:12 +00003287 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3288 else
3289 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregorf6961522010-08-27 21:18:54 +00003290
David Blaikie4e4d0842012-03-11 07:00:24 +00003291 if (getLangOpts().CPlusPlus)
Stephen Hines6bcf27b2014-05-29 04:14:42 -07003292 MaybeAddOverrideCalls(*this, /*InContext=*/nullptr, Results);
Douglas Gregor01dfea02010-01-10 23:08:15 +00003293 break;
Douglas Gregordc845342010-05-25 05:58:43 +00003294
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003295 case PCC_RecoveryInFunction:
Douglas Gregordc845342010-05-25 05:58:43 +00003296 // Unfiltered
3297 break;
Douglas Gregor01dfea02010-01-10 23:08:15 +00003298 }
3299
Douglas Gregor3cdee122010-08-26 16:36:48 +00003300 // If we are in a C++ non-static member function, check the qualifiers on
3301 // the member function to filter/prioritize the results list.
3302 if (CXXMethodDecl *CurMethod = dyn_cast<CXXMethodDecl>(CurContext))
3303 if (CurMethod->isInstance())
3304 Results.setObjectTypeQualifiers(
3305 Qualifiers::fromCVRMask(CurMethod->getTypeQualifiers()));
3306
Douglas Gregor1ca6ae82010-01-14 01:09:38 +00003307 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00003308 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3309 CodeCompleter->includeGlobals());
Douglas Gregor2a7925c2009-12-07 09:54:55 +00003310
Douglas Gregorbca403c2010-01-13 23:51:12 +00003311 AddOrdinaryNameResults(CompletionContext, S, *this, Results);
Douglas Gregor2a7925c2009-12-07 09:54:55 +00003312 Results.ExitScope();
3313
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003314 switch (CompletionContext) {
Douglas Gregor02688102010-09-14 23:59:36 +00003315 case PCC_ParenthesizedExpression:
Douglas Gregor72db1082010-08-24 01:11:00 +00003316 case PCC_Expression:
3317 case PCC_Statement:
3318 case PCC_RecoveryInFunction:
3319 if (S->getFnParent())
David Blaikie4e4d0842012-03-11 07:00:24 +00003320 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregor72db1082010-08-24 01:11:00 +00003321 break;
3322
3323 case PCC_Namespace:
3324 case PCC_Class:
3325 case PCC_ObjCInterface:
3326 case PCC_ObjCImplementation:
3327 case PCC_ObjCInstanceVariableList:
3328 case PCC_Template:
3329 case PCC_MemberTemplate:
3330 case PCC_ForInit:
3331 case PCC_Condition:
3332 case PCC_Type:
Douglas Gregor68e3c2e2011-02-15 20:33:25 +00003333 case PCC_LocalDeclarationSpecifiers:
Douglas Gregor72db1082010-08-24 01:11:00 +00003334 break;
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003335 }
3336
Douglas Gregor0c8296d2009-11-07 00:00:49 +00003337 if (CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00003338 AddMacroResults(PP, Results, false);
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003339
Douglas Gregorcee9ff12010-09-20 22:39:41 +00003340 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003341 Results.data(),Results.size());
Douglas Gregor791215b2009-09-21 20:51:25 +00003342}
3343
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003344static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
3345 ParsedType Receiver,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00003346 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00003347 bool AtArgumentExpression,
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003348 bool IsSuper,
3349 ResultBuilder &Results);
3350
3351void Sema::CodeCompleteDeclSpec(Scope *S, DeclSpec &DS,
3352 bool AllowNonIdentifiers,
3353 bool AllowNestedNameSpecifiers) {
John McCall0a2c5e22010-08-25 06:19:51 +00003354 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00003355 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003356 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00003357 AllowNestedNameSpecifiers
3358 ? CodeCompletionContext::CCC_PotentiallyQualifiedName
3359 : CodeCompletionContext::CCC_Name);
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003360 Results.EnterNewScope();
3361
3362 // Type qualifiers can come after names.
3363 Results.AddResult(Result("const"));
3364 Results.AddResult(Result("volatile"));
David Blaikie4e4d0842012-03-11 07:00:24 +00003365 if (getLangOpts().C99)
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003366 Results.AddResult(Result("restrict"));
3367
David Blaikie4e4d0842012-03-11 07:00:24 +00003368 if (getLangOpts().CPlusPlus) {
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003369 if (AllowNonIdentifiers) {
3370 Results.AddResult(Result("operator"));
3371 }
3372
3373 // Add nested-name-specifiers.
3374 if (AllowNestedNameSpecifiers) {
3375 Results.allowNestedNameSpecifiers();
Douglas Gregor52779fb2010-09-23 23:01:17 +00003376 Results.setFilter(&ResultBuilder::IsImpossibleToSatisfy);
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003377 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3378 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer,
3379 CodeCompleter->includeGlobals());
Stephen Hines6bcf27b2014-05-29 04:14:42 -07003380 Results.setFilter(nullptr);
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003381 }
3382 }
3383 Results.ExitScope();
3384
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003385 // If we're in a context where we might have an expression (rather than a
3386 // declaration), and what we've seen so far is an Objective-C type that could
3387 // be a receiver of a class message, this may be a class message send with
3388 // the initial opening bracket '[' missing. Add appropriate completions.
3389 if (AllowNonIdentifiers && !AllowNestedNameSpecifiers &&
Richard Smithec642442013-04-12 22:46:28 +00003390 DS.getParsedSpecifiers() == DeclSpec::PQ_TypeSpecifier &&
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003391 DS.getTypeSpecType() == DeclSpec::TST_typename &&
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003392 DS.getTypeSpecComplex() == DeclSpec::TSC_unspecified &&
3393 DS.getTypeSpecSign() == DeclSpec::TSS_unspecified &&
Richard Smithec642442013-04-12 22:46:28 +00003394 !DS.isTypeAltiVecVector() &&
3395 S &&
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003396 (S->getFlags() & Scope::DeclScope) != 0 &&
3397 (S->getFlags() & (Scope::ClassScope | Scope::TemplateParamScope |
3398 Scope::FunctionPrototypeScope |
3399 Scope::AtCatchScope)) == 0) {
3400 ParsedType T = DS.getRepAsType();
3401 if (!T.get().isNull() && T.get()->isObjCObjectOrInterfaceType())
Dmitri Gribenko050315b2013-06-16 03:47:57 +00003402 AddClassMessageCompletions(*this, S, T, None, false, false, Results);
Douglas Gregorc7b6d882010-09-16 15:14:18 +00003403 }
3404
Douglas Gregor4497dd42010-08-24 04:59:56 +00003405 // Note that we intentionally suppress macro results here, since we do not
3406 // encourage using macros to produce the names of entities.
3407
Douglas Gregor52779fb2010-09-23 23:01:17 +00003408 HandleCodeCompleteResults(this, CodeCompleter,
3409 Results.getCompletionContext(),
Douglas Gregor2ccccb32010-08-23 18:23:48 +00003410 Results.data(), Results.size());
3411}
3412
Douglas Gregorfb629412010-08-23 21:17:50 +00003413struct Sema::CodeCompleteExpressionData {
3414 CodeCompleteExpressionData(QualType PreferredType = QualType())
3415 : PreferredType(PreferredType), IntegralConstantExpression(false),
3416 ObjCCollection(false) { }
3417
3418 QualType PreferredType;
3419 bool IntegralConstantExpression;
3420 bool ObjCCollection;
Chris Lattner5f9e2722011-07-23 10:55:15 +00003421 SmallVector<Decl *, 4> IgnoreDecls;
Douglas Gregorfb629412010-08-23 21:17:50 +00003422};
3423
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003424/// \brief Perform code-completion in an expression context when we know what
3425/// type we're looking for.
Douglas Gregorfb629412010-08-23 21:17:50 +00003426void Sema::CodeCompleteExpression(Scope *S,
3427 const CodeCompleteExpressionData &Data) {
Douglas Gregor218937c2011-02-01 19:23:04 +00003428 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003429 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00003430 CodeCompletionContext::CCC_Expression);
Douglas Gregorfb629412010-08-23 21:17:50 +00003431 if (Data.ObjCCollection)
3432 Results.setFilter(&ResultBuilder::IsObjCCollection);
3433 else if (Data.IntegralConstantExpression)
Douglas Gregorf9578432010-07-28 21:50:18 +00003434 Results.setFilter(&ResultBuilder::IsIntegralConstantValue);
David Blaikie4e4d0842012-03-11 07:00:24 +00003435 else if (WantTypesInContext(PCC_Expression, getLangOpts()))
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003436 Results.setFilter(&ResultBuilder::IsOrdinaryName);
3437 else
3438 Results.setFilter(&ResultBuilder::IsOrdinaryNonTypeName);
Douglas Gregorfb629412010-08-23 21:17:50 +00003439
3440 if (!Data.PreferredType.isNull())
3441 Results.setPreferredType(Data.PreferredType.getNonReferenceType());
3442
3443 // Ignore any declarations that we were told that we don't care about.
3444 for (unsigned I = 0, N = Data.IgnoreDecls.size(); I != N; ++I)
3445 Results.Ignore(Data.IgnoreDecls[I]);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003446
3447 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00003448 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
3449 CodeCompleter->includeGlobals());
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003450
3451 Results.EnterNewScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003452 AddOrdinaryNameResults(PCC_Expression, S, *this, Results);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003453 Results.ExitScope();
3454
Douglas Gregor590c7d52010-07-08 20:55:51 +00003455 bool PreferredTypeIsPointer = false;
Douglas Gregorfb629412010-08-23 21:17:50 +00003456 if (!Data.PreferredType.isNull())
3457 PreferredTypeIsPointer = Data.PreferredType->isAnyPointerType()
3458 || Data.PreferredType->isMemberPointerType()
3459 || Data.PreferredType->isBlockPointerType();
Douglas Gregor590c7d52010-07-08 20:55:51 +00003460
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003461 if (S->getFnParent() &&
3462 !Data.ObjCCollection &&
3463 !Data.IntegralConstantExpression)
David Blaikie4e4d0842012-03-11 07:00:24 +00003464 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregoraa5f77b2010-08-23 21:54:33 +00003465
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003466 if (CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00003467 AddMacroResults(PP, Results, false, PreferredTypeIsPointer);
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003468 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregorfb629412010-08-23 21:17:50 +00003469 CodeCompletionContext(CodeCompletionContext::CCC_Expression,
3470 Data.PreferredType),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003471 Results.data(),Results.size());
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003472}
3473
Douglas Gregorac5fd842010-09-18 01:28:11 +00003474void Sema::CodeCompletePostfixExpression(Scope *S, ExprResult E) {
3475 if (E.isInvalid())
3476 CodeCompleteOrdinaryName(S, PCC_RecoveryInFunction);
David Blaikie4e4d0842012-03-11 07:00:24 +00003477 else if (getLangOpts().ObjC1)
Stephen Hinesc568f1e2014-07-21 00:47:37 -07003478 CodeCompleteObjCInstanceMessage(S, E.get(), None, false);
Douglas Gregor78edf512010-09-15 16:23:04 +00003479}
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003480
Douglas Gregor73449212010-12-09 23:01:55 +00003481/// \brief The set of properties that have already been added, referenced by
3482/// property name.
3483typedef llvm::SmallPtrSet<IdentifierInfo*, 16> AddedPropertiesSet;
3484
Douglas Gregorb92a4082012-06-12 13:44:08 +00003485/// \brief Retrieve the container definition, if any?
3486static ObjCContainerDecl *getContainerDef(ObjCContainerDecl *Container) {
3487 if (ObjCInterfaceDecl *Interface = dyn_cast<ObjCInterfaceDecl>(Container)) {
3488 if (Interface->hasDefinition())
3489 return Interface->getDefinition();
3490
3491 return Interface;
3492 }
3493
3494 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
3495 if (Protocol->hasDefinition())
3496 return Protocol->getDefinition();
3497
3498 return Protocol;
3499 }
3500 return Container;
3501}
3502
3503static void AddObjCProperties(ObjCContainerDecl *Container,
Douglas Gregor322328b2009-11-18 22:32:06 +00003504 bool AllowCategories,
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003505 bool AllowNullaryMethods,
Douglas Gregor95ac6552009-11-18 01:29:26 +00003506 DeclContext *CurContext,
Douglas Gregor73449212010-12-09 23:01:55 +00003507 AddedPropertiesSet &AddedProperties,
Douglas Gregor95ac6552009-11-18 01:29:26 +00003508 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00003509 typedef CodeCompletionResult Result;
Douglas Gregor95ac6552009-11-18 01:29:26 +00003510
Douglas Gregorb92a4082012-06-12 13:44:08 +00003511 // Retrieve the definition.
3512 Container = getContainerDef(Container);
3513
Douglas Gregor95ac6552009-11-18 01:29:26 +00003514 // Add properties in this container.
Stephen Hines651f13c2014-04-23 16:59:28 -07003515 for (const auto *P : Container->properties())
Stephen Hines176edba2014-12-01 14:53:08 -08003516 if (AddedProperties.insert(P->getIdentifier()).second)
Stephen Hines6bcf27b2014-05-29 04:14:42 -07003517 Results.MaybeAddResult(Result(P, Results.getBasePriority(P), nullptr),
Douglas Gregord1f09b42013-01-31 04:52:16 +00003518 CurContext);
Stephen Hines6bcf27b2014-05-29 04:14:42 -07003519
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003520 // Add nullary methods
3521 if (AllowNullaryMethods) {
3522 ASTContext &Context = Container->getASTContext();
Douglas Gregor8987b232011-09-27 23:30:47 +00003523 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
Stephen Hines651f13c2014-04-23 16:59:28 -07003524 for (auto *M : Container->methods()) {
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003525 if (M->getSelector().isUnarySelector())
3526 if (IdentifierInfo *Name = M->getSelector().getIdentifierInfoForSlot(0))
Stephen Hines176edba2014-12-01 14:53:08 -08003527 if (AddedProperties.insert(Name).second) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003528 CodeCompletionBuilder Builder(Results.getAllocator(),
3529 Results.getCodeCompletionTUInfo());
Stephen Hines651f13c2014-04-23 16:59:28 -07003530 AddResultTypeChunk(Context, Policy, M, Builder);
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003531 Builder.AddTypedTextChunk(
3532 Results.getAllocator().CopyString(Name->getName()));
3533
Stephen Hines651f13c2014-04-23 16:59:28 -07003534 Results.MaybeAddResult(Result(Builder.TakeString(), M,
Douglas Gregorba103062012-03-27 23:34:16 +00003535 CCP_MemberDeclaration + CCD_MethodAsProperty),
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003536 CurContext);
3537 }
3538 }
3539 }
3540
3541
Douglas Gregor95ac6552009-11-18 01:29:26 +00003542 // Add properties in referenced protocols.
3543 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Stephen Hines651f13c2014-04-23 16:59:28 -07003544 for (auto *P : Protocol->protocols())
3545 AddObjCProperties(P, AllowCategories, AllowNullaryMethods, CurContext,
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003546 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003547 } else if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)){
Douglas Gregor322328b2009-11-18 22:32:06 +00003548 if (AllowCategories) {
3549 // Look through categories.
Stephen Hines651f13c2014-04-23 16:59:28 -07003550 for (auto *Cat : IFace->known_categories())
3551 AddObjCProperties(Cat, AllowCategories, AllowNullaryMethods, CurContext,
3552 AddedProperties, Results);
Douglas Gregor322328b2009-11-18 22:32:06 +00003553 }
Stephen Hines651f13c2014-04-23 16:59:28 -07003554
Douglas Gregor95ac6552009-11-18 01:29:26 +00003555 // Look through protocols.
Stephen Hines651f13c2014-04-23 16:59:28 -07003556 for (auto *I : IFace->all_referenced_protocols())
3557 AddObjCProperties(I, AllowCategories, AllowNullaryMethods, CurContext,
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003558 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003559
3560 // Look in the superclass.
3561 if (IFace->getSuperClass())
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003562 AddObjCProperties(IFace->getSuperClass(), AllowCategories,
3563 AllowNullaryMethods, CurContext,
Douglas Gregor73449212010-12-09 23:01:55 +00003564 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003565 } else if (const ObjCCategoryDecl *Category
3566 = dyn_cast<ObjCCategoryDecl>(Container)) {
3567 // Look through protocols.
Stephen Hines651f13c2014-04-23 16:59:28 -07003568 for (auto *P : Category->protocols())
3569 AddObjCProperties(P, AllowCategories, AllowNullaryMethods, CurContext,
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003570 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003571 }
3572}
3573
Douglas Gregorf5cd27d2012-01-23 15:59:30 +00003574void Sema::CodeCompleteMemberReferenceExpr(Scope *S, Expr *Base,
Douglas Gregor81b747b2009-09-17 21:32:03 +00003575 SourceLocation OpLoc,
3576 bool IsArrow) {
Douglas Gregorf5cd27d2012-01-23 15:59:30 +00003577 if (!Base || !CodeCompleter)
Douglas Gregor81b747b2009-09-17 21:32:03 +00003578 return;
3579
Douglas Gregorf5cd27d2012-01-23 15:59:30 +00003580 ExprResult ConvertedBase = PerformMemberExprBaseConversion(Base, IsArrow);
3581 if (ConvertedBase.isInvalid())
3582 return;
3583 Base = ConvertedBase.get();
3584
John McCall0a2c5e22010-08-25 06:19:51 +00003585 typedef CodeCompletionResult Result;
Douglas Gregor86d9a522009-09-21 16:56:56 +00003586
Douglas Gregor81b747b2009-09-17 21:32:03 +00003587 QualType BaseType = Base->getType();
Douglas Gregor86d9a522009-09-21 16:56:56 +00003588
3589 if (IsArrow) {
3590 if (const PointerType *Ptr = BaseType->getAs<PointerType>())
3591 BaseType = Ptr->getPointeeType();
3592 else if (BaseType->isObjCObjectPointerType())
Douglas Gregor3cdee122010-08-26 16:36:48 +00003593 /*Do nothing*/ ;
Douglas Gregor86d9a522009-09-21 16:56:56 +00003594 else
3595 return;
3596 }
3597
Douglas Gregor3da626b2011-07-07 16:03:39 +00003598 enum CodeCompletionContext::Kind contextKind;
3599
3600 if (IsArrow) {
3601 contextKind = CodeCompletionContext::CCC_ArrowMemberAccess;
3602 }
3603 else {
3604 if (BaseType->isObjCObjectPointerType() ||
3605 BaseType->isObjCObjectOrInterfaceType()) {
3606 contextKind = CodeCompletionContext::CCC_ObjCPropertyAccess;
3607 }
3608 else {
3609 contextKind = CodeCompletionContext::CCC_DotMemberAccess;
3610 }
3611 }
3612
Douglas Gregor218937c2011-02-01 19:23:04 +00003613 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003614 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor3da626b2011-07-07 16:03:39 +00003615 CodeCompletionContext(contextKind,
Douglas Gregor52779fb2010-09-23 23:01:17 +00003616 BaseType),
3617 &ResultBuilder::IsMember);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003618 Results.EnterNewScope();
3619 if (const RecordType *Record = BaseType->getAs<RecordType>()) {
Douglas Gregor3cdee122010-08-26 16:36:48 +00003620 // Indicate that we are performing a member access, and the cv-qualifiers
3621 // for the base object type.
3622 Results.setObjectTypeQualifiers(BaseType.getQualifiers());
3623
Douglas Gregor95ac6552009-11-18 01:29:26 +00003624 // Access to a C/C++ class, struct, or union.
Douglas Gregor45bcd432010-01-14 03:21:49 +00003625 Results.allowNestedNameSpecifiers();
Douglas Gregor0cc84042010-01-14 15:47:35 +00003626 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00003627 LookupVisibleDecls(Record->getDecl(), LookupMemberName, Consumer,
3628 CodeCompleter->includeGlobals());
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003629
David Blaikie4e4d0842012-03-11 07:00:24 +00003630 if (getLangOpts().CPlusPlus) {
Douglas Gregor95ac6552009-11-18 01:29:26 +00003631 if (!Results.empty()) {
3632 // The "template" keyword can follow "->" or "." in the grammar.
3633 // However, we only want to suggest the template keyword if something
3634 // is dependent.
3635 bool IsDependent = BaseType->isDependentType();
3636 if (!IsDependent) {
3637 for (Scope *DepScope = S; DepScope; DepScope = DepScope->getParent())
Ted Kremenekf0d58612013-10-08 17:08:03 +00003638 if (DeclContext *Ctx = DepScope->getEntity()) {
Douglas Gregor95ac6552009-11-18 01:29:26 +00003639 IsDependent = Ctx->isDependentContext();
3640 break;
3641 }
3642 }
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003643
Douglas Gregor95ac6552009-11-18 01:29:26 +00003644 if (IsDependent)
Douglas Gregora4477812010-01-14 16:01:26 +00003645 Results.AddResult(Result("template"));
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003646 }
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003647 }
Douglas Gregor95ac6552009-11-18 01:29:26 +00003648 } else if (!IsArrow && BaseType->getAsObjCInterfacePointerType()) {
3649 // Objective-C property reference.
Douglas Gregor73449212010-12-09 23:01:55 +00003650 AddedPropertiesSet AddedProperties;
Douglas Gregor95ac6552009-11-18 01:29:26 +00003651
3652 // Add property results based on our interface.
3653 const ObjCObjectPointerType *ObjCPtr
3654 = BaseType->getAsObjCInterfacePointerType();
3655 assert(ObjCPtr && "Non-NULL pointer guaranteed above!");
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003656 AddObjCProperties(ObjCPtr->getInterfaceDecl(), true,
3657 /*AllowNullaryMethods=*/true, CurContext,
Douglas Gregor73449212010-12-09 23:01:55 +00003658 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003659
3660 // Add properties from the protocols in a qualified interface.
Stephen Hines651f13c2014-04-23 16:59:28 -07003661 for (auto *I : ObjCPtr->quals())
3662 AddObjCProperties(I, true, /*AllowNullaryMethods=*/true, CurContext,
Douglas Gregor4b81cde2011-05-05 15:50:42 +00003663 AddedProperties, Results);
Douglas Gregor95ac6552009-11-18 01:29:26 +00003664 } else if ((IsArrow && BaseType->isObjCObjectPointerType()) ||
John McCallc12c5bb2010-05-15 11:32:37 +00003665 (!IsArrow && BaseType->isObjCObjectType())) {
Douglas Gregor95ac6552009-11-18 01:29:26 +00003666 // Objective-C instance variable access.
Stephen Hines6bcf27b2014-05-29 04:14:42 -07003667 ObjCInterfaceDecl *Class = nullptr;
Douglas Gregor95ac6552009-11-18 01:29:26 +00003668 if (const ObjCObjectPointerType *ObjCPtr
3669 = BaseType->getAs<ObjCObjectPointerType>())
3670 Class = ObjCPtr->getInterfaceDecl();
3671 else
John McCallc12c5bb2010-05-15 11:32:37 +00003672 Class = BaseType->getAs<ObjCObjectType>()->getInterface();
Douglas Gregor95ac6552009-11-18 01:29:26 +00003673
3674 // Add all ivars from this class and its superclasses.
Douglas Gregor80f4f4c2010-01-14 16:08:12 +00003675 if (Class) {
3676 CodeCompletionDeclConsumer Consumer(Results, CurContext);
3677 Results.setFilter(&ResultBuilder::IsObjCIvar);
Douglas Gregor8071e422010-08-15 06:18:01 +00003678 LookupVisibleDecls(Class, LookupMemberName, Consumer,
3679 CodeCompleter->includeGlobals());
Douglas Gregor95ac6552009-11-18 01:29:26 +00003680 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00003681 }
Douglas Gregor95ac6552009-11-18 01:29:26 +00003682
3683 // FIXME: How do we cope with isa?
3684
3685 Results.ExitScope();
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003686
Daniel Dunbar3a2838d2009-11-13 08:58:20 +00003687 // Hand off the results found for code completion.
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003688 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00003689 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003690 Results.data(),Results.size());
Douglas Gregor81b747b2009-09-17 21:32:03 +00003691}
3692
Douglas Gregor374929f2009-09-18 15:37:17 +00003693void Sema::CodeCompleteTag(Scope *S, unsigned TagSpec) {
3694 if (!CodeCompleter)
3695 return;
Stephen Hines6bcf27b2014-05-29 04:14:42 -07003696
3697 ResultBuilder::LookupFilter Filter = nullptr;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003698 enum CodeCompletionContext::Kind ContextKind
3699 = CodeCompletionContext::CCC_Other;
Douglas Gregor374929f2009-09-18 15:37:17 +00003700 switch ((DeclSpec::TST)TagSpec) {
3701 case DeclSpec::TST_enum:
Douglas Gregor86d9a522009-09-21 16:56:56 +00003702 Filter = &ResultBuilder::IsEnum;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003703 ContextKind = CodeCompletionContext::CCC_EnumTag;
Douglas Gregor374929f2009-09-18 15:37:17 +00003704 break;
3705
3706 case DeclSpec::TST_union:
Douglas Gregor86d9a522009-09-21 16:56:56 +00003707 Filter = &ResultBuilder::IsUnion;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003708 ContextKind = CodeCompletionContext::CCC_UnionTag;
Douglas Gregor374929f2009-09-18 15:37:17 +00003709 break;
3710
3711 case DeclSpec::TST_struct:
Douglas Gregor374929f2009-09-18 15:37:17 +00003712 case DeclSpec::TST_class:
Joao Matos6666ed42012-08-31 18:45:21 +00003713 case DeclSpec::TST_interface:
Douglas Gregor86d9a522009-09-21 16:56:56 +00003714 Filter = &ResultBuilder::IsClassOrStruct;
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003715 ContextKind = CodeCompletionContext::CCC_ClassOrStructTag;
Douglas Gregor374929f2009-09-18 15:37:17 +00003716 break;
3717
3718 default:
David Blaikieb219cfc2011-09-23 05:06:16 +00003719 llvm_unreachable("Unknown type specifier kind in CodeCompleteTag");
Douglas Gregor374929f2009-09-18 15:37:17 +00003720 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00003721
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003722 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
3723 CodeCompleter->getCodeCompletionTUInfo(), ContextKind);
Douglas Gregor5d2fc402010-01-14 03:27:13 +00003724 CodeCompletionDeclConsumer Consumer(Results, CurContext);
John McCall0d6b1642010-04-23 18:46:30 +00003725
3726 // First pass: look for tags.
3727 Results.setFilter(Filter);
Douglas Gregor8071e422010-08-15 06:18:01 +00003728 LookupVisibleDecls(S, LookupTagName, Consumer,
3729 CodeCompleter->includeGlobals());
John McCall0d6b1642010-04-23 18:46:30 +00003730
Douglas Gregor8071e422010-08-15 06:18:01 +00003731 if (CodeCompleter->includeGlobals()) {
3732 // Second pass: look for nested name specifiers.
3733 Results.setFilter(&ResultBuilder::IsNestedNameSpecifier);
3734 LookupVisibleDecls(S, LookupNestedNameSpecifierName, Consumer);
3735 }
Douglas Gregor86d9a522009-09-21 16:56:56 +00003736
Douglas Gregor52779fb2010-09-23 23:01:17 +00003737 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003738 Results.data(),Results.size());
Douglas Gregor374929f2009-09-18 15:37:17 +00003739}
3740
Douglas Gregor1a480c42010-08-27 17:35:51 +00003741void Sema::CodeCompleteTypeQualifiers(DeclSpec &DS) {
Douglas Gregor218937c2011-02-01 19:23:04 +00003742 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003743 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00003744 CodeCompletionContext::CCC_TypeQualifiers);
Douglas Gregor1a480c42010-08-27 17:35:51 +00003745 Results.EnterNewScope();
3746 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_const))
3747 Results.AddResult("const");
3748 if (!(DS.getTypeQualifiers() & DeclSpec::TQ_volatile))
3749 Results.AddResult("volatile");
David Blaikie4e4d0842012-03-11 07:00:24 +00003750 if (getLangOpts().C99 &&
Douglas Gregor1a480c42010-08-27 17:35:51 +00003751 !(DS.getTypeQualifiers() & DeclSpec::TQ_restrict))
3752 Results.AddResult("restrict");
Richard Smith4cf4a5e2013-03-28 01:55:44 +00003753 if (getLangOpts().C11 &&
3754 !(DS.getTypeQualifiers() & DeclSpec::TQ_atomic))
3755 Results.AddResult("_Atomic");
Douglas Gregor1a480c42010-08-27 17:35:51 +00003756 Results.ExitScope();
3757 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00003758 Results.getCompletionContext(),
Douglas Gregor1a480c42010-08-27 17:35:51 +00003759 Results.data(), Results.size());
3760}
3761
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003762void Sema::CodeCompleteCase(Scope *S) {
John McCall781472f2010-08-25 08:40:02 +00003763 if (getCurFunction()->SwitchStack.empty() || !CodeCompleter)
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003764 return;
John McCalla8e0cd82011-08-06 07:30:58 +00003765
John McCall781472f2010-08-25 08:40:02 +00003766 SwitchStmt *Switch = getCurFunction()->SwitchStack.back();
John McCalla8e0cd82011-08-06 07:30:58 +00003767 QualType type = Switch->getCond()->IgnoreImplicit()->getType();
3768 if (!type->isEnumeralType()) {
3769 CodeCompleteExpressionData Data(type);
Douglas Gregorfb629412010-08-23 21:17:50 +00003770 Data.IntegralConstantExpression = true;
3771 CodeCompleteExpression(S, Data);
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003772 return;
Douglas Gregorf9578432010-07-28 21:50:18 +00003773 }
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003774
3775 // Code-complete the cases of a switch statement over an enumeration type
3776 // by providing the list of
John McCalla8e0cd82011-08-06 07:30:58 +00003777 EnumDecl *Enum = type->castAs<EnumType>()->getDecl();
Douglas Gregorb92a4082012-06-12 13:44:08 +00003778 if (EnumDecl *Def = Enum->getDefinition())
3779 Enum = Def;
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003780
3781 // Determine which enumerators we have already seen in the switch statement.
3782 // FIXME: Ideally, we would also be able to look *past* the code-completion
3783 // token, in case we are code-completing in the middle of the switch and not
3784 // at the end. However, we aren't able to do so at the moment.
3785 llvm::SmallPtrSet<EnumConstantDecl *, 8> EnumeratorsSeen;
Stephen Hines6bcf27b2014-05-29 04:14:42 -07003786 NestedNameSpecifier *Qualifier = nullptr;
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003787 for (SwitchCase *SC = Switch->getSwitchCaseList(); SC;
3788 SC = SC->getNextSwitchCase()) {
3789 CaseStmt *Case = dyn_cast<CaseStmt>(SC);
3790 if (!Case)
3791 continue;
3792
3793 Expr *CaseVal = Case->getLHS()->IgnoreParenCasts();
3794 if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(CaseVal))
3795 if (EnumConstantDecl *Enumerator
3796 = dyn_cast<EnumConstantDecl>(DRE->getDecl())) {
3797 // We look into the AST of the case statement to determine which
3798 // enumerator was named. Alternatively, we could compute the value of
3799 // the integral constant expression, then compare it against the
3800 // values of each enumerator. However, value-based approach would not
3801 // work as well with C++ templates where enumerators declared within a
3802 // template are type- and value-dependent.
3803 EnumeratorsSeen.insert(Enumerator);
3804
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00003805 // If this is a qualified-id, keep track of the nested-name-specifier
3806 // so that we can reproduce it as part of code completion, e.g.,
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003807 //
3808 // switch (TagD.getKind()) {
3809 // case TagDecl::TK_enum:
3810 // break;
3811 // case XXX
3812 //
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00003813 // At the XXX, our completions are TagDecl::TK_union,
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003814 // TagDecl::TK_struct, and TagDecl::TK_class, rather than TK_union,
3815 // TK_struct, and TK_class.
Douglas Gregora2813ce2009-10-23 18:54:35 +00003816 Qualifier = DRE->getQualifier();
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003817 }
3818 }
3819
David Blaikie4e4d0842012-03-11 07:00:24 +00003820 if (getLangOpts().CPlusPlus && !Qualifier && EnumeratorsSeen.empty()) {
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00003821 // If there are no prior enumerators in C++, check whether we have to
3822 // qualify the names of the enumerators that we suggest, because they
3823 // may not be visible in this scope.
Douglas Gregorb223d8c2012-02-01 05:02:47 +00003824 Qualifier = getRequiredQualification(Context, CurContext, Enum);
Douglas Gregorb9d0ef72009-09-21 19:57:38 +00003825 }
3826
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003827 // Add any enumerators that have not yet been mentioned.
Douglas Gregor218937c2011-02-01 19:23:04 +00003828 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00003829 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00003830 CodeCompletionContext::CCC_Expression);
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003831 Results.EnterNewScope();
Stephen Hines651f13c2014-04-23 16:59:28 -07003832 for (auto *E : Enum->enumerators()) {
3833 if (EnumeratorsSeen.count(E))
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003834 continue;
3835
Stephen Hines651f13c2014-04-23 16:59:28 -07003836 CodeCompletionResult R(E, CCP_EnumInCase, Qualifier);
Stephen Hines6bcf27b2014-05-29 04:14:42 -07003837 Results.AddResult(R, CurContext, nullptr, false);
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003838 }
3839 Results.ExitScope();
Douglas Gregor2f880e42010-04-06 20:02:15 +00003840
Douglas Gregor3da626b2011-07-07 16:03:39 +00003841 //We need to make sure we're setting the right context,
3842 //so only say we include macros if the code completer says we do
3843 enum CodeCompletionContext::Kind kind = CodeCompletionContext::CCC_Other;
3844 if (CodeCompleter->includeMacros()) {
Douglas Gregor3644d972012-10-09 16:01:50 +00003845 AddMacroResults(PP, Results, false);
Douglas Gregor3da626b2011-07-07 16:03:39 +00003846 kind = CodeCompletionContext::CCC_OtherWithMacros;
3847 }
3848
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003849 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor3da626b2011-07-07 16:03:39 +00003850 kind,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003851 Results.data(),Results.size());
Douglas Gregor3e1005f2009-09-21 18:10:23 +00003852}
3853
Robert Wilhelm834c0582013-08-09 18:02:13 +00003854static bool anyNullArguments(ArrayRef<Expr *> Args) {
Ahmed Charles13a140c2012-02-25 11:00:22 +00003855 if (Args.size() && !Args.data())
Douglas Gregord28dcd72010-05-30 06:10:08 +00003856 return true;
Ahmed Charles13a140c2012-02-25 11:00:22 +00003857
3858 for (unsigned I = 0; I != Args.size(); ++I)
Douglas Gregord28dcd72010-05-30 06:10:08 +00003859 if (!Args[I])
3860 return true;
Ahmed Charles13a140c2012-02-25 11:00:22 +00003861
Douglas Gregord28dcd72010-05-30 06:10:08 +00003862 return false;
3863}
3864
Stephen Hines0e2c34f2015-03-23 12:09:02 -07003865typedef CodeCompleteConsumer::OverloadCandidate ResultCandidate;
3866
3867static void mergeCandidatesWithResults(Sema &SemaRef,
3868 SmallVectorImpl<ResultCandidate> &Results,
3869 OverloadCandidateSet &CandidateSet,
3870 SourceLocation Loc) {
3871 if (!CandidateSet.empty()) {
3872 // Sort the overload candidate set by placing the best overloads first.
3873 std::stable_sort(
3874 CandidateSet.begin(), CandidateSet.end(),
3875 [&](const OverloadCandidate &X, const OverloadCandidate &Y) {
3876 return isBetterOverloadCandidate(SemaRef, X, Y, Loc);
3877 });
3878
3879 // Add the remaining viable overload candidates as code-completion results.
3880 for (auto &Candidate : CandidateSet)
3881 if (Candidate.Viable)
3882 Results.push_back(ResultCandidate(Candidate.Function));
3883 }
3884}
3885
3886/// \brief Get the type of the Nth parameter from a given set of overload
3887/// candidates.
3888static QualType getParamType(Sema &SemaRef,
3889 ArrayRef<ResultCandidate> Candidates,
3890 unsigned N) {
3891
3892 // Given the overloads 'Candidates' for a function call matching all arguments
3893 // up to N, return the type of the Nth parameter if it is the same for all
3894 // overload candidates.
3895 QualType ParamType;
3896 for (auto &Candidate : Candidates) {
3897 if (auto FType = Candidate.getFunctionType())
3898 if (auto Proto = dyn_cast<FunctionProtoType>(FType))
3899 if (N < Proto->getNumParams()) {
3900 if (ParamType.isNull())
3901 ParamType = Proto->getParamType(N);
3902 else if (!SemaRef.Context.hasSameUnqualifiedType(
3903 ParamType.getNonReferenceType(),
3904 Proto->getParamType(N).getNonReferenceType()))
3905 // Otherwise return a default-constructed QualType.
3906 return QualType();
3907 }
3908 }
3909
3910 return ParamType;
3911}
3912
3913static void CodeCompleteOverloadResults(Sema &SemaRef, Scope *S,
3914 MutableArrayRef<ResultCandidate> Candidates,
3915 unsigned CurrentArg,
3916 bool CompleteExpressionWithCurrentArg = true) {
3917 QualType ParamType;
3918 if (CompleteExpressionWithCurrentArg)
3919 ParamType = getParamType(SemaRef, Candidates, CurrentArg);
3920
3921 if (ParamType.isNull())
3922 SemaRef.CodeCompleteOrdinaryName(S, Sema::PCC_Expression);
3923 else
3924 SemaRef.CodeCompleteExpression(S, ParamType);
3925
3926 if (!Candidates.empty())
3927 SemaRef.CodeCompleter->ProcessOverloadCandidates(SemaRef, CurrentArg,
3928 Candidates.data(),
3929 Candidates.size());
3930}
3931
3932void Sema::CodeCompleteCall(Scope *S, Expr *Fn, ArrayRef<Expr *> Args) {
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003933 if (!CodeCompleter)
3934 return;
Douglas Gregoref96eac2009-12-11 19:06:04 +00003935
3936 // When we're code-completing for a call, we fall back to ordinary
3937 // name code-completion whenever we can't produce specific
3938 // results. We may want to revisit this strategy in the future,
3939 // e.g., by merging the two kinds of results.
3940
Stephen Hines0e2c34f2015-03-23 12:09:02 -07003941 // FIXME: Provide support for variadic template functions.
Douglas Gregoref96eac2009-12-11 19:06:04 +00003942
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003943 // Ignore type-dependent call expressions entirely.
Ahmed Charles13a140c2012-02-25 11:00:22 +00003944 if (!Fn || Fn->isTypeDependent() || anyNullArguments(Args) ||
3945 Expr::hasAnyTypeDependentArguments(Args)) {
Douglas Gregore6b1bb62010-08-11 21:23:17 +00003946 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003947 return;
Douglas Gregoref96eac2009-12-11 19:06:04 +00003948 }
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00003949
John McCall3b4294e2009-12-16 12:17:52 +00003950 // Build an overload candidate set based on the functions we find.
John McCall5769d612010-02-08 23:07:23 +00003951 SourceLocation Loc = Fn->getExprLoc();
Stephen Hines6bcf27b2014-05-29 04:14:42 -07003952 OverloadCandidateSet CandidateSet(Loc, OverloadCandidateSet::CSK_Normal);
John McCall3b4294e2009-12-16 12:17:52 +00003953
Chris Lattner5f9e2722011-07-23 10:55:15 +00003954 SmallVector<ResultCandidate, 8> Results;
Douglas Gregorc0265402010-01-21 15:46:19 +00003955
John McCall3b4294e2009-12-16 12:17:52 +00003956 Expr *NakedFn = Fn->IgnoreParenCasts();
Stephen Hines0e2c34f2015-03-23 12:09:02 -07003957 if (auto ULE = dyn_cast<UnresolvedLookupExpr>(NakedFn))
Ahmed Charles13a140c2012-02-25 11:00:22 +00003958 AddOverloadedCallCandidates(ULE, Args, CandidateSet,
Stephen Hines0e2c34f2015-03-23 12:09:02 -07003959 /*PartialOverloading=*/true);
3960 else if (auto UME = dyn_cast<UnresolvedMemberExpr>(NakedFn)) {
3961 TemplateArgumentListInfo TemplateArgsBuffer, *TemplateArgs = nullptr;
3962 if (UME->hasExplicitTemplateArgs()) {
3963 UME->copyTemplateArgumentsInto(TemplateArgsBuffer);
3964 TemplateArgs = &TemplateArgsBuffer;
Douglas Gregorc0265402010-01-21 15:46:19 +00003965 }
Stephen Hines0e2c34f2015-03-23 12:09:02 -07003966 SmallVector<Expr *, 12> ArgExprs(1, UME->getBase());
3967 ArgExprs.append(Args.begin(), Args.end());
3968 UnresolvedSet<8> Decls;
3969 Decls.append(UME->decls_begin(), UME->decls_end());
3970 AddFunctionCandidates(Decls, ArgExprs, CandidateSet, TemplateArgs,
3971 /*SuppressUsedConversions=*/false,
3972 /*PartialOverloading=*/true);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00003973 } else {
Stephen Hines0e2c34f2015-03-23 12:09:02 -07003974 FunctionDecl *FD = nullptr;
3975 if (auto MCE = dyn_cast<MemberExpr>(NakedFn))
3976 FD = dyn_cast<FunctionDecl>(MCE->getMemberDecl());
3977 else if (auto DRE = dyn_cast<DeclRefExpr>(NakedFn))
3978 FD = dyn_cast<FunctionDecl>(DRE->getDecl());
3979 if (FD) { // We check whether it's a resolved function declaration.
3980 if (!getLangOpts().CPlusPlus ||
3981 !FD->getType()->getAs<FunctionProtoType>())
3982 Results.push_back(ResultCandidate(FD));
3983 else
3984 AddOverloadCandidate(FD, DeclAccessPair::make(FD, FD->getAccess()),
3985 Args, CandidateSet,
3986 /*SuppressUsedConversions=*/false,
3987 /*PartialOverloading=*/true);
3988
3989 } else if (auto DC = NakedFn->getType()->getAsCXXRecordDecl()) {
3990 // If expression's type is CXXRecordDecl, it may overload the function
3991 // call operator, so we check if it does and add them as candidates.
3992 // A complete type is needed to lookup for member function call operators.
3993 if (!RequireCompleteType(Loc, NakedFn->getType(), 0)) {
3994 DeclarationName OpName = Context.DeclarationNames
3995 .getCXXOperatorName(OO_Call);
3996 LookupResult R(*this, OpName, Loc, LookupOrdinaryName);
3997 LookupQualifiedName(R, DC);
3998 R.suppressDiagnostics();
3999 SmallVector<Expr *, 12> ArgExprs(1, NakedFn);
4000 ArgExprs.append(Args.begin(), Args.end());
4001 AddFunctionCandidates(R.asUnresolvedSet(), ArgExprs, CandidateSet,
4002 /*ExplicitArgs=*/nullptr,
4003 /*SuppressUsedConversions=*/false,
4004 /*PartialOverloading=*/true);
4005 }
4006 } else {
4007 // Lastly we check whether expression's type is function pointer or
4008 // function.
4009 QualType T = NakedFn->getType();
4010 if (!T->getPointeeType().isNull())
4011 T = T->getPointeeType();
4012
4013 if (auto FP = T->getAs<FunctionProtoType>()) {
4014 if (!TooManyArguments(FP->getNumParams(), Args.size(),
4015 /*PartialOverloading=*/true) ||
4016 FP->isVariadic())
4017 Results.push_back(ResultCandidate(FP));
4018 } else if (auto FT = T->getAs<FunctionType>())
4019 // No prototype and declaration, it may be a K & R style function.
4020 Results.push_back(ResultCandidate(FT));
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00004021 }
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00004022 }
Douglas Gregoref96eac2009-12-11 19:06:04 +00004023
Stephen Hines0e2c34f2015-03-23 12:09:02 -07004024 mergeCandidatesWithResults(*this, Results, CandidateSet, Loc);
4025 CodeCompleteOverloadResults(*this, S, Results, Args.size(),
4026 !CandidateSet.empty());
4027}
4028
4029void Sema::CodeCompleteConstructor(Scope *S, QualType Type, SourceLocation Loc,
4030 ArrayRef<Expr *> Args) {
4031 if (!CodeCompleter)
4032 return;
4033
4034 // A complete type is needed to lookup for constructors.
4035 if (RequireCompleteType(Loc, Type, 0))
4036 return;
4037
Pirama Arumuga Nainar3ea9e332015-04-08 08:57:32 -07004038 CXXRecordDecl *RD = Type->getAsCXXRecordDecl();
4039 if (!RD) {
4040 CodeCompleteExpression(S, Type);
4041 return;
4042 }
4043
Stephen Hines0e2c34f2015-03-23 12:09:02 -07004044 // FIXME: Provide support for member initializers.
4045 // FIXME: Provide support for variadic template constructors.
4046
4047 OverloadCandidateSet CandidateSet(Loc, OverloadCandidateSet::CSK_Normal);
4048
Pirama Arumuga Nainar3ea9e332015-04-08 08:57:32 -07004049 for (auto C : LookupConstructors(RD)) {
Stephen Hines0e2c34f2015-03-23 12:09:02 -07004050 if (auto FD = dyn_cast<FunctionDecl>(C)) {
4051 AddOverloadCandidate(FD, DeclAccessPair::make(FD, C->getAccess()),
4052 Args, CandidateSet,
4053 /*SuppressUsedConversions=*/false,
4054 /*PartialOverloading=*/true);
4055 } else if (auto FTD = dyn_cast<FunctionTemplateDecl>(C)) {
4056 AddTemplateOverloadCandidate(FTD,
4057 DeclAccessPair::make(FTD, C->getAccess()),
4058 /*ExplicitTemplateArgs=*/nullptr,
4059 Args, CandidateSet,
4060 /*SuppressUsedConversions=*/false,
4061 /*PartialOverloading=*/true);
4062 }
4063 }
4064
4065 SmallVector<ResultCandidate, 8> Results;
4066 mergeCandidatesWithResults(*this, Results, CandidateSet, Loc);
4067 CodeCompleteOverloadResults(*this, S, Results, Args.size());
Douglas Gregor9c6a0e92009-09-22 15:41:20 +00004068}
4069
John McCalld226f652010-08-21 09:40:31 +00004070void Sema::CodeCompleteInitializer(Scope *S, Decl *D) {
4071 ValueDecl *VD = dyn_cast_or_null<ValueDecl>(D);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00004072 if (!VD) {
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004073 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00004074 return;
4075 }
4076
4077 CodeCompleteExpression(S, VD->getType());
4078}
4079
4080void Sema::CodeCompleteReturn(Scope *S) {
4081 QualType ResultType;
4082 if (isa<BlockDecl>(CurContext)) {
4083 if (BlockScopeInfo *BSI = getCurBlock())
4084 ResultType = BSI->ReturnType;
4085 } else if (FunctionDecl *Function = dyn_cast<FunctionDecl>(CurContext))
Stephen Hines651f13c2014-04-23 16:59:28 -07004086 ResultType = Function->getReturnType();
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00004087 else if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(CurContext))
Stephen Hines651f13c2014-04-23 16:59:28 -07004088 ResultType = Method->getReturnType();
4089
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00004090 if (ResultType.isNull())
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004091 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00004092 else
4093 CodeCompleteExpression(S, ResultType);
4094}
4095
Douglas Gregord2d8be62011-07-30 08:36:53 +00004096void Sema::CodeCompleteAfterIf(Scope *S) {
Douglas Gregord2d8be62011-07-30 08:36:53 +00004097 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004098 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregord2d8be62011-07-30 08:36:53 +00004099 mapCodeCompletionContext(*this, PCC_Statement));
4100 Results.setFilter(&ResultBuilder::IsOrdinaryName);
4101 Results.EnterNewScope();
4102
4103 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4104 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4105 CodeCompleter->includeGlobals());
4106
4107 AddOrdinaryNameResults(PCC_Statement, S, *this, Results);
4108
4109 // "else" block
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004110 CodeCompletionBuilder Builder(Results.getAllocator(),
4111 Results.getCodeCompletionTUInfo());
Douglas Gregord2d8be62011-07-30 08:36:53 +00004112 Builder.AddTypedTextChunk("else");
Douglas Gregorf11641a2012-02-16 17:49:04 +00004113 if (Results.includeCodePatterns()) {
4114 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4115 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4116 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4117 Builder.AddPlaceholderChunk("statements");
4118 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4119 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4120 }
Douglas Gregord2d8be62011-07-30 08:36:53 +00004121 Results.AddResult(Builder.TakeString());
4122
4123 // "else if" block
4124 Builder.AddTypedTextChunk("else");
4125 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4126 Builder.AddTextChunk("if");
4127 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4128 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
David Blaikie4e4d0842012-03-11 07:00:24 +00004129 if (getLangOpts().CPlusPlus)
Douglas Gregord2d8be62011-07-30 08:36:53 +00004130 Builder.AddPlaceholderChunk("condition");
4131 else
4132 Builder.AddPlaceholderChunk("expression");
4133 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorf11641a2012-02-16 17:49:04 +00004134 if (Results.includeCodePatterns()) {
4135 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4136 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4137 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4138 Builder.AddPlaceholderChunk("statements");
4139 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
4140 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4141 }
Douglas Gregord2d8be62011-07-30 08:36:53 +00004142 Results.AddResult(Builder.TakeString());
4143
4144 Results.ExitScope();
4145
4146 if (S->getFnParent())
David Blaikie4e4d0842012-03-11 07:00:24 +00004147 AddPrettyFunctionResults(PP.getLangOpts(), Results);
Douglas Gregord2d8be62011-07-30 08:36:53 +00004148
4149 if (CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00004150 AddMacroResults(PP, Results, false);
Douglas Gregord2d8be62011-07-30 08:36:53 +00004151
4152 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4153 Results.data(),Results.size());
4154}
4155
Richard Trieuf81e5a92011-09-09 02:00:50 +00004156void Sema::CodeCompleteAssignmentRHS(Scope *S, Expr *LHS) {
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00004157 if (LHS)
4158 CodeCompleteExpression(S, static_cast<Expr *>(LHS)->getType());
4159 else
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004160 CodeCompleteOrdinaryName(S, PCC_Expression);
Douglas Gregor5ac3bdb2010-05-30 01:49:25 +00004161}
4162
Jeffrey Yasskin9ab14542010-04-08 16:38:48 +00004163void Sema::CodeCompleteQualifiedId(Scope *S, CXXScopeSpec &SS,
Douglas Gregor81b747b2009-09-17 21:32:03 +00004164 bool EnteringContext) {
4165 if (!SS.getScopeRep() || !CodeCompleter)
4166 return;
4167
Douglas Gregor86d9a522009-09-21 16:56:56 +00004168 DeclContext *Ctx = computeDeclContext(SS, EnteringContext);
4169 if (!Ctx)
4170 return;
Douglas Gregord1cd31a2009-12-11 18:28:39 +00004171
4172 // Try to instantiate any non-dependent declaration contexts before
4173 // we look in them.
John McCall77bb1aa2010-05-01 00:40:08 +00004174 if (!isDependentScopeSpecifier(SS) && RequireCompleteDeclContext(SS, Ctx))
Douglas Gregord1cd31a2009-12-11 18:28:39 +00004175 return;
4176
Douglas Gregor218937c2011-02-01 19:23:04 +00004177 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004178 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004179 CodeCompletionContext::CCC_Name);
Douglas Gregorf6961522010-08-27 21:18:54 +00004180 Results.EnterNewScope();
Douglas Gregor52779fb2010-09-23 23:01:17 +00004181
Douglas Gregor86d9a522009-09-21 16:56:56 +00004182 // The "template" keyword can follow "::" in the grammar, but only
4183 // put it into the grammar if the nested-name-specifier is dependent.
Stephen Hines651f13c2014-04-23 16:59:28 -07004184 NestedNameSpecifier *NNS = SS.getScopeRep();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004185 if (!Results.empty() && NNS->isDependent())
Douglas Gregora4477812010-01-14 16:01:26 +00004186 Results.AddResult("template");
Douglas Gregorf6961522010-08-27 21:18:54 +00004187
4188 // Add calls to overridden virtual functions, if there are any.
4189 //
4190 // FIXME: This isn't wonderful, because we don't know whether we're actually
4191 // in a context that permits expressions. This is a general issue with
4192 // qualified-id completions.
4193 if (!EnteringContext)
4194 MaybeAddOverrideCalls(*this, Ctx, Results);
4195 Results.ExitScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004196
Douglas Gregorf6961522010-08-27 21:18:54 +00004197 CodeCompletionDeclConsumer Consumer(Results, CurContext);
4198 LookupVisibleDecls(Ctx, LookupOrdinaryName, Consumer);
4199
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004200 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor430d7a12011-07-25 17:48:11 +00004201 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004202 Results.data(),Results.size());
Douglas Gregor81b747b2009-09-17 21:32:03 +00004203}
Douglas Gregor49f40bd2009-09-18 19:03:04 +00004204
4205void Sema::CodeCompleteUsing(Scope *S) {
4206 if (!CodeCompleter)
4207 return;
4208
Douglas Gregor218937c2011-02-01 19:23:04 +00004209 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004210 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00004211 CodeCompletionContext::CCC_PotentiallyQualifiedName,
4212 &ResultBuilder::IsNestedNameSpecifier);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004213 Results.EnterNewScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004214
4215 // If we aren't in class scope, we could see the "namespace" keyword.
4216 if (!S->isClassScope())
John McCall0a2c5e22010-08-25 06:19:51 +00004217 Results.AddResult(CodeCompletionResult("namespace"));
Douglas Gregor86d9a522009-09-21 16:56:56 +00004218
4219 // After "using", we can see anything that would start a
4220 // nested-name-specifier.
Douglas Gregor5d2fc402010-01-14 03:27:13 +00004221 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00004222 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4223 CodeCompleter->includeGlobals());
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004224 Results.ExitScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004225
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004226 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004227 CodeCompletionContext::CCC_PotentiallyQualifiedName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004228 Results.data(),Results.size());
Douglas Gregor49f40bd2009-09-18 19:03:04 +00004229}
4230
4231void Sema::CodeCompleteUsingDirective(Scope *S) {
4232 if (!CodeCompleter)
4233 return;
4234
Douglas Gregor86d9a522009-09-21 16:56:56 +00004235 // After "using namespace", we expect to see a namespace name or namespace
4236 // alias.
Douglas Gregor218937c2011-02-01 19:23:04 +00004237 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004238 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004239 CodeCompletionContext::CCC_Namespace,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004240 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004241 Results.EnterNewScope();
Douglas Gregor5d2fc402010-01-14 03:27:13 +00004242 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00004243 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4244 CodeCompleter->includeGlobals());
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004245 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004246 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor8071e422010-08-15 06:18:01 +00004247 CodeCompletionContext::CCC_Namespace,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004248 Results.data(),Results.size());
Douglas Gregor49f40bd2009-09-18 19:03:04 +00004249}
4250
4251void Sema::CodeCompleteNamespaceDecl(Scope *S) {
4252 if (!CodeCompleter)
4253 return;
4254
Ted Kremenekf0d58612013-10-08 17:08:03 +00004255 DeclContext *Ctx = S->getEntity();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004256 if (!S->getParent())
4257 Ctx = Context.getTranslationUnitDecl();
4258
Douglas Gregor52779fb2010-09-23 23:01:17 +00004259 bool SuppressedGlobalResults
4260 = Ctx && !CodeCompleter->includeGlobals() && isa<TranslationUnitDecl>(Ctx);
4261
Douglas Gregor218937c2011-02-01 19:23:04 +00004262 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004263 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00004264 SuppressedGlobalResults
4265 ? CodeCompletionContext::CCC_Namespace
4266 : CodeCompletionContext::CCC_Other,
4267 &ResultBuilder::IsNamespace);
4268
4269 if (Ctx && Ctx->isFileContext() && !SuppressedGlobalResults) {
Douglas Gregor86d9a522009-09-21 16:56:56 +00004270 // We only want to see those namespaces that have already been defined
4271 // within this scope, because its likely that the user is creating an
4272 // extended namespace declaration. Keep track of the most recent
4273 // definition of each namespace.
4274 std::map<NamespaceDecl *, NamespaceDecl *> OrigToLatest;
4275 for (DeclContext::specific_decl_iterator<NamespaceDecl>
4276 NS(Ctx->decls_begin()), NSEnd(Ctx->decls_end());
4277 NS != NSEnd; ++NS)
David Blaikie581deb32012-06-06 20:45:41 +00004278 OrigToLatest[NS->getOriginalNamespace()] = *NS;
Douglas Gregor86d9a522009-09-21 16:56:56 +00004279
4280 // Add the most recent definition (or extended definition) of each
4281 // namespace to the list of results.
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004282 Results.EnterNewScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004283 for (std::map<NamespaceDecl *, NamespaceDecl *>::iterator
Douglas Gregorba103062012-03-27 23:34:16 +00004284 NS = OrigToLatest.begin(),
4285 NSEnd = OrigToLatest.end();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004286 NS != NSEnd; ++NS)
Douglas Gregord1f09b42013-01-31 04:52:16 +00004287 Results.AddResult(CodeCompletionResult(
Stephen Hines6bcf27b2014-05-29 04:14:42 -07004288 NS->second, Results.getBasePriority(NS->second),
4289 nullptr),
4290 CurContext, nullptr, false);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004291 Results.ExitScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004292 }
4293
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004294 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004295 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004296 Results.data(),Results.size());
Douglas Gregor49f40bd2009-09-18 19:03:04 +00004297}
4298
4299void Sema::CodeCompleteNamespaceAliasDecl(Scope *S) {
4300 if (!CodeCompleter)
4301 return;
4302
Douglas Gregor86d9a522009-09-21 16:56:56 +00004303 // After "namespace", we expect to see a namespace or alias.
Douglas Gregor218937c2011-02-01 19:23:04 +00004304 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004305 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004306 CodeCompletionContext::CCC_Namespace,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004307 &ResultBuilder::IsNamespaceOrAlias);
Douglas Gregor5d2fc402010-01-14 03:27:13 +00004308 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00004309 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4310 CodeCompleter->includeGlobals());
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004311 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004312 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004313 Results.data(),Results.size());
Douglas Gregor49f40bd2009-09-18 19:03:04 +00004314}
4315
Douglas Gregored8d3222009-09-18 20:05:18 +00004316void Sema::CodeCompleteOperatorName(Scope *S) {
4317 if (!CodeCompleter)
4318 return;
Douglas Gregor86d9a522009-09-21 16:56:56 +00004319
John McCall0a2c5e22010-08-25 06:19:51 +00004320 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00004321 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004322 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004323 CodeCompletionContext::CCC_Type,
Douglas Gregor52779fb2010-09-23 23:01:17 +00004324 &ResultBuilder::IsType);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004325 Results.EnterNewScope();
Douglas Gregored8d3222009-09-18 20:05:18 +00004326
Douglas Gregor86d9a522009-09-21 16:56:56 +00004327 // Add the names of overloadable operators.
4328#define OVERLOADED_OPERATOR(Name,Spelling,Token,Unary,Binary,MemberOnly) \
4329 if (std::strcmp(Spelling, "?")) \
Douglas Gregora4477812010-01-14 16:01:26 +00004330 Results.AddResult(Result(Spelling));
Douglas Gregor86d9a522009-09-21 16:56:56 +00004331#include "clang/Basic/OperatorKinds.def"
4332
4333 // Add any type names visible from the current scope
Douglas Gregor45bcd432010-01-14 03:21:49 +00004334 Results.allowNestedNameSpecifiers();
Douglas Gregor5d2fc402010-01-14 03:27:13 +00004335 CodeCompletionDeclConsumer Consumer(Results, CurContext);
Douglas Gregor8071e422010-08-15 06:18:01 +00004336 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
4337 CodeCompleter->includeGlobals());
Douglas Gregor86d9a522009-09-21 16:56:56 +00004338
4339 // Add any type specifiers
David Blaikie4e4d0842012-03-11 07:00:24 +00004340 AddTypeSpecifierResults(getLangOpts(), Results);
Douglas Gregor8e0a0e42009-09-22 23:31:26 +00004341 Results.ExitScope();
Douglas Gregor86d9a522009-09-21 16:56:56 +00004342
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004343 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor8071e422010-08-15 06:18:01 +00004344 CodeCompletionContext::CCC_Type,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004345 Results.data(),Results.size());
Douglas Gregored8d3222009-09-18 20:05:18 +00004346}
Douglas Gregor49f40bd2009-09-18 19:03:04 +00004347
Dmitri Gribenko572cf582013-06-23 22:58:02 +00004348void Sema::CodeCompleteConstructorInitializer(
4349 Decl *ConstructorD,
4350 ArrayRef <CXXCtorInitializer *> Initializers) {
Douglas Gregor8987b232011-09-27 23:30:47 +00004351 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregor0133f522010-08-28 00:00:50 +00004352 CXXConstructorDecl *Constructor
4353 = static_cast<CXXConstructorDecl *>(ConstructorD);
4354 if (!Constructor)
4355 return;
4356
Douglas Gregor218937c2011-02-01 19:23:04 +00004357 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004358 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00004359 CodeCompletionContext::CCC_PotentiallyQualifiedName);
Douglas Gregor0133f522010-08-28 00:00:50 +00004360 Results.EnterNewScope();
4361
4362 // Fill in any already-initialized fields or base classes.
4363 llvm::SmallPtrSet<FieldDecl *, 4> InitializedFields;
4364 llvm::SmallPtrSet<CanQualType, 4> InitializedBases;
Dmitri Gribenko572cf582013-06-23 22:58:02 +00004365 for (unsigned I = 0, E = Initializers.size(); I != E; ++I) {
Douglas Gregor0133f522010-08-28 00:00:50 +00004366 if (Initializers[I]->isBaseInitializer())
4367 InitializedBases.insert(
4368 Context.getCanonicalType(QualType(Initializers[I]->getBaseClass(), 0)));
4369 else
Francois Pichet00eb3f92010-12-04 09:14:42 +00004370 InitializedFields.insert(cast<FieldDecl>(
4371 Initializers[I]->getAnyMember()));
Douglas Gregor0133f522010-08-28 00:00:50 +00004372 }
4373
4374 // Add completions for base classes.
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004375 CodeCompletionBuilder Builder(Results.getAllocator(),
4376 Results.getCodeCompletionTUInfo());
Dmitri Gribenko572cf582013-06-23 22:58:02 +00004377 bool SawLastInitializer = Initializers.empty();
Douglas Gregor0133f522010-08-28 00:00:50 +00004378 CXXRecordDecl *ClassDecl = Constructor->getParent();
Stephen Hines651f13c2014-04-23 16:59:28 -07004379 for (const auto &Base : ClassDecl->bases()) {
Stephen Hines176edba2014-12-01 14:53:08 -08004380 if (!InitializedBases.insert(Context.getCanonicalType(Base.getType()))
4381 .second) {
Douglas Gregor0c431c82010-08-29 19:27:27 +00004382 SawLastInitializer
Dmitri Gribenko572cf582013-06-23 22:58:02 +00004383 = !Initializers.empty() &&
4384 Initializers.back()->isBaseInitializer() &&
Stephen Hines651f13c2014-04-23 16:59:28 -07004385 Context.hasSameUnqualifiedType(Base.getType(),
Dmitri Gribenko572cf582013-06-23 22:58:02 +00004386 QualType(Initializers.back()->getBaseClass(), 0));
Douglas Gregor0133f522010-08-28 00:00:50 +00004387 continue;
Douglas Gregor0c431c82010-08-29 19:27:27 +00004388 }
Douglas Gregor0133f522010-08-28 00:00:50 +00004389
Douglas Gregor218937c2011-02-01 19:23:04 +00004390 Builder.AddTypedTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00004391 Results.getAllocator().CopyString(
Stephen Hines651f13c2014-04-23 16:59:28 -07004392 Base.getType().getAsString(Policy)));
Douglas Gregor218937c2011-02-01 19:23:04 +00004393 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4394 Builder.AddPlaceholderChunk("args");
4395 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4396 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor0c431c82010-08-29 19:27:27 +00004397 SawLastInitializer? CCP_NextInitializer
4398 : CCP_MemberDeclaration));
4399 SawLastInitializer = false;
Douglas Gregor0133f522010-08-28 00:00:50 +00004400 }
4401
4402 // Add completions for virtual base classes.
Stephen Hines651f13c2014-04-23 16:59:28 -07004403 for (const auto &Base : ClassDecl->vbases()) {
Stephen Hines176edba2014-12-01 14:53:08 -08004404 if (!InitializedBases.insert(Context.getCanonicalType(Base.getType()))
4405 .second) {
Douglas Gregor0c431c82010-08-29 19:27:27 +00004406 SawLastInitializer
Dmitri Gribenko572cf582013-06-23 22:58:02 +00004407 = !Initializers.empty() &&
4408 Initializers.back()->isBaseInitializer() &&
Stephen Hines651f13c2014-04-23 16:59:28 -07004409 Context.hasSameUnqualifiedType(Base.getType(),
Dmitri Gribenko572cf582013-06-23 22:58:02 +00004410 QualType(Initializers.back()->getBaseClass(), 0));
Douglas Gregor0133f522010-08-28 00:00:50 +00004411 continue;
Douglas Gregor0c431c82010-08-29 19:27:27 +00004412 }
Douglas Gregor0133f522010-08-28 00:00:50 +00004413
Douglas Gregor218937c2011-02-01 19:23:04 +00004414 Builder.AddTypedTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00004415 Builder.getAllocator().CopyString(
Stephen Hines651f13c2014-04-23 16:59:28 -07004416 Base.getType().getAsString(Policy)));
Douglas Gregor218937c2011-02-01 19:23:04 +00004417 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4418 Builder.AddPlaceholderChunk("args");
4419 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4420 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor0c431c82010-08-29 19:27:27 +00004421 SawLastInitializer? CCP_NextInitializer
4422 : CCP_MemberDeclaration));
4423 SawLastInitializer = false;
Douglas Gregor0133f522010-08-28 00:00:50 +00004424 }
4425
4426 // Add completions for members.
Stephen Hines651f13c2014-04-23 16:59:28 -07004427 for (auto *Field : ClassDecl->fields()) {
Stephen Hines176edba2014-12-01 14:53:08 -08004428 if (!InitializedFields.insert(cast<FieldDecl>(Field->getCanonicalDecl()))
4429 .second) {
Douglas Gregor0c431c82010-08-29 19:27:27 +00004430 SawLastInitializer
Dmitri Gribenko572cf582013-06-23 22:58:02 +00004431 = !Initializers.empty() &&
4432 Initializers.back()->isAnyMemberInitializer() &&
Stephen Hines651f13c2014-04-23 16:59:28 -07004433 Initializers.back()->getAnyMember() == Field;
Douglas Gregor0133f522010-08-28 00:00:50 +00004434 continue;
Douglas Gregor0c431c82010-08-29 19:27:27 +00004435 }
Douglas Gregor0133f522010-08-28 00:00:50 +00004436
4437 if (!Field->getDeclName())
4438 continue;
4439
Douglas Gregordae68752011-02-01 22:57:45 +00004440 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00004441 Field->getIdentifier()->getName()));
4442 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4443 Builder.AddPlaceholderChunk("args");
4444 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4445 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
Douglas Gregor0c431c82010-08-29 19:27:27 +00004446 SawLastInitializer? CCP_NextInitializer
Douglas Gregora67e03f2010-09-09 21:42:20 +00004447 : CCP_MemberDeclaration,
Douglas Gregorba103062012-03-27 23:34:16 +00004448 CXCursor_MemberRef,
4449 CXAvailability_Available,
Stephen Hines651f13c2014-04-23 16:59:28 -07004450 Field));
Douglas Gregor0c431c82010-08-29 19:27:27 +00004451 SawLastInitializer = false;
Douglas Gregor0133f522010-08-28 00:00:50 +00004452 }
4453 Results.ExitScope();
4454
Douglas Gregor52779fb2010-09-23 23:01:17 +00004455 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor0133f522010-08-28 00:00:50 +00004456 Results.data(), Results.size());
4457}
4458
Douglas Gregor81f3bff2012-02-15 15:34:24 +00004459/// \brief Determine whether this scope denotes a namespace.
4460static bool isNamespaceScope(Scope *S) {
Ted Kremenekf0d58612013-10-08 17:08:03 +00004461 DeclContext *DC = S->getEntity();
Douglas Gregor81f3bff2012-02-15 15:34:24 +00004462 if (!DC)
4463 return false;
4464
4465 return DC->isFileContext();
4466}
4467
4468void Sema::CodeCompleteLambdaIntroducer(Scope *S, LambdaIntroducer &Intro,
4469 bool AfterAmpersand) {
4470 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004471 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor81f3bff2012-02-15 15:34:24 +00004472 CodeCompletionContext::CCC_Other);
4473 Results.EnterNewScope();
4474
4475 // Note what has already been captured.
4476 llvm::SmallPtrSet<IdentifierInfo *, 4> Known;
4477 bool IncludedThis = false;
Stephen Hines6bcf27b2014-05-29 04:14:42 -07004478 for (const auto &C : Intro.Captures) {
4479 if (C.Kind == LCK_This) {
Douglas Gregor81f3bff2012-02-15 15:34:24 +00004480 IncludedThis = true;
4481 continue;
4482 }
4483
Stephen Hines6bcf27b2014-05-29 04:14:42 -07004484 Known.insert(C.Id);
Douglas Gregor81f3bff2012-02-15 15:34:24 +00004485 }
4486
4487 // Look for other capturable variables.
4488 for (; S && !isNamespaceScope(S); S = S->getParent()) {
Stephen Hines651f13c2014-04-23 16:59:28 -07004489 for (const auto *D : S->decls()) {
4490 const auto *Var = dyn_cast<VarDecl>(D);
Douglas Gregor81f3bff2012-02-15 15:34:24 +00004491 if (!Var ||
4492 !Var->hasLocalStorage() ||
4493 Var->hasAttr<BlocksAttr>())
4494 continue;
4495
Stephen Hines176edba2014-12-01 14:53:08 -08004496 if (Known.insert(Var->getIdentifier()).second)
Douglas Gregord1f09b42013-01-31 04:52:16 +00004497 Results.AddResult(CodeCompletionResult(Var, CCP_LocalDeclaration),
Stephen Hines6bcf27b2014-05-29 04:14:42 -07004498 CurContext, nullptr, false);
Douglas Gregor81f3bff2012-02-15 15:34:24 +00004499 }
4500 }
4501
4502 // Add 'this', if it would be valid.
4503 if (!IncludedThis && !AfterAmpersand && Intro.Default != LCD_ByCopy)
4504 addThisCompletion(*this, Results);
4505
4506 Results.ExitScope();
4507
4508 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
4509 Results.data(), Results.size());
4510}
4511
James Dennetta40f7922012-06-14 03:11:41 +00004512/// Macro that optionally prepends an "@" to the string literal passed in via
4513/// Keyword, depending on whether NeedAt is true or false.
4514#define OBJC_AT_KEYWORD_NAME(NeedAt,Keyword) ((NeedAt)? "@" Keyword : Keyword)
4515
Douglas Gregorbca403c2010-01-13 23:51:12 +00004516static void AddObjCImplementationResults(const LangOptions &LangOpts,
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004517 ResultBuilder &Results,
4518 bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004519 typedef CodeCompletionResult Result;
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004520 // Since we have an implementation, we can end it.
James Dennetta40f7922012-06-14 03:11:41 +00004521 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004522
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004523 CodeCompletionBuilder Builder(Results.getAllocator(),
4524 Results.getCodeCompletionTUInfo());
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004525 if (LangOpts.ObjC2) {
4526 // @dynamic
James Dennetta40f7922012-06-14 03:11:41 +00004527 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"dynamic"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004528 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4529 Builder.AddPlaceholderChunk("property");
4530 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004531
4532 // @synthesize
James Dennetta40f7922012-06-14 03:11:41 +00004533 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synthesize"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004534 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4535 Builder.AddPlaceholderChunk("property");
4536 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004537 }
4538}
4539
Douglas Gregorbca403c2010-01-13 23:51:12 +00004540static void AddObjCInterfaceResults(const LangOptions &LangOpts,
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004541 ResultBuilder &Results,
4542 bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004543 typedef CodeCompletionResult Result;
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004544
4545 // Since we have an interface or protocol, we can end it.
James Dennetta40f7922012-06-14 03:11:41 +00004546 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"end")));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004547
4548 if (LangOpts.ObjC2) {
4549 // @property
James Dennetta40f7922012-06-14 03:11:41 +00004550 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"property")));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004551
4552 // @required
James Dennetta40f7922012-06-14 03:11:41 +00004553 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"required")));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004554
4555 // @optional
James Dennetta40f7922012-06-14 03:11:41 +00004556 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"optional")));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004557 }
4558}
4559
Douglas Gregorbca403c2010-01-13 23:51:12 +00004560static void AddObjCTopLevelResults(ResultBuilder &Results, bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004561 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004562 CodeCompletionBuilder Builder(Results.getAllocator(),
4563 Results.getCodeCompletionTUInfo());
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004564
4565 // @class name ;
James Dennetta40f7922012-06-14 03:11:41 +00004566 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"class"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004567 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4568 Builder.AddPlaceholderChunk("name");
4569 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004570
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004571 if (Results.includeCodePatterns()) {
4572 // @interface name
4573 // FIXME: Could introduce the whole pattern, including superclasses and
4574 // such.
James Dennetta40f7922012-06-14 03:11:41 +00004575 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"interface"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004576 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4577 Builder.AddPlaceholderChunk("class");
4578 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004579
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004580 // @protocol name
James Dennetta40f7922012-06-14 03:11:41 +00004581 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004582 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4583 Builder.AddPlaceholderChunk("protocol");
4584 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004585
4586 // @implementation name
James Dennetta40f7922012-06-14 03:11:41 +00004587 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"implementation"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004588 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4589 Builder.AddPlaceholderChunk("class");
4590 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004591 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004592
4593 // @compatibility_alias name
James Dennetta40f7922012-06-14 03:11:41 +00004594 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"compatibility_alias"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004595 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4596 Builder.AddPlaceholderChunk("alias");
4597 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4598 Builder.AddPlaceholderChunk("class");
4599 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor06898632013-03-07 23:26:24 +00004600
4601 if (Results.getSema().getLangOpts().Modules) {
4602 // @import name
4603 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "import"));
4604 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4605 Builder.AddPlaceholderChunk("module");
4606 Results.AddResult(Result(Builder.TakeString()));
4607 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004608}
4609
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004610void Sema::CodeCompleteObjCAtDirective(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004611 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004612 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004613 CodeCompletionContext::CCC_Other);
Douglas Gregorc464ae82009-12-07 09:27:33 +00004614 Results.EnterNewScope();
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004615 if (isa<ObjCImplDecl>(CurContext))
David Blaikie4e4d0842012-03-11 07:00:24 +00004616 AddObjCImplementationResults(getLangOpts(), Results, false);
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00004617 else if (CurContext->isObjCContainer())
David Blaikie4e4d0842012-03-11 07:00:24 +00004618 AddObjCInterfaceResults(getLangOpts(), Results, false);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004619 else
Douglas Gregorbca403c2010-01-13 23:51:12 +00004620 AddObjCTopLevelResults(Results, false);
Douglas Gregorc464ae82009-12-07 09:27:33 +00004621 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004622 HandleCodeCompleteResults(this, CodeCompleter,
4623 CodeCompletionContext::CCC_Other,
4624 Results.data(),Results.size());
Douglas Gregorc464ae82009-12-07 09:27:33 +00004625}
4626
Douglas Gregorbca403c2010-01-13 23:51:12 +00004627static void AddObjCExpressionResults(ResultBuilder &Results, bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004628 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004629 CodeCompletionBuilder Builder(Results.getAllocator(),
4630 Results.getCodeCompletionTUInfo());
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004631
4632 // @encode ( type-name )
Douglas Gregor8ca72082011-10-18 21:20:17 +00004633 const char *EncodeType = "char[]";
David Blaikie4e4d0842012-03-11 07:00:24 +00004634 if (Results.getSema().getLangOpts().CPlusPlus ||
4635 Results.getSema().getLangOpts().ConstStrings)
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004636 EncodeType = "const char[]";
Douglas Gregor8ca72082011-10-18 21:20:17 +00004637 Builder.AddResultTypeChunk(EncodeType);
James Dennetta40f7922012-06-14 03:11:41 +00004638 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"encode"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004639 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4640 Builder.AddPlaceholderChunk("type-name");
4641 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4642 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004643
4644 // @protocol ( protocol-name )
Douglas Gregor8ca72082011-10-18 21:20:17 +00004645 Builder.AddResultTypeChunk("Protocol *");
James Dennetta40f7922012-06-14 03:11:41 +00004646 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"protocol"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004647 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4648 Builder.AddPlaceholderChunk("protocol-name");
4649 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4650 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004651
4652 // @selector ( selector )
Douglas Gregor8ca72082011-10-18 21:20:17 +00004653 Builder.AddResultTypeChunk("SEL");
James Dennetta40f7922012-06-14 03:11:41 +00004654 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"selector"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004655 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4656 Builder.AddPlaceholderChunk("selector");
4657 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4658 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004659
4660 // @"string"
4661 Builder.AddResultTypeChunk("NSString *");
4662 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"\""));
4663 Builder.AddPlaceholderChunk("string");
4664 Builder.AddTextChunk("\"");
4665 Results.AddResult(Result(Builder.TakeString()));
4666
Douglas Gregor79615892012-07-17 23:24:47 +00004667 // @[objects, ...]
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004668 Builder.AddResultTypeChunk("NSArray *");
James Dennetta40f7922012-06-14 03:11:41 +00004669 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"["));
Ted Kremenekebcb57a2012-03-06 20:05:56 +00004670 Builder.AddPlaceholderChunk("objects, ...");
Ted Kremenekebcb57a2012-03-06 20:05:56 +00004671 Builder.AddChunk(CodeCompletionString::CK_RightBracket);
4672 Results.AddResult(Result(Builder.TakeString()));
4673
Douglas Gregor79615892012-07-17 23:24:47 +00004674 // @{key : object, ...}
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004675 Builder.AddResultTypeChunk("NSDictionary *");
James Dennetta40f7922012-06-14 03:11:41 +00004676 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"{"));
Ted Kremenekebcb57a2012-03-06 20:05:56 +00004677 Builder.AddPlaceholderChunk("key");
Ted Kremenekebcb57a2012-03-06 20:05:56 +00004678 Builder.AddChunk(CodeCompletionString::CK_Colon);
4679 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4680 Builder.AddPlaceholderChunk("object, ...");
Ted Kremenekebcb57a2012-03-06 20:05:56 +00004681 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4682 Results.AddResult(Result(Builder.TakeString()));
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004683
Douglas Gregor79615892012-07-17 23:24:47 +00004684 // @(expression)
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004685 Builder.AddResultTypeChunk("id");
4686 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt, "("));
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004687 Builder.AddPlaceholderChunk("expression");
Jordan Rose1f6e22d2012-06-15 18:19:56 +00004688 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4689 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004690}
4691
Douglas Gregorbca403c2010-01-13 23:51:12 +00004692static void AddObjCStatementResults(ResultBuilder &Results, bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004693 typedef CodeCompletionResult Result;
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004694 CodeCompletionBuilder Builder(Results.getAllocator(),
4695 Results.getCodeCompletionTUInfo());
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004696
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004697 if (Results.includeCodePatterns()) {
4698 // @try { statements } @catch ( declaration ) { statements } @finally
4699 // { statements }
James Dennetta40f7922012-06-14 03:11:41 +00004700 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"try"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004701 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4702 Builder.AddPlaceholderChunk("statements");
4703 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4704 Builder.AddTextChunk("@catch");
4705 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4706 Builder.AddPlaceholderChunk("parameter");
4707 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4708 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4709 Builder.AddPlaceholderChunk("statements");
4710 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4711 Builder.AddTextChunk("@finally");
4712 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4713 Builder.AddPlaceholderChunk("statements");
4714 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4715 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004716 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004717
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004718 // @throw
James Dennetta40f7922012-06-14 03:11:41 +00004719 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"throw"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004720 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4721 Builder.AddPlaceholderChunk("expression");
4722 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004723
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004724 if (Results.includeCodePatterns()) {
4725 // @synchronized ( expression ) { statements }
James Dennetta40f7922012-06-14 03:11:41 +00004726 Builder.AddTypedTextChunk(OBJC_AT_KEYWORD_NAME(NeedAt,"synchronized"));
Douglas Gregor218937c2011-02-01 19:23:04 +00004727 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
4728 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
4729 Builder.AddPlaceholderChunk("expression");
4730 Builder.AddChunk(CodeCompletionString::CK_RightParen);
4731 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
4732 Builder.AddPlaceholderChunk("statements");
4733 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
4734 Results.AddResult(Result(Builder.TakeString()));
Douglas Gregorc8bddde2010-05-28 00:22:41 +00004735 }
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004736}
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004737
Douglas Gregorbca403c2010-01-13 23:51:12 +00004738static void AddObjCVisibilityResults(const LangOptions &LangOpts,
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004739 ResultBuilder &Results,
4740 bool NeedAt) {
John McCall0a2c5e22010-08-25 06:19:51 +00004741 typedef CodeCompletionResult Result;
James Dennetta40f7922012-06-14 03:11:41 +00004742 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"private")));
4743 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"protected")));
4744 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"public")));
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004745 if (LangOpts.ObjC2)
James Dennetta40f7922012-06-14 03:11:41 +00004746 Results.AddResult(Result(OBJC_AT_KEYWORD_NAME(NeedAt,"package")));
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004747}
4748
4749void Sema::CodeCompleteObjCAtVisibility(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004750 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004751 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004752 CodeCompletionContext::CCC_Other);
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004753 Results.EnterNewScope();
David Blaikie4e4d0842012-03-11 07:00:24 +00004754 AddObjCVisibilityResults(getLangOpts(), Results, false);
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004755 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004756 HandleCodeCompleteResults(this, CodeCompleter,
4757 CodeCompletionContext::CCC_Other,
4758 Results.data(),Results.size());
Douglas Gregorc38c3e12010-01-13 21:54:15 +00004759}
4760
4761void Sema::CodeCompleteObjCAtStatement(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004762 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004763 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004764 CodeCompletionContext::CCC_Other);
Douglas Gregorb6ac2452010-01-13 21:24:21 +00004765 Results.EnterNewScope();
Douglas Gregorbca403c2010-01-13 23:51:12 +00004766 AddObjCStatementResults(Results, false);
4767 AddObjCExpressionResults(Results, false);
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004768 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004769 HandleCodeCompleteResults(this, CodeCompleter,
4770 CodeCompletionContext::CCC_Other,
4771 Results.data(),Results.size());
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004772}
4773
4774void Sema::CodeCompleteObjCAtExpression(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00004775 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004776 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004777 CodeCompletionContext::CCC_Other);
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004778 Results.EnterNewScope();
Douglas Gregorbca403c2010-01-13 23:51:12 +00004779 AddObjCExpressionResults(Results, false);
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004780 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004781 HandleCodeCompleteResults(this, CodeCompleter,
4782 CodeCompletionContext::CCC_Other,
4783 Results.data(),Results.size());
Douglas Gregor9a0c85e2009-12-07 09:51:25 +00004784}
4785
Douglas Gregor988358f2009-11-19 00:14:45 +00004786/// \brief Determine whether the addition of the given flag to an Objective-C
4787/// property's attributes will cause a conflict.
Bill Wendlingad017fa2012-12-20 19:22:21 +00004788static bool ObjCPropertyFlagConflicts(unsigned Attributes, unsigned NewFlag) {
Douglas Gregor988358f2009-11-19 00:14:45 +00004789 // Check if we've already added this flag.
Bill Wendlingad017fa2012-12-20 19:22:21 +00004790 if (Attributes & NewFlag)
Douglas Gregor988358f2009-11-19 00:14:45 +00004791 return true;
4792
Bill Wendlingad017fa2012-12-20 19:22:21 +00004793 Attributes |= NewFlag;
Douglas Gregor988358f2009-11-19 00:14:45 +00004794
4795 // Check for collisions with "readonly".
Bill Wendlingad017fa2012-12-20 19:22:21 +00004796 if ((Attributes & ObjCDeclSpec::DQ_PR_readonly) &&
4797 (Attributes & ObjCDeclSpec::DQ_PR_readwrite))
Douglas Gregor988358f2009-11-19 00:14:45 +00004798 return true;
4799
Jordan Rosed7403a72012-08-20 20:01:13 +00004800 // Check for more than one of { assign, copy, retain, strong, weak }.
Bill Wendlingad017fa2012-12-20 19:22:21 +00004801 unsigned AssignCopyRetMask = Attributes & (ObjCDeclSpec::DQ_PR_assign |
John McCallf85e1932011-06-15 23:02:42 +00004802 ObjCDeclSpec::DQ_PR_unsafe_unretained |
Douglas Gregor988358f2009-11-19 00:14:45 +00004803 ObjCDeclSpec::DQ_PR_copy |
Jordan Rosed7403a72012-08-20 20:01:13 +00004804 ObjCDeclSpec::DQ_PR_retain |
4805 ObjCDeclSpec::DQ_PR_strong |
4806 ObjCDeclSpec::DQ_PR_weak);
Douglas Gregor988358f2009-11-19 00:14:45 +00004807 if (AssignCopyRetMask &&
4808 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_assign &&
John McCallf85e1932011-06-15 23:02:42 +00004809 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_unsafe_unretained &&
Douglas Gregor988358f2009-11-19 00:14:45 +00004810 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_copy &&
John McCallf85e1932011-06-15 23:02:42 +00004811 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_retain &&
Jordan Rosed7403a72012-08-20 20:01:13 +00004812 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_strong &&
4813 AssignCopyRetMask != ObjCDeclSpec::DQ_PR_weak)
Douglas Gregor988358f2009-11-19 00:14:45 +00004814 return true;
4815
4816 return false;
4817}
4818
Douglas Gregora93b1082009-11-18 23:08:07 +00004819void Sema::CodeCompleteObjCPropertyFlags(Scope *S, ObjCDeclSpec &ODS) {
Steve Naroffece8e712009-10-08 21:55:05 +00004820 if (!CodeCompleter)
4821 return;
Douglas Gregord3c68542009-11-19 01:08:35 +00004822
Bill Wendlingad017fa2012-12-20 19:22:21 +00004823 unsigned Attributes = ODS.getPropertyAttributes();
Steve Naroffece8e712009-10-08 21:55:05 +00004824
Douglas Gregor218937c2011-02-01 19:23:04 +00004825 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004826 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00004827 CodeCompletionContext::CCC_Other);
Steve Naroffece8e712009-10-08 21:55:05 +00004828 Results.EnterNewScope();
Bill Wendlingad017fa2012-12-20 19:22:21 +00004829 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readonly))
John McCall0a2c5e22010-08-25 06:19:51 +00004830 Results.AddResult(CodeCompletionResult("readonly"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004831 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_assign))
John McCall0a2c5e22010-08-25 06:19:51 +00004832 Results.AddResult(CodeCompletionResult("assign"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004833 if (!ObjCPropertyFlagConflicts(Attributes,
John McCallf85e1932011-06-15 23:02:42 +00004834 ObjCDeclSpec::DQ_PR_unsafe_unretained))
4835 Results.AddResult(CodeCompletionResult("unsafe_unretained"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004836 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_readwrite))
John McCall0a2c5e22010-08-25 06:19:51 +00004837 Results.AddResult(CodeCompletionResult("readwrite"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004838 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_retain))
John McCall0a2c5e22010-08-25 06:19:51 +00004839 Results.AddResult(CodeCompletionResult("retain"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004840 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_strong))
John McCallf85e1932011-06-15 23:02:42 +00004841 Results.AddResult(CodeCompletionResult("strong"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004842 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_copy))
John McCall0a2c5e22010-08-25 06:19:51 +00004843 Results.AddResult(CodeCompletionResult("copy"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004844 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_nonatomic))
John McCall0a2c5e22010-08-25 06:19:51 +00004845 Results.AddResult(CodeCompletionResult("nonatomic"));
Bill Wendlingad017fa2012-12-20 19:22:21 +00004846 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_atomic))
Fariborz Jahanian27f45232011-06-11 17:14:27 +00004847 Results.AddResult(CodeCompletionResult("atomic"));
Jordan Rosed7403a72012-08-20 20:01:13 +00004848
4849 // Only suggest "weak" if we're compiling for ARC-with-weak-references or GC.
John McCall0a7dd782012-08-21 02:47:43 +00004850 if (getLangOpts().ObjCARCWeak || getLangOpts().getGC() != LangOptions::NonGC)
Bill Wendlingad017fa2012-12-20 19:22:21 +00004851 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_weak))
Jordan Rosed7403a72012-08-20 20:01:13 +00004852 Results.AddResult(CodeCompletionResult("weak"));
4853
Bill Wendlingad017fa2012-12-20 19:22:21 +00004854 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_setter)) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004855 CodeCompletionBuilder Setter(Results.getAllocator(),
4856 Results.getCodeCompletionTUInfo());
Douglas Gregor218937c2011-02-01 19:23:04 +00004857 Setter.AddTypedTextChunk("setter");
Stephen Hines651f13c2014-04-23 16:59:28 -07004858 Setter.AddTextChunk("=");
Douglas Gregor218937c2011-02-01 19:23:04 +00004859 Setter.AddPlaceholderChunk("method");
4860 Results.AddResult(CodeCompletionResult(Setter.TakeString()));
Douglas Gregor54f01612009-11-19 00:01:57 +00004861 }
Bill Wendlingad017fa2012-12-20 19:22:21 +00004862 if (!ObjCPropertyFlagConflicts(Attributes, ObjCDeclSpec::DQ_PR_getter)) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00004863 CodeCompletionBuilder Getter(Results.getAllocator(),
4864 Results.getCodeCompletionTUInfo());
Douglas Gregor218937c2011-02-01 19:23:04 +00004865 Getter.AddTypedTextChunk("getter");
Stephen Hines651f13c2014-04-23 16:59:28 -07004866 Getter.AddTextChunk("=");
Douglas Gregor218937c2011-02-01 19:23:04 +00004867 Getter.AddPlaceholderChunk("method");
4868 Results.AddResult(CodeCompletionResult(Getter.TakeString()));
Douglas Gregor54f01612009-11-19 00:01:57 +00004869 }
Steve Naroffece8e712009-10-08 21:55:05 +00004870 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00004871 HandleCodeCompleteResults(this, CodeCompleter,
4872 CodeCompletionContext::CCC_Other,
4873 Results.data(),Results.size());
Steve Naroffece8e712009-10-08 21:55:05 +00004874}
Steve Naroffc4df6d22009-11-07 02:08:14 +00004875
James Dennettde23c7e2012-06-17 05:33:25 +00004876/// \brief Describes the kind of Objective-C method that we want to find
Douglas Gregor4ad96852009-11-19 07:41:15 +00004877/// via code completion.
4878enum ObjCMethodKind {
Dmitri Gribenko49fdccb2012-06-08 23:13:42 +00004879 MK_Any, ///< Any kind of method, provided it means other specified criteria.
4880 MK_ZeroArgSelector, ///< Zero-argument (unary) selector.
4881 MK_OneArgSelector ///< One-argument selector.
Douglas Gregor4ad96852009-11-19 07:41:15 +00004882};
4883
Douglas Gregor458433d2010-08-26 15:07:07 +00004884static bool isAcceptableObjCSelector(Selector Sel,
4885 ObjCMethodKind WantKind,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00004886 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004887 bool AllowSameLength = true) {
Dmitri Gribenko050315b2013-06-16 03:47:57 +00004888 unsigned NumSelIdents = SelIdents.size();
Douglas Gregor458433d2010-08-26 15:07:07 +00004889 if (NumSelIdents > Sel.getNumArgs())
4890 return false;
4891
4892 switch (WantKind) {
4893 case MK_Any: break;
4894 case MK_ZeroArgSelector: return Sel.isUnarySelector();
4895 case MK_OneArgSelector: return Sel.getNumArgs() == 1;
4896 }
4897
Douglas Gregorcf544262010-11-17 21:36:08 +00004898 if (!AllowSameLength && NumSelIdents && NumSelIdents == Sel.getNumArgs())
4899 return false;
4900
Douglas Gregor458433d2010-08-26 15:07:07 +00004901 for (unsigned I = 0; I != NumSelIdents; ++I)
4902 if (SelIdents[I] != Sel.getIdentifierInfoForSlot(I))
4903 return false;
4904
4905 return true;
4906}
4907
Douglas Gregor4ad96852009-11-19 07:41:15 +00004908static bool isAcceptableObjCMethod(ObjCMethodDecl *Method,
4909 ObjCMethodKind WantKind,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00004910 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004911 bool AllowSameLength = true) {
Douglas Gregor458433d2010-08-26 15:07:07 +00004912 return isAcceptableObjCSelector(Method->getSelector(), WantKind, SelIdents,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00004913 AllowSameLength);
Douglas Gregor4ad96852009-11-19 07:41:15 +00004914}
Douglas Gregord36adf52010-09-16 16:06:31 +00004915
4916namespace {
4917 /// \brief A set of selectors, which is used to avoid introducing multiple
4918 /// completions with the same selector into the result set.
4919 typedef llvm::SmallPtrSet<Selector, 16> VisitedSelectorSet;
4920}
4921
Douglas Gregor36ecb042009-11-17 23:22:23 +00004922/// \brief Add all of the Objective-C methods in the given Objective-C
4923/// container to the set of results.
4924///
4925/// The container will be a class, protocol, category, or implementation of
4926/// any of the above. This mether will recurse to include methods from
4927/// the superclasses of classes along with their categories, protocols, and
4928/// implementations.
4929///
4930/// \param Container the container in which we'll look to find methods.
4931///
James Dennetta40f7922012-06-14 03:11:41 +00004932/// \param WantInstanceMethods Whether to add instance methods (only); if
4933/// false, this routine will add factory methods (only).
Douglas Gregor36ecb042009-11-17 23:22:23 +00004934///
4935/// \param CurContext the context in which we're performing the lookup that
4936/// finds methods.
4937///
Douglas Gregorcf544262010-11-17 21:36:08 +00004938/// \param AllowSameLength Whether we allow a method to be added to the list
4939/// when it has the same number of parameters as we have selector identifiers.
4940///
Douglas Gregor36ecb042009-11-17 23:22:23 +00004941/// \param Results the structure into which we'll add results.
4942static void AddObjCMethods(ObjCContainerDecl *Container,
4943 bool WantInstanceMethods,
Douglas Gregor4ad96852009-11-19 07:41:15 +00004944 ObjCMethodKind WantKind,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00004945 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregor36ecb042009-11-17 23:22:23 +00004946 DeclContext *CurContext,
Douglas Gregord36adf52010-09-16 16:06:31 +00004947 VisitedSelectorSet &Selectors,
Douglas Gregorcf544262010-11-17 21:36:08 +00004948 bool AllowSameLength,
Douglas Gregor408be5a2010-08-25 01:08:01 +00004949 ResultBuilder &Results,
4950 bool InOriginalClass = true) {
John McCall0a2c5e22010-08-25 06:19:51 +00004951 typedef CodeCompletionResult Result;
Douglas Gregorb92a4082012-06-12 13:44:08 +00004952 Container = getContainerDef(Container);
Douglas Gregor5824b802013-01-30 06:58:39 +00004953 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container);
4954 bool isRootClass = IFace && !IFace->getSuperClass();
Stephen Hines651f13c2014-04-23 16:59:28 -07004955 for (auto *M : Container->methods()) {
Douglas Gregor5824b802013-01-30 06:58:39 +00004956 // The instance methods on the root class can be messaged via the
4957 // metaclass.
4958 if (M->isInstanceMethod() == WantInstanceMethods ||
4959 (isRootClass && !WantInstanceMethods)) {
Douglas Gregord3c68542009-11-19 01:08:35 +00004960 // Check whether the selector identifiers we've been given are a
4961 // subset of the identifiers for this particular method.
Stephen Hines651f13c2014-04-23 16:59:28 -07004962 if (!isAcceptableObjCMethod(M, WantKind, SelIdents, AllowSameLength))
Douglas Gregord3c68542009-11-19 01:08:35 +00004963 continue;
Douglas Gregor4ad96852009-11-19 07:41:15 +00004964
Stephen Hines176edba2014-12-01 14:53:08 -08004965 if (!Selectors.insert(M->getSelector()).second)
Douglas Gregord36adf52010-09-16 16:06:31 +00004966 continue;
Stephen Hines6bcf27b2014-05-29 04:14:42 -07004967
4968 Result R = Result(M, Results.getBasePriority(M), nullptr);
Dmitri Gribenko050315b2013-06-16 03:47:57 +00004969 R.StartParameter = SelIdents.size();
Douglas Gregor4ad96852009-11-19 07:41:15 +00004970 R.AllParametersAreInformative = (WantKind != MK_Any);
Douglas Gregor408be5a2010-08-25 01:08:01 +00004971 if (!InOriginalClass)
4972 R.Priority += CCD_InBaseClass;
Douglas Gregord3c68542009-11-19 01:08:35 +00004973 Results.MaybeAddResult(R, CurContext);
4974 }
Douglas Gregor36ecb042009-11-17 23:22:23 +00004975 }
4976
Douglas Gregore396c7b2010-09-16 15:34:59 +00004977 // Visit the protocols of protocols.
4978 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregor5e2a1ff2012-01-01 19:29:29 +00004979 if (Protocol->hasDefinition()) {
4980 const ObjCList<ObjCProtocolDecl> &Protocols
4981 = Protocol->getReferencedProtocols();
4982 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
4983 E = Protocols.end();
4984 I != E; ++I)
4985 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00004986 CurContext, Selectors, AllowSameLength, Results, false);
Douglas Gregor5e2a1ff2012-01-01 19:29:29 +00004987 }
Douglas Gregore396c7b2010-09-16 15:34:59 +00004988 }
4989
Douglas Gregor2e5c15b2011-12-15 05:27:12 +00004990 if (!IFace || !IFace->hasDefinition())
Douglas Gregor36ecb042009-11-17 23:22:23 +00004991 return;
4992
4993 // Add methods in protocols.
Stephen Hines651f13c2014-04-23 16:59:28 -07004994 for (auto *I : IFace->protocols())
4995 AddObjCMethods(I, WantInstanceMethods, WantKind, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00004996 CurContext, Selectors, AllowSameLength, Results, false);
Douglas Gregor36ecb042009-11-17 23:22:23 +00004997
4998 // Add methods in categories.
Stephen Hines651f13c2014-04-23 16:59:28 -07004999 for (auto *CatDecl : IFace->known_categories()) {
Douglas Gregord3297242013-01-16 23:00:23 +00005000 AddObjCMethods(CatDecl, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005001 CurContext, Selectors, AllowSameLength,
Douglas Gregorcf544262010-11-17 21:36:08 +00005002 Results, InOriginalClass);
Douglas Gregor36ecb042009-11-17 23:22:23 +00005003
5004 // Add a categories protocol methods.
5005 const ObjCList<ObjCProtocolDecl> &Protocols
5006 = CatDecl->getReferencedProtocols();
5007 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
5008 E = Protocols.end();
5009 I != E; ++I)
Douglas Gregor4ad96852009-11-19 07:41:15 +00005010 AddObjCMethods(*I, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005011 CurContext, Selectors, AllowSameLength,
Douglas Gregorcf544262010-11-17 21:36:08 +00005012 Results, false);
Douglas Gregor36ecb042009-11-17 23:22:23 +00005013
5014 // Add methods in category implementations.
5015 if (ObjCCategoryImplDecl *Impl = CatDecl->getImplementation())
Douglas Gregor4ad96852009-11-19 07:41:15 +00005016 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005017 CurContext, Selectors, AllowSameLength,
Douglas Gregorcf544262010-11-17 21:36:08 +00005018 Results, InOriginalClass);
Douglas Gregor36ecb042009-11-17 23:22:23 +00005019 }
5020
5021 // Add methods in superclass.
5022 if (IFace->getSuperClass())
Douglas Gregor4ad96852009-11-19 07:41:15 +00005023 AddObjCMethods(IFace->getSuperClass(), WantInstanceMethods, WantKind,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005024 SelIdents, CurContext, Selectors,
Douglas Gregorcf544262010-11-17 21:36:08 +00005025 AllowSameLength, Results, false);
Douglas Gregor36ecb042009-11-17 23:22:23 +00005026
5027 // Add methods in our implementation, if any.
5028 if (ObjCImplementationDecl *Impl = IFace->getImplementation())
Douglas Gregor4ad96852009-11-19 07:41:15 +00005029 AddObjCMethods(Impl, WantInstanceMethods, WantKind, SelIdents,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005030 CurContext, Selectors, AllowSameLength,
Douglas Gregorcf544262010-11-17 21:36:08 +00005031 Results, InOriginalClass);
Douglas Gregor4ad96852009-11-19 07:41:15 +00005032}
5033
5034
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00005035void Sema::CodeCompleteObjCPropertyGetter(Scope *S) {
Douglas Gregor4ad96852009-11-19 07:41:15 +00005036 // Try to find the interface where getters might live.
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00005037 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregor4ad96852009-11-19 07:41:15 +00005038 if (!Class) {
5039 if (ObjCCategoryDecl *Category
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00005040 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregor4ad96852009-11-19 07:41:15 +00005041 Class = Category->getClassInterface();
5042
5043 if (!Class)
5044 return;
5045 }
5046
5047 // Find all of the potential getters.
Douglas Gregor218937c2011-02-01 19:23:04 +00005048 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005049 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005050 CodeCompletionContext::CCC_Other);
Douglas Gregor4ad96852009-11-19 07:41:15 +00005051 Results.EnterNewScope();
5052
Douglas Gregord36adf52010-09-16 16:06:31 +00005053 VisitedSelectorSet Selectors;
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005054 AddObjCMethods(Class, true, MK_ZeroArgSelector, None, CurContext, Selectors,
Douglas Gregorcf544262010-11-17 21:36:08 +00005055 /*AllowSameLength=*/true, Results);
Douglas Gregor4ad96852009-11-19 07:41:15 +00005056 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005057 HandleCodeCompleteResults(this, CodeCompleter,
5058 CodeCompletionContext::CCC_Other,
5059 Results.data(),Results.size());
Douglas Gregor4ad96852009-11-19 07:41:15 +00005060}
5061
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00005062void Sema::CodeCompleteObjCPropertySetter(Scope *S) {
Douglas Gregor4ad96852009-11-19 07:41:15 +00005063 // Try to find the interface where setters might live.
5064 ObjCInterfaceDecl *Class
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00005065 = dyn_cast_or_null<ObjCInterfaceDecl>(CurContext);
Douglas Gregor4ad96852009-11-19 07:41:15 +00005066 if (!Class) {
5067 if (ObjCCategoryDecl *Category
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00005068 = dyn_cast_or_null<ObjCCategoryDecl>(CurContext))
Douglas Gregor4ad96852009-11-19 07:41:15 +00005069 Class = Category->getClassInterface();
5070
5071 if (!Class)
5072 return;
5073 }
5074
5075 // Find all of the potential getters.
Douglas Gregor218937c2011-02-01 19:23:04 +00005076 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005077 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005078 CodeCompletionContext::CCC_Other);
Douglas Gregor4ad96852009-11-19 07:41:15 +00005079 Results.EnterNewScope();
5080
Douglas Gregord36adf52010-09-16 16:06:31 +00005081 VisitedSelectorSet Selectors;
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005082 AddObjCMethods(Class, true, MK_OneArgSelector, None, CurContext,
Douglas Gregorcf544262010-11-17 21:36:08 +00005083 Selectors, /*AllowSameLength=*/true, Results);
Douglas Gregor4ad96852009-11-19 07:41:15 +00005084
5085 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005086 HandleCodeCompleteResults(this, CodeCompleter,
5087 CodeCompletionContext::CCC_Other,
5088 Results.data(),Results.size());
Douglas Gregor36ecb042009-11-17 23:22:23 +00005089}
5090
Douglas Gregorafc45782011-02-15 22:19:42 +00005091void Sema::CodeCompleteObjCPassingType(Scope *S, ObjCDeclSpec &DS,
5092 bool IsParameter) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005093 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005094 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005095 CodeCompletionContext::CCC_Type);
Douglas Gregord32b0222010-08-24 01:06:58 +00005096 Results.EnterNewScope();
5097
5098 // Add context-sensitive, Objective-C parameter-passing keywords.
5099 bool AddedInOut = false;
5100 if ((DS.getObjCDeclQualifier() &
5101 (ObjCDeclSpec::DQ_In | ObjCDeclSpec::DQ_Inout)) == 0) {
5102 Results.AddResult("in");
5103 Results.AddResult("inout");
5104 AddedInOut = true;
5105 }
5106 if ((DS.getObjCDeclQualifier() &
5107 (ObjCDeclSpec::DQ_Out | ObjCDeclSpec::DQ_Inout)) == 0) {
5108 Results.AddResult("out");
5109 if (!AddedInOut)
5110 Results.AddResult("inout");
5111 }
5112 if ((DS.getObjCDeclQualifier() &
5113 (ObjCDeclSpec::DQ_Bycopy | ObjCDeclSpec::DQ_Byref |
5114 ObjCDeclSpec::DQ_Oneway)) == 0) {
5115 Results.AddResult("bycopy");
5116 Results.AddResult("byref");
5117 Results.AddResult("oneway");
5118 }
5119
Douglas Gregorafc45782011-02-15 22:19:42 +00005120 // If we're completing the return type of an Objective-C method and the
5121 // identifier IBAction refers to a macro, provide a completion item for
5122 // an action, e.g.,
5123 // IBAction)<#selector#>:(id)sender
5124 if (DS.getObjCDeclQualifier() == 0 && !IsParameter &&
5125 Context.Idents.get("IBAction").hasMacroDefinition()) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005126 CodeCompletionBuilder Builder(Results.getAllocator(),
5127 Results.getCodeCompletionTUInfo(),
5128 CCP_CodePattern, CXAvailability_Available);
Douglas Gregorafc45782011-02-15 22:19:42 +00005129 Builder.AddTypedTextChunk("IBAction");
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00005130 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorafc45782011-02-15 22:19:42 +00005131 Builder.AddPlaceholderChunk("selector");
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00005132 Builder.AddChunk(CodeCompletionString::CK_Colon);
5133 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregorafc45782011-02-15 22:19:42 +00005134 Builder.AddTextChunk("id");
Benjamin Kramer1eb18af2012-03-26 16:57:36 +00005135 Builder.AddChunk(CodeCompletionString::CK_RightParen);
Douglas Gregorafc45782011-02-15 22:19:42 +00005136 Builder.AddTextChunk("sender");
5137 Results.AddResult(CodeCompletionResult(Builder.TakeString()));
5138 }
Douglas Gregor31aa5772013-01-30 07:11:43 +00005139
5140 // If we're completing the return type, provide 'instancetype'.
5141 if (!IsParameter) {
5142 Results.AddResult(CodeCompletionResult("instancetype"));
5143 }
Douglas Gregorafc45782011-02-15 22:19:42 +00005144
Douglas Gregord32b0222010-08-24 01:06:58 +00005145 // Add various builtin type names and specifiers.
5146 AddOrdinaryNameResults(PCC_Type, S, *this, Results);
5147 Results.ExitScope();
5148
5149 // Add the various type names
5150 Results.setFilter(&ResultBuilder::IsOrdinaryNonValueName);
5151 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5152 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5153 CodeCompleter->includeGlobals());
5154
5155 if (CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00005156 AddMacroResults(PP, Results, false);
Douglas Gregord32b0222010-08-24 01:06:58 +00005157
5158 HandleCodeCompleteResults(this, CodeCompleter,
5159 CodeCompletionContext::CCC_Type,
5160 Results.data(), Results.size());
5161}
5162
Douglas Gregor22f56992010-04-06 19:22:33 +00005163/// \brief When we have an expression with type "id", we may assume
5164/// that it has some more-specific class type based on knowledge of
5165/// common uses of Objective-C. This routine returns that class type,
5166/// or NULL if no better result could be determined.
5167static ObjCInterfaceDecl *GetAssumedMessageSendExprType(Expr *E) {
Douglas Gregor78edf512010-09-15 16:23:04 +00005168 ObjCMessageExpr *Msg = dyn_cast_or_null<ObjCMessageExpr>(E);
Douglas Gregor22f56992010-04-06 19:22:33 +00005169 if (!Msg)
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005170 return nullptr;
Douglas Gregor22f56992010-04-06 19:22:33 +00005171
5172 Selector Sel = Msg->getSelector();
5173 if (Sel.isNull())
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005174 return nullptr;
Douglas Gregor22f56992010-04-06 19:22:33 +00005175
5176 IdentifierInfo *Id = Sel.getIdentifierInfoForSlot(0);
5177 if (!Id)
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005178 return nullptr;
Douglas Gregor22f56992010-04-06 19:22:33 +00005179
5180 ObjCMethodDecl *Method = Msg->getMethodDecl();
5181 if (!Method)
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005182 return nullptr;
Douglas Gregor22f56992010-04-06 19:22:33 +00005183
5184 // Determine the class that we're sending the message to.
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005185 ObjCInterfaceDecl *IFace = nullptr;
Douglas Gregor04badcf2010-04-21 00:45:42 +00005186 switch (Msg->getReceiverKind()) {
5187 case ObjCMessageExpr::Class:
John McCallc12c5bb2010-05-15 11:32:37 +00005188 if (const ObjCObjectType *ObjType
5189 = Msg->getClassReceiver()->getAs<ObjCObjectType>())
5190 IFace = ObjType->getInterface();
Douglas Gregor04badcf2010-04-21 00:45:42 +00005191 break;
5192
5193 case ObjCMessageExpr::Instance: {
5194 QualType T = Msg->getInstanceReceiver()->getType();
5195 if (const ObjCObjectPointerType *Ptr = T->getAs<ObjCObjectPointerType>())
5196 IFace = Ptr->getInterfaceDecl();
5197 break;
5198 }
5199
5200 case ObjCMessageExpr::SuperInstance:
5201 case ObjCMessageExpr::SuperClass:
5202 break;
Douglas Gregor22f56992010-04-06 19:22:33 +00005203 }
5204
5205 if (!IFace)
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005206 return nullptr;
Douglas Gregor22f56992010-04-06 19:22:33 +00005207
5208 ObjCInterfaceDecl *Super = IFace->getSuperClass();
5209 if (Method->isInstanceMethod())
5210 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5211 .Case("retain", IFace)
John McCallf85e1932011-06-15 23:02:42 +00005212 .Case("strong", IFace)
Douglas Gregor22f56992010-04-06 19:22:33 +00005213 .Case("autorelease", IFace)
5214 .Case("copy", IFace)
5215 .Case("copyWithZone", IFace)
5216 .Case("mutableCopy", IFace)
5217 .Case("mutableCopyWithZone", IFace)
5218 .Case("awakeFromCoder", IFace)
5219 .Case("replacementObjectFromCoder", IFace)
5220 .Case("class", IFace)
5221 .Case("classForCoder", IFace)
5222 .Case("superclass", Super)
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005223 .Default(nullptr);
Douglas Gregor22f56992010-04-06 19:22:33 +00005224
5225 return llvm::StringSwitch<ObjCInterfaceDecl *>(Id->getName())
5226 .Case("new", IFace)
5227 .Case("alloc", IFace)
5228 .Case("allocWithZone", IFace)
5229 .Case("class", IFace)
5230 .Case("superclass", Super)
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005231 .Default(nullptr);
Douglas Gregor22f56992010-04-06 19:22:33 +00005232}
5233
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005234// Add a special completion for a message send to "super", which fills in the
5235// most likely case of forwarding all of our arguments to the superclass
5236// function.
5237///
5238/// \param S The semantic analysis object.
5239///
Dmitri Gribenko70517ca2012-08-23 17:58:28 +00005240/// \param NeedSuperKeyword Whether we need to prefix this completion with
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005241/// the "super" keyword. Otherwise, we just need to provide the arguments.
5242///
5243/// \param SelIdents The identifiers in the selector that have already been
5244/// provided as arguments for a send to "super".
5245///
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005246/// \param Results The set of results to augment.
5247///
5248/// \returns the Objective-C method declaration that would be invoked by
5249/// this "super" completion. If NULL, no completion was added.
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005250static ObjCMethodDecl *AddSuperSendCompletion(
5251 Sema &S, bool NeedSuperKeyword,
5252 ArrayRef<IdentifierInfo *> SelIdents,
5253 ResultBuilder &Results) {
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005254 ObjCMethodDecl *CurMethod = S.getCurMethodDecl();
5255 if (!CurMethod)
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005256 return nullptr;
5257
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005258 ObjCInterfaceDecl *Class = CurMethod->getClassInterface();
5259 if (!Class)
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005260 return nullptr;
5261
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005262 // Try to find a superclass method with the same selector.
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005263 ObjCMethodDecl *SuperMethod = nullptr;
Douglas Gregor78bcd912011-02-16 00:51:18 +00005264 while ((Class = Class->getSuperClass()) && !SuperMethod) {
5265 // Check in the class
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005266 SuperMethod = Class->getMethod(CurMethod->getSelector(),
5267 CurMethod->isInstanceMethod());
5268
Douglas Gregor78bcd912011-02-16 00:51:18 +00005269 // Check in categories or class extensions.
5270 if (!SuperMethod) {
Stephen Hines651f13c2014-04-23 16:59:28 -07005271 for (const auto *Cat : Class->known_categories()) {
Douglas Gregord3297242013-01-16 23:00:23 +00005272 if ((SuperMethod = Cat->getMethod(CurMethod->getSelector(),
Douglas Gregor78bcd912011-02-16 00:51:18 +00005273 CurMethod->isInstanceMethod())))
5274 break;
Douglas Gregord3297242013-01-16 23:00:23 +00005275 }
Douglas Gregor78bcd912011-02-16 00:51:18 +00005276 }
5277 }
5278
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005279 if (!SuperMethod)
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005280 return nullptr;
5281
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005282 // Check whether the superclass method has the same signature.
5283 if (CurMethod->param_size() != SuperMethod->param_size() ||
5284 CurMethod->isVariadic() != SuperMethod->isVariadic())
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005285 return nullptr;
5286
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005287 for (ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin(),
5288 CurPEnd = CurMethod->param_end(),
5289 SuperP = SuperMethod->param_begin();
5290 CurP != CurPEnd; ++CurP, ++SuperP) {
5291 // Make sure the parameter types are compatible.
5292 if (!S.Context.hasSameUnqualifiedType((*CurP)->getType(),
5293 (*SuperP)->getType()))
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005294 return nullptr;
5295
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005296 // Make sure we have a parameter name to forward!
5297 if (!(*CurP)->getIdentifier())
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005298 return nullptr;
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005299 }
5300
5301 // We have a superclass method. Now, form the send-to-super completion.
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005302 CodeCompletionBuilder Builder(Results.getAllocator(),
5303 Results.getCodeCompletionTUInfo());
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005304
5305 // Give this completion a return type.
Douglas Gregor8987b232011-09-27 23:30:47 +00005306 AddResultTypeChunk(S.Context, getCompletionPrintingPolicy(S), SuperMethod,
5307 Builder);
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005308
5309 // If we need the "super" keyword, add it (plus some spacing).
5310 if (NeedSuperKeyword) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005311 Builder.AddTypedTextChunk("super");
5312 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005313 }
5314
5315 Selector Sel = CurMethod->getSelector();
5316 if (Sel.isUnarySelector()) {
5317 if (NeedSuperKeyword)
Douglas Gregordae68752011-02-01 22:57:45 +00005318 Builder.AddTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005319 Sel.getNameForSlot(0)));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005320 else
Douglas Gregordae68752011-02-01 22:57:45 +00005321 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005322 Sel.getNameForSlot(0)));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005323 } else {
5324 ObjCMethodDecl::param_iterator CurP = CurMethod->param_begin();
5325 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I, ++CurP) {
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005326 if (I > SelIdents.size())
Douglas Gregor218937c2011-02-01 19:23:04 +00005327 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005328
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005329 if (I < SelIdents.size())
Douglas Gregor218937c2011-02-01 19:23:04 +00005330 Builder.AddInformativeChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00005331 Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005332 Sel.getNameForSlot(I) + ":"));
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005333 else if (NeedSuperKeyword || I > SelIdents.size()) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005334 Builder.AddTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00005335 Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005336 Sel.getNameForSlot(I) + ":"));
Douglas Gregordae68752011-02-01 22:57:45 +00005337 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00005338 (*CurP)->getIdentifier()->getName()));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005339 } else {
Douglas Gregor218937c2011-02-01 19:23:04 +00005340 Builder.AddTypedTextChunk(
Douglas Gregordae68752011-02-01 22:57:45 +00005341 Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005342 Sel.getNameForSlot(I) + ":"));
Douglas Gregordae68752011-02-01 22:57:45 +00005343 Builder.AddPlaceholderChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00005344 (*CurP)->getIdentifier()->getName()));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005345 }
5346 }
5347 }
5348
Douglas Gregorba103062012-03-27 23:34:16 +00005349 Results.AddResult(CodeCompletionResult(Builder.TakeString(), SuperMethod,
5350 CCP_SuperCompletion));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005351 return SuperMethod;
5352}
5353
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005354void Sema::CodeCompleteObjCMessageReceiver(Scope *S) {
John McCall0a2c5e22010-08-25 06:19:51 +00005355 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00005356 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005357 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005358 CodeCompletionContext::CCC_ObjCMessageReceiver,
Richard Smith80ad52f2013-01-02 11:42:31 +00005359 getLangOpts().CPlusPlus11
Douglas Gregor81f3bff2012-02-15 15:34:24 +00005360 ? &ResultBuilder::IsObjCMessageReceiverOrLambdaCapture
5361 : &ResultBuilder::IsObjCMessageReceiver);
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005362
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005363 CodeCompletionDeclConsumer Consumer(Results, CurContext);
5364 Results.EnterNewScope();
Douglas Gregor8071e422010-08-15 06:18:01 +00005365 LookupVisibleDecls(S, LookupOrdinaryName, Consumer,
5366 CodeCompleter->includeGlobals());
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005367
5368 // If we are in an Objective-C method inside a class that has a superclass,
5369 // add "super" as an option.
5370 if (ObjCMethodDecl *Method = getCurMethodDecl())
5371 if (ObjCInterfaceDecl *Iface = Method->getClassInterface())
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005372 if (Iface->getSuperClass()) {
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005373 Results.AddResult(Result("super"));
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005374
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005375 AddSuperSendCompletion(*this, /*NeedSuperKeyword=*/true, None, Results);
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005376 }
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005377
Richard Smith80ad52f2013-01-02 11:42:31 +00005378 if (getLangOpts().CPlusPlus11)
Douglas Gregor81f3bff2012-02-15 15:34:24 +00005379 addThisCompletion(*this, Results);
5380
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005381 Results.ExitScope();
5382
5383 if (CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00005384 AddMacroResults(PP, Results, false);
Douglas Gregorcee9ff12010-09-20 22:39:41 +00005385 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005386 Results.data(), Results.size());
Douglas Gregor8e254cf2010-05-27 23:06:34 +00005387
5388}
5389
Douglas Gregor2725ca82010-04-21 19:57:20 +00005390void Sema::CodeCompleteObjCSuperMessage(Scope *S, SourceLocation SuperLoc,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005391 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005392 bool AtArgumentExpression) {
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005393 ObjCInterfaceDecl *CDecl = nullptr;
Douglas Gregor2725ca82010-04-21 19:57:20 +00005394 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5395 // Figure out which interface we're in.
5396 CDecl = CurMethod->getClassInterface();
5397 if (!CDecl)
5398 return;
5399
5400 // Find the superclass of this class.
5401 CDecl = CDecl->getSuperClass();
5402 if (!CDecl)
5403 return;
5404
5405 if (CurMethod->isInstanceMethod()) {
5406 // We are inside an instance method, which means that the message
5407 // send [super ...] is actually calling an instance method on the
Douglas Gregor6b0656a2010-10-13 21:24:53 +00005408 // current object.
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005409 return CodeCompleteObjCInstanceMessage(S, nullptr, SelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005410 AtArgumentExpression,
Douglas Gregor6b0656a2010-10-13 21:24:53 +00005411 CDecl);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005412 }
5413
5414 // Fall through to send to the superclass in CDecl.
5415 } else {
5416 // "super" may be the name of a type or variable. Figure out which
5417 // it is.
Argyrios Kyrtzidis57f8da52013-03-14 22:56:43 +00005418 IdentifierInfo *Super = getSuperIdentifier();
Douglas Gregor2725ca82010-04-21 19:57:20 +00005419 NamedDecl *ND = LookupSingleName(S, Super, SuperLoc,
5420 LookupOrdinaryName);
5421 if ((CDecl = dyn_cast_or_null<ObjCInterfaceDecl>(ND))) {
5422 // "super" names an interface. Use it.
5423 } else if (TypeDecl *TD = dyn_cast_or_null<TypeDecl>(ND)) {
John McCallc12c5bb2010-05-15 11:32:37 +00005424 if (const ObjCObjectType *Iface
5425 = Context.getTypeDeclType(TD)->getAs<ObjCObjectType>())
5426 CDecl = Iface->getInterface();
Douglas Gregor2725ca82010-04-21 19:57:20 +00005427 } else if (ND && isa<UnresolvedUsingTypenameDecl>(ND)) {
5428 // "super" names an unresolved type; we can't be more specific.
5429 } else {
5430 // Assume that "super" names some kind of value and parse that way.
5431 CXXScopeSpec SS;
Abramo Bagnarae4b92762012-01-27 09:46:47 +00005432 SourceLocation TemplateKWLoc;
Douglas Gregor2725ca82010-04-21 19:57:20 +00005433 UnqualifiedId id;
5434 id.setIdentifier(Super, SuperLoc);
Abramo Bagnarae4b92762012-01-27 09:46:47 +00005435 ExprResult SuperExpr = ActOnIdExpression(S, SS, TemplateKWLoc, id,
5436 false, false);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005437 return CodeCompleteObjCInstanceMessage(S, (Expr *)SuperExpr.get(),
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005438 SelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005439 AtArgumentExpression);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005440 }
5441
5442 // Fall through
5443 }
5444
John McCallb3d87482010-08-24 05:47:05 +00005445 ParsedType Receiver;
Douglas Gregor2725ca82010-04-21 19:57:20 +00005446 if (CDecl)
John McCallb3d87482010-08-24 05:47:05 +00005447 Receiver = ParsedType::make(Context.getObjCInterfaceType(CDecl));
Douglas Gregor2725ca82010-04-21 19:57:20 +00005448 return CodeCompleteObjCClassMessage(S, Receiver, SelIdents,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005449 AtArgumentExpression,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005450 /*IsSuper=*/true);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005451}
5452
Douglas Gregorb9d77572010-09-21 00:03:25 +00005453/// \brief Given a set of code-completion results for the argument of a message
5454/// send, determine the preferred type (if any) for that argument expression.
5455static QualType getPreferredArgumentTypeForMessageSend(ResultBuilder &Results,
5456 unsigned NumSelIdents) {
5457 typedef CodeCompletionResult Result;
5458 ASTContext &Context = Results.getSema().Context;
5459
5460 QualType PreferredType;
5461 unsigned BestPriority = CCP_Unlikely * 2;
5462 Result *ResultsData = Results.data();
5463 for (unsigned I = 0, N = Results.size(); I != N; ++I) {
5464 Result &R = ResultsData[I];
5465 if (R.Kind == Result::RK_Declaration &&
5466 isa<ObjCMethodDecl>(R.Declaration)) {
5467 if (R.Priority <= BestPriority) {
Dmitri Gribenko89cf4252013-01-23 17:21:11 +00005468 const ObjCMethodDecl *Method = cast<ObjCMethodDecl>(R.Declaration);
Douglas Gregorb9d77572010-09-21 00:03:25 +00005469 if (NumSelIdents <= Method->param_size()) {
Stephen Hinesc568f1e2014-07-21 00:47:37 -07005470 QualType MyPreferredType = Method->parameters()[NumSelIdents - 1]
Douglas Gregorb9d77572010-09-21 00:03:25 +00005471 ->getType();
5472 if (R.Priority < BestPriority || PreferredType.isNull()) {
5473 BestPriority = R.Priority;
5474 PreferredType = MyPreferredType;
5475 } else if (!Context.hasSameUnqualifiedType(PreferredType,
5476 MyPreferredType)) {
5477 PreferredType = QualType();
5478 }
5479 }
5480 }
5481 }
5482 }
5483
5484 return PreferredType;
5485}
5486
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005487static void AddClassMessageCompletions(Sema &SemaRef, Scope *S,
5488 ParsedType Receiver,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005489 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005490 bool AtArgumentExpression,
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005491 bool IsSuper,
5492 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00005493 typedef CodeCompletionResult Result;
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005494 ObjCInterfaceDecl *CDecl = nullptr;
5495
Douglas Gregor24a069f2009-11-17 17:59:40 +00005496 // If the given name refers to an interface type, retrieve the
5497 // corresponding declaration.
Douglas Gregor2725ca82010-04-21 19:57:20 +00005498 if (Receiver) {
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005499 QualType T = SemaRef.GetTypeFromParser(Receiver, nullptr);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005500 if (!T.isNull())
John McCallc12c5bb2010-05-15 11:32:37 +00005501 if (const ObjCObjectType *Interface = T->getAs<ObjCObjectType>())
5502 CDecl = Interface->getInterface();
Douglas Gregor24a069f2009-11-17 17:59:40 +00005503 }
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005504
Douglas Gregor36ecb042009-11-17 23:22:23 +00005505 // Add all of the factory methods in this Objective-C class, its protocols,
5506 // superclasses, categories, implementation, etc.
Steve Naroffc4df6d22009-11-07 02:08:14 +00005507 Results.EnterNewScope();
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005508
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005509 // If this is a send-to-super, try to add the special "super" send
5510 // completion.
5511 if (IsSuper) {
5512 if (ObjCMethodDecl *SuperMethod
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005513 = AddSuperSendCompletion(SemaRef, false, SelIdents, Results))
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005514 Results.Ignore(SuperMethod);
5515 }
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005516
Douglas Gregor265f7492010-08-27 15:29:55 +00005517 // If we're inside an Objective-C method definition, prefer its selector to
5518 // others.
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005519 if (ObjCMethodDecl *CurMethod = SemaRef.getCurMethodDecl())
Douglas Gregor265f7492010-08-27 15:29:55 +00005520 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005521
Douglas Gregord36adf52010-09-16 16:06:31 +00005522 VisitedSelectorSet Selectors;
Douglas Gregor13438f92010-04-06 16:40:00 +00005523 if (CDecl)
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005524 AddObjCMethods(CDecl, false, MK_Any, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00005525 SemaRef.CurContext, Selectors, AtArgumentExpression,
5526 Results);
Douglas Gregor2725ca82010-04-21 19:57:20 +00005527 else {
Douglas Gregor13438f92010-04-06 16:40:00 +00005528 // We're messaging "id" as a type; provide all class/factory methods.
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005529
Douglas Gregor719770d2010-04-06 17:30:22 +00005530 // If we have an external source, load the entire class method
Sebastian Redl3c7f4132010-08-18 23:57:06 +00005531 // pool from the AST file.
Axel Naumann0ec56b72012-10-18 19:05:02 +00005532 if (SemaRef.getExternalSource()) {
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005533 for (uint32_t I = 0,
Axel Naumann0ec56b72012-10-18 19:05:02 +00005534 N = SemaRef.getExternalSource()->GetNumExternalSelectors();
John McCall76bd1f32010-06-01 09:23:16 +00005535 I != N; ++I) {
Axel Naumann0ec56b72012-10-18 19:05:02 +00005536 Selector Sel = SemaRef.getExternalSource()->GetExternalSelector(I);
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005537 if (Sel.isNull() || SemaRef.MethodPool.count(Sel))
Douglas Gregor719770d2010-04-06 17:30:22 +00005538 continue;
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005539
5540 SemaRef.ReadMethodPool(Sel);
Douglas Gregor719770d2010-04-06 17:30:22 +00005541 }
5542 }
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005543
5544 for (Sema::GlobalMethodPool::iterator M = SemaRef.MethodPool.begin(),
5545 MEnd = SemaRef.MethodPool.end();
Sebastian Redldb9d2142010-08-02 23:18:59 +00005546 M != MEnd; ++M) {
5547 for (ObjCMethodList *MethList = &M->second.second;
Stephen Hines0e2c34f2015-03-23 12:09:02 -07005548 MethList && MethList->getMethod();
Argyrios Kyrtzidis2e3d8c02013-04-17 00:08:58 +00005549 MethList = MethList->getNext()) {
Stephen Hines0e2c34f2015-03-23 12:09:02 -07005550 if (!isAcceptableObjCMethod(MethList->getMethod(), MK_Any, SelIdents))
Douglas Gregor13438f92010-04-06 16:40:00 +00005551 continue;
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005552
Stephen Hines0e2c34f2015-03-23 12:09:02 -07005553 Result R(MethList->getMethod(),
5554 Results.getBasePriority(MethList->getMethod()), nullptr);
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005555 R.StartParameter = SelIdents.size();
Douglas Gregor13438f92010-04-06 16:40:00 +00005556 R.AllParametersAreInformative = false;
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005557 Results.MaybeAddResult(R, SemaRef.CurContext);
Douglas Gregor13438f92010-04-06 16:40:00 +00005558 }
5559 }
5560 }
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005561
5562 Results.ExitScope();
5563}
Douglas Gregor13438f92010-04-06 16:40:00 +00005564
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005565void Sema::CodeCompleteObjCClassMessage(Scope *S, ParsedType Receiver,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005566 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005567 bool AtArgumentExpression,
Douglas Gregorc7b6d882010-09-16 15:14:18 +00005568 bool IsSuper) {
Douglas Gregore081a612011-07-21 01:05:26 +00005569
5570 QualType T = this->GetTypeFromParser(Receiver);
5571
Douglas Gregor218937c2011-02-01 19:23:04 +00005572 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005573 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregore081a612011-07-21 01:05:26 +00005574 CodeCompletionContext(CodeCompletionContext::CCC_ObjCClassMessage,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005575 T, SelIdents));
Douglas Gregore081a612011-07-21 01:05:26 +00005576
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005577 AddClassMessageCompletions(*this, S, Receiver, SelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005578 AtArgumentExpression, IsSuper, Results);
Douglas Gregorb9d77572010-09-21 00:03:25 +00005579
5580 // If we're actually at the argument expression (rather than prior to the
5581 // selector), we're actually performing code completion for an expression.
5582 // Determine whether we have a single, best method. If so, we can
5583 // code-complete the expression using the corresponding parameter type as
5584 // our preferred type, improving completion results.
5585 if (AtArgumentExpression) {
5586 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005587 SelIdents.size());
Douglas Gregorb9d77572010-09-21 00:03:25 +00005588 if (PreferredType.isNull())
5589 CodeCompleteOrdinaryName(S, PCC_Expression);
5590 else
5591 CodeCompleteExpression(S, PreferredType);
5592 return;
5593 }
5594
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005595 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregore081a612011-07-21 01:05:26 +00005596 Results.getCompletionContext(),
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005597 Results.data(), Results.size());
Steve Naroffc4df6d22009-11-07 02:08:14 +00005598}
5599
Richard Trieuf81e5a92011-09-09 02:00:50 +00005600void Sema::CodeCompleteObjCInstanceMessage(Scope *S, Expr *Receiver,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005601 ArrayRef<IdentifierInfo *> SelIdents,
Douglas Gregor70c5ac72010-09-20 23:34:21 +00005602 bool AtArgumentExpression,
Douglas Gregor6b0656a2010-10-13 21:24:53 +00005603 ObjCInterfaceDecl *Super) {
John McCall0a2c5e22010-08-25 06:19:51 +00005604 typedef CodeCompletionResult Result;
Steve Naroffc4df6d22009-11-07 02:08:14 +00005605
5606 Expr *RecExpr = static_cast<Expr *>(Receiver);
Steve Naroffc4df6d22009-11-07 02:08:14 +00005607
Douglas Gregor36ecb042009-11-17 23:22:23 +00005608 // If necessary, apply function/array conversion to the receiver.
5609 // C99 6.7.5.3p[7,8].
John Wiegley429bb272011-04-08 18:41:53 +00005610 if (RecExpr) {
5611 ExprResult Conv = DefaultFunctionArrayLvalueConversion(RecExpr);
5612 if (Conv.isInvalid()) // conversion failed. bail.
5613 return;
Stephen Hinesc568f1e2014-07-21 00:47:37 -07005614 RecExpr = Conv.get();
John Wiegley429bb272011-04-08 18:41:53 +00005615 }
Douglas Gregor6b0656a2010-10-13 21:24:53 +00005616 QualType ReceiverType = RecExpr? RecExpr->getType()
5617 : Super? Context.getObjCObjectPointerType(
5618 Context.getObjCInterfaceType(Super))
5619 : Context.getObjCIdType();
Steve Naroffc4df6d22009-11-07 02:08:14 +00005620
Douglas Gregorda892642010-11-08 21:12:30 +00005621 // If we're messaging an expression with type "id" or "Class", check
5622 // whether we know something special about the receiver that allows
5623 // us to assume a more-specific receiver type.
Stephen Hines651f13c2014-04-23 16:59:28 -07005624 if (ReceiverType->isObjCIdType() || ReceiverType->isObjCClassType()) {
Douglas Gregorda892642010-11-08 21:12:30 +00005625 if (ObjCInterfaceDecl *IFace = GetAssumedMessageSendExprType(RecExpr)) {
5626 if (ReceiverType->isObjCClassType())
5627 return CodeCompleteObjCClassMessage(S,
5628 ParsedType::make(Context.getObjCInterfaceType(IFace)),
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005629 SelIdents,
Douglas Gregorda892642010-11-08 21:12:30 +00005630 AtArgumentExpression, Super);
5631
5632 ReceiverType = Context.getObjCObjectPointerType(
5633 Context.getObjCInterfaceType(IFace));
5634 }
Stephen Hines651f13c2014-04-23 16:59:28 -07005635 } else if (RecExpr && getLangOpts().CPlusPlus) {
5636 ExprResult Conv = PerformContextuallyConvertToObjCPointer(RecExpr);
5637 if (Conv.isUsable()) {
Stephen Hinesc568f1e2014-07-21 00:47:37 -07005638 RecExpr = Conv.get();
Stephen Hines651f13c2014-04-23 16:59:28 -07005639 ReceiverType = RecExpr->getType();
5640 }
5641 }
Douglas Gregorda892642010-11-08 21:12:30 +00005642
Douglas Gregor36ecb042009-11-17 23:22:23 +00005643 // Build the set of methods we can see.
Douglas Gregor218937c2011-02-01 19:23:04 +00005644 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005645 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregore081a612011-07-21 01:05:26 +00005646 CodeCompletionContext(CodeCompletionContext::CCC_ObjCInstanceMessage,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005647 ReceiverType, SelIdents));
Douglas Gregore081a612011-07-21 01:05:26 +00005648
Douglas Gregor36ecb042009-11-17 23:22:23 +00005649 Results.EnterNewScope();
Douglas Gregor22f56992010-04-06 19:22:33 +00005650
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005651 // If this is a send-to-super, try to add the special "super" send
5652 // completion.
Douglas Gregor6b0656a2010-10-13 21:24:53 +00005653 if (Super) {
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005654 if (ObjCMethodDecl *SuperMethod
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005655 = AddSuperSendCompletion(*this, false, SelIdents, Results))
Douglas Gregor03d8aec2010-08-27 15:10:57 +00005656 Results.Ignore(SuperMethod);
5657 }
5658
Douglas Gregor265f7492010-08-27 15:29:55 +00005659 // If we're inside an Objective-C method definition, prefer its selector to
5660 // others.
5661 if (ObjCMethodDecl *CurMethod = getCurMethodDecl())
5662 Results.setPreferredSelector(CurMethod->getSelector());
Douglas Gregor36ecb042009-11-17 23:22:23 +00005663
Douglas Gregord36adf52010-09-16 16:06:31 +00005664 // Keep track of the selectors we've already added.
5665 VisitedSelectorSet Selectors;
5666
Douglas Gregorf74a4192009-11-18 00:06:18 +00005667 // Handle messages to Class. This really isn't a message to an instance
5668 // method, so we treat it the same way we would treat a message send to a
5669 // class method.
5670 if (ReceiverType->isObjCClassType() ||
5671 ReceiverType->isObjCQualifiedClassType()) {
5672 if (ObjCMethodDecl *CurMethod = getCurMethodDecl()) {
5673 if (ObjCInterfaceDecl *ClassDecl = CurMethod->getClassInterface())
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005674 AddObjCMethods(ClassDecl, false, MK_Any, SelIdents,
Douglas Gregorcf544262010-11-17 21:36:08 +00005675 CurContext, Selectors, AtArgumentExpression, Results);
Douglas Gregorf74a4192009-11-18 00:06:18 +00005676 }
5677 }
5678 // Handle messages to a qualified ID ("id<foo>").
5679 else if (const ObjCObjectPointerType *QualID
5680 = ReceiverType->getAsObjCQualifiedIdType()) {
5681 // Search protocols for instance methods.
Stephen Hines651f13c2014-04-23 16:59:28 -07005682 for (auto *I : QualID->quals())
5683 AddObjCMethods(I, true, MK_Any, SelIdents, CurContext,
Douglas Gregorcf544262010-11-17 21:36:08 +00005684 Selectors, AtArgumentExpression, Results);
Douglas Gregorf74a4192009-11-18 00:06:18 +00005685 }
5686 // Handle messages to a pointer to interface type.
5687 else if (const ObjCObjectPointerType *IFacePtr
5688 = ReceiverType->getAsObjCInterfacePointerType()) {
5689 // Search the class, its superclasses, etc., for instance methods.
Douglas Gregor4ad96852009-11-19 07:41:15 +00005690 AddObjCMethods(IFacePtr->getInterfaceDecl(), true, MK_Any, SelIdents,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005691 CurContext, Selectors, AtArgumentExpression,
Douglas Gregorcf544262010-11-17 21:36:08 +00005692 Results);
Douglas Gregorf74a4192009-11-18 00:06:18 +00005693
5694 // Search protocols for instance methods.
Stephen Hines651f13c2014-04-23 16:59:28 -07005695 for (auto *I : IFacePtr->quals())
5696 AddObjCMethods(I, true, MK_Any, SelIdents, CurContext,
Douglas Gregorcf544262010-11-17 21:36:08 +00005697 Selectors, AtArgumentExpression, Results);
Douglas Gregorf74a4192009-11-18 00:06:18 +00005698 }
Douglas Gregor13438f92010-04-06 16:40:00 +00005699 // Handle messages to "id".
5700 else if (ReceiverType->isObjCIdType()) {
Douglas Gregor719770d2010-04-06 17:30:22 +00005701 // We're messaging "id", so provide all instance methods we know
5702 // about as code-completion results.
5703
5704 // If we have an external source, load the entire class method
Sebastian Redl3c7f4132010-08-18 23:57:06 +00005705 // pool from the AST file.
Douglas Gregor719770d2010-04-06 17:30:22 +00005706 if (ExternalSource) {
John McCall76bd1f32010-06-01 09:23:16 +00005707 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5708 I != N; ++I) {
5709 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redldb9d2142010-08-02 23:18:59 +00005710 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregor719770d2010-04-06 17:30:22 +00005711 continue;
5712
Sebastian Redldb9d2142010-08-02 23:18:59 +00005713 ReadMethodPool(Sel);
Douglas Gregor719770d2010-04-06 17:30:22 +00005714 }
5715 }
5716
Sebastian Redldb9d2142010-08-02 23:18:59 +00005717 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5718 MEnd = MethodPool.end();
5719 M != MEnd; ++M) {
5720 for (ObjCMethodList *MethList = &M->second.first;
Stephen Hines0e2c34f2015-03-23 12:09:02 -07005721 MethList && MethList->getMethod();
Argyrios Kyrtzidis2e3d8c02013-04-17 00:08:58 +00005722 MethList = MethList->getNext()) {
Stephen Hines0e2c34f2015-03-23 12:09:02 -07005723 if (!isAcceptableObjCMethod(MethList->getMethod(), MK_Any, SelIdents))
Douglas Gregor13438f92010-04-06 16:40:00 +00005724 continue;
Douglas Gregord36adf52010-09-16 16:06:31 +00005725
Stephen Hines0e2c34f2015-03-23 12:09:02 -07005726 if (!Selectors.insert(MethList->getMethod()->getSelector()).second)
Douglas Gregord36adf52010-09-16 16:06:31 +00005727 continue;
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005728
Stephen Hines0e2c34f2015-03-23 12:09:02 -07005729 Result R(MethList->getMethod(),
5730 Results.getBasePriority(MethList->getMethod()), nullptr);
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005731 R.StartParameter = SelIdents.size();
Douglas Gregor13438f92010-04-06 16:40:00 +00005732 R.AllParametersAreInformative = false;
5733 Results.MaybeAddResult(R, CurContext);
5734 }
5735 }
5736 }
Steve Naroffc4df6d22009-11-07 02:08:14 +00005737 Results.ExitScope();
Douglas Gregorb9d77572010-09-21 00:03:25 +00005738
5739
5740 // If we're actually at the argument expression (rather than prior to the
5741 // selector), we're actually performing code completion for an expression.
5742 // Determine whether we have a single, best method. If so, we can
5743 // code-complete the expression using the corresponding parameter type as
5744 // our preferred type, improving completion results.
5745 if (AtArgumentExpression) {
5746 QualType PreferredType = getPreferredArgumentTypeForMessageSend(Results,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005747 SelIdents.size());
Douglas Gregorb9d77572010-09-21 00:03:25 +00005748 if (PreferredType.isNull())
5749 CodeCompleteOrdinaryName(S, PCC_Expression);
5750 else
5751 CodeCompleteExpression(S, PreferredType);
5752 return;
5753 }
5754
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005755 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregore081a612011-07-21 01:05:26 +00005756 Results.getCompletionContext(),
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005757 Results.data(),Results.size());
Steve Naroffc4df6d22009-11-07 02:08:14 +00005758}
Douglas Gregor55385fe2009-11-18 04:19:12 +00005759
Douglas Gregorfb629412010-08-23 21:17:50 +00005760void Sema::CodeCompleteObjCForCollection(Scope *S,
5761 DeclGroupPtrTy IterationVar) {
5762 CodeCompleteExpressionData Data;
5763 Data.ObjCCollection = true;
5764
5765 if (IterationVar.getAsOpaquePtr()) {
Serge Pavlov18062392013-08-27 13:15:56 +00005766 DeclGroupRef DG = IterationVar.get();
Douglas Gregorfb629412010-08-23 21:17:50 +00005767 for (DeclGroupRef::iterator I = DG.begin(), End = DG.end(); I != End; ++I) {
5768 if (*I)
5769 Data.IgnoreDecls.push_back(*I);
5770 }
5771 }
5772
5773 CodeCompleteExpression(S, Data);
5774}
5775
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005776void Sema::CodeCompleteObjCSelector(Scope *S,
5777 ArrayRef<IdentifierInfo *> SelIdents) {
Douglas Gregor458433d2010-08-26 15:07:07 +00005778 // If we have an external source, load the entire class method
5779 // pool from the AST file.
5780 if (ExternalSource) {
5781 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
5782 I != N; ++I) {
5783 Selector Sel = ExternalSource->GetExternalSelector(I);
5784 if (Sel.isNull() || MethodPool.count(Sel))
5785 continue;
5786
5787 ReadMethodPool(Sel);
5788 }
5789 }
5790
Douglas Gregor218937c2011-02-01 19:23:04 +00005791 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005792 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005793 CodeCompletionContext::CCC_SelectorName);
Douglas Gregor458433d2010-08-26 15:07:07 +00005794 Results.EnterNewScope();
5795 for (GlobalMethodPool::iterator M = MethodPool.begin(),
5796 MEnd = MethodPool.end();
5797 M != MEnd; ++M) {
5798
5799 Selector Sel = M->first;
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005800 if (!isAcceptableObjCSelector(Sel, MK_Any, SelIdents))
Douglas Gregor458433d2010-08-26 15:07:07 +00005801 continue;
5802
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005803 CodeCompletionBuilder Builder(Results.getAllocator(),
5804 Results.getCodeCompletionTUInfo());
Douglas Gregor458433d2010-08-26 15:07:07 +00005805 if (Sel.isUnarySelector()) {
Douglas Gregordae68752011-02-01 22:57:45 +00005806 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00005807 Sel.getNameForSlot(0)));
Douglas Gregor218937c2011-02-01 19:23:04 +00005808 Results.AddResult(Builder.TakeString());
Douglas Gregor458433d2010-08-26 15:07:07 +00005809 continue;
5810 }
5811
Douglas Gregor2d9e21f2010-08-26 16:46:39 +00005812 std::string Accumulator;
Douglas Gregor458433d2010-08-26 15:07:07 +00005813 for (unsigned I = 0, N = Sel.getNumArgs(); I != N; ++I) {
Dmitri Gribenko050315b2013-06-16 03:47:57 +00005814 if (I == SelIdents.size()) {
Douglas Gregor2d9e21f2010-08-26 16:46:39 +00005815 if (!Accumulator.empty()) {
Douglas Gregordae68752011-02-01 22:57:45 +00005816 Builder.AddInformativeChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00005817 Accumulator));
Douglas Gregor2d9e21f2010-08-26 16:46:39 +00005818 Accumulator.clear();
5819 }
5820 }
5821
Benjamin Kramera0651c52011-07-26 16:59:25 +00005822 Accumulator += Sel.getNameForSlot(I);
Douglas Gregor2d9e21f2010-08-26 16:46:39 +00005823 Accumulator += ':';
Douglas Gregor458433d2010-08-26 15:07:07 +00005824 }
Douglas Gregordae68752011-02-01 22:57:45 +00005825 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString( Accumulator));
Douglas Gregor218937c2011-02-01 19:23:04 +00005826 Results.AddResult(Builder.TakeString());
Douglas Gregor458433d2010-08-26 15:07:07 +00005827 }
5828 Results.ExitScope();
5829
5830 HandleCodeCompleteResults(this, CodeCompleter,
5831 CodeCompletionContext::CCC_SelectorName,
5832 Results.data(), Results.size());
5833}
5834
Douglas Gregor55385fe2009-11-18 04:19:12 +00005835/// \brief Add all of the protocol declarations that we find in the given
5836/// (translation unit) context.
5837static void AddProtocolResults(DeclContext *Ctx, DeclContext *CurContext,
Douglas Gregor083128f2009-11-18 04:49:41 +00005838 bool OnlyForwardDeclarations,
Douglas Gregor55385fe2009-11-18 04:19:12 +00005839 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00005840 typedef CodeCompletionResult Result;
Douglas Gregor55385fe2009-11-18 04:19:12 +00005841
Stephen Hines651f13c2014-04-23 16:59:28 -07005842 for (const auto *D : Ctx->decls()) {
Douglas Gregor55385fe2009-11-18 04:19:12 +00005843 // Record any protocols we find.
Stephen Hines651f13c2014-04-23 16:59:28 -07005844 if (const auto *Proto = dyn_cast<ObjCProtocolDecl>(D))
Douglas Gregor5e2a1ff2012-01-01 19:29:29 +00005845 if (!OnlyForwardDeclarations || !Proto->hasDefinition())
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005846 Results.AddResult(Result(Proto, Results.getBasePriority(Proto),nullptr),
5847 CurContext, nullptr, false);
Douglas Gregor55385fe2009-11-18 04:19:12 +00005848 }
5849}
5850
5851void Sema::CodeCompleteObjCProtocolReferences(IdentifierLocPair *Protocols,
5852 unsigned NumProtocols) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005853 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005854 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005855 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregor55385fe2009-11-18 04:19:12 +00005856
Douglas Gregor70c23352010-12-09 21:44:02 +00005857 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5858 Results.EnterNewScope();
5859
5860 // Tell the result set to ignore all of the protocols we have
5861 // already seen.
5862 // FIXME: This doesn't work when caching code-completion results.
5863 for (unsigned I = 0; I != NumProtocols; ++I)
5864 if (ObjCProtocolDecl *Protocol = LookupProtocol(Protocols[I].first,
5865 Protocols[I].second))
5866 Results.Ignore(Protocol);
Douglas Gregor55385fe2009-11-18 04:19:12 +00005867
Douglas Gregor70c23352010-12-09 21:44:02 +00005868 // Add all protocols.
5869 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, false,
5870 Results);
Douglas Gregor083128f2009-11-18 04:49:41 +00005871
Douglas Gregor70c23352010-12-09 21:44:02 +00005872 Results.ExitScope();
5873 }
5874
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005875 HandleCodeCompleteResults(this, CodeCompleter,
5876 CodeCompletionContext::CCC_ObjCProtocolName,
5877 Results.data(),Results.size());
Douglas Gregor083128f2009-11-18 04:49:41 +00005878}
5879
5880void Sema::CodeCompleteObjCProtocolDecl(Scope *) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005881 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005882 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005883 CodeCompletionContext::CCC_ObjCProtocolName);
Douglas Gregor083128f2009-11-18 04:49:41 +00005884
Douglas Gregor70c23352010-12-09 21:44:02 +00005885 if (CodeCompleter && CodeCompleter->includeGlobals()) {
5886 Results.EnterNewScope();
5887
5888 // Add all protocols.
5889 AddProtocolResults(Context.getTranslationUnitDecl(), CurContext, true,
5890 Results);
Douglas Gregor55385fe2009-11-18 04:19:12 +00005891
Douglas Gregor70c23352010-12-09 21:44:02 +00005892 Results.ExitScope();
5893 }
5894
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005895 HandleCodeCompleteResults(this, CodeCompleter,
5896 CodeCompletionContext::CCC_ObjCProtocolName,
5897 Results.data(),Results.size());
Douglas Gregor55385fe2009-11-18 04:19:12 +00005898}
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005899
5900/// \brief Add all of the Objective-C interface declarations that we find in
5901/// the given (translation unit) context.
5902static void AddInterfaceResults(DeclContext *Ctx, DeclContext *CurContext,
5903 bool OnlyForwardDeclarations,
5904 bool OnlyUnimplemented,
5905 ResultBuilder &Results) {
John McCall0a2c5e22010-08-25 06:19:51 +00005906 typedef CodeCompletionResult Result;
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005907
Stephen Hines651f13c2014-04-23 16:59:28 -07005908 for (const auto *D : Ctx->decls()) {
Douglas Gregordeacbdc2010-08-11 12:19:30 +00005909 // Record any interfaces we find.
Stephen Hines651f13c2014-04-23 16:59:28 -07005910 if (const auto *Class = dyn_cast<ObjCInterfaceDecl>(D))
Douglas Gregor7723fec2011-12-15 20:29:51 +00005911 if ((!OnlyForwardDeclarations || !Class->hasDefinition()) &&
Douglas Gregordeacbdc2010-08-11 12:19:30 +00005912 (!OnlyUnimplemented || !Class->getImplementation()))
Stephen Hines6bcf27b2014-05-29 04:14:42 -07005913 Results.AddResult(Result(Class, Results.getBasePriority(Class),nullptr),
5914 CurContext, nullptr, false);
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005915 }
5916}
5917
5918void Sema::CodeCompleteObjCInterfaceDecl(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005919 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005920 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005921 CodeCompletionContext::CCC_Other);
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005922 Results.EnterNewScope();
5923
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005924 if (CodeCompleter->includeGlobals()) {
5925 // Add all classes.
5926 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5927 false, Results);
5928 }
5929
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005930 Results.ExitScope();
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005931
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005932 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005933 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005934 Results.data(),Results.size());
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005935}
5936
Douglas Gregorc83c6872010-04-15 22:33:43 +00005937void Sema::CodeCompleteObjCSuperclass(Scope *S, IdentifierInfo *ClassName,
5938 SourceLocation ClassNameLoc) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005939 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005940 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005941 CodeCompletionContext::CCC_ObjCInterfaceName);
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005942 Results.EnterNewScope();
5943
5944 // Make sure that we ignore the class we're currently defining.
5945 NamedDecl *CurClass
Douglas Gregorc83c6872010-04-15 22:33:43 +00005946 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005947 if (CurClass && isa<ObjCInterfaceDecl>(CurClass))
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005948 Results.Ignore(CurClass);
5949
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005950 if (CodeCompleter->includeGlobals()) {
5951 // Add all classes.
5952 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5953 false, Results);
5954 }
5955
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005956 Results.ExitScope();
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005957
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005958 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005959 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005960 Results.data(),Results.size());
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005961}
5962
5963void Sema::CodeCompleteObjCImplementationDecl(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00005964 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005965 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00005966 CodeCompletionContext::CCC_Other);
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005967 Results.EnterNewScope();
5968
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005969 if (CodeCompleter->includeGlobals()) {
5970 // Add all unimplemented classes.
5971 AddInterfaceResults(Context.getTranslationUnitDecl(), CurContext, false,
5972 true, Results);
5973 }
5974
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005975 Results.ExitScope();
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005976
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005977 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor0f91c8c2011-07-30 06:55:39 +00005978 CodeCompletionContext::CCC_ObjCInterfaceName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00005979 Results.data(),Results.size());
Douglas Gregor3b49aca2009-11-18 16:26:39 +00005980}
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005981
5982void Sema::CodeCompleteObjCInterfaceCategory(Scope *S,
Douglas Gregorc83c6872010-04-15 22:33:43 +00005983 IdentifierInfo *ClassName,
5984 SourceLocation ClassNameLoc) {
John McCall0a2c5e22010-08-25 06:19:51 +00005985 typedef CodeCompletionResult Result;
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005986
Douglas Gregor218937c2011-02-01 19:23:04 +00005987 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00005988 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor3da626b2011-07-07 16:03:39 +00005989 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00005990
5991 // Ignore any categories we find that have already been implemented by this
5992 // interface.
5993 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
5994 NamedDecl *CurClass
Douglas Gregorc83c6872010-04-15 22:33:43 +00005995 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregord3297242013-01-16 23:00:23 +00005996 if (ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass)){
Stephen Hines651f13c2014-04-23 16:59:28 -07005997 for (const auto *Cat : Class->visible_categories())
Douglas Gregord3297242013-01-16 23:00:23 +00005998 CategoryNames.insert(Cat->getIdentifier());
Douglas Gregord3297242013-01-16 23:00:23 +00005999 }
6000
Douglas Gregor33ced0b2009-11-18 19:08:43 +00006001 // Add all of the categories we know about.
6002 Results.EnterNewScope();
6003 TranslationUnitDecl *TU = Context.getTranslationUnitDecl();
Stephen Hines651f13c2014-04-23 16:59:28 -07006004 for (const auto *D : TU->decls())
6005 if (const auto *Category = dyn_cast<ObjCCategoryDecl>(D))
Stephen Hines176edba2014-12-01 14:53:08 -08006006 if (CategoryNames.insert(Category->getIdentifier()).second)
Stephen Hines6bcf27b2014-05-29 04:14:42 -07006007 Results.AddResult(Result(Category, Results.getBasePriority(Category),
6008 nullptr),
6009 CurContext, nullptr, false);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00006010 Results.ExitScope();
6011
Douglas Gregore6b1bb62010-08-11 21:23:17 +00006012 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor3da626b2011-07-07 16:03:39 +00006013 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00006014 Results.data(),Results.size());
Douglas Gregor33ced0b2009-11-18 19:08:43 +00006015}
6016
6017void Sema::CodeCompleteObjCImplementationCategory(Scope *S,
Douglas Gregorc83c6872010-04-15 22:33:43 +00006018 IdentifierInfo *ClassName,
6019 SourceLocation ClassNameLoc) {
John McCall0a2c5e22010-08-25 06:19:51 +00006020 typedef CodeCompletionResult Result;
Douglas Gregor33ced0b2009-11-18 19:08:43 +00006021
6022 // Find the corresponding interface. If we couldn't find the interface, the
6023 // program itself is ill-formed. However, we'll try to be helpful still by
6024 // providing the list of all of the categories we know about.
6025 NamedDecl *CurClass
Douglas Gregorc83c6872010-04-15 22:33:43 +00006026 = LookupSingleName(TUScope, ClassName, ClassNameLoc, LookupOrdinaryName);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00006027 ObjCInterfaceDecl *Class = dyn_cast_or_null<ObjCInterfaceDecl>(CurClass);
6028 if (!Class)
Douglas Gregorc83c6872010-04-15 22:33:43 +00006029 return CodeCompleteObjCInterfaceCategory(S, ClassName, ClassNameLoc);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00006030
Douglas Gregor218937c2011-02-01 19:23:04 +00006031 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00006032 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor3da626b2011-07-07 16:03:39 +00006033 CodeCompletionContext::CCC_ObjCCategoryName);
Douglas Gregor33ced0b2009-11-18 19:08:43 +00006034
6035 // Add all of the categories that have have corresponding interface
6036 // declarations in this class and any of its superclasses, except for
6037 // already-implemented categories in the class itself.
6038 llvm::SmallPtrSet<IdentifierInfo *, 16> CategoryNames;
6039 Results.EnterNewScope();
6040 bool IgnoreImplemented = true;
6041 while (Class) {
Stephen Hines651f13c2014-04-23 16:59:28 -07006042 for (const auto *Cat : Class->visible_categories()) {
Douglas Gregord3297242013-01-16 23:00:23 +00006043 if ((!IgnoreImplemented || !Cat->getImplementation()) &&
Stephen Hines176edba2014-12-01 14:53:08 -08006044 CategoryNames.insert(Cat->getIdentifier()).second)
Stephen Hines6bcf27b2014-05-29 04:14:42 -07006045 Results.AddResult(Result(Cat, Results.getBasePriority(Cat), nullptr),
6046 CurContext, nullptr, false);
Douglas Gregord3297242013-01-16 23:00:23 +00006047 }
Douglas Gregor33ced0b2009-11-18 19:08:43 +00006048
6049 Class = Class->getSuperClass();
6050 IgnoreImplemented = false;
6051 }
6052 Results.ExitScope();
6053
Douglas Gregore6b1bb62010-08-11 21:23:17 +00006054 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor3da626b2011-07-07 16:03:39 +00006055 CodeCompletionContext::CCC_ObjCCategoryName,
Douglas Gregore6b1bb62010-08-11 21:23:17 +00006056 Results.data(),Results.size());
Douglas Gregor33ced0b2009-11-18 19:08:43 +00006057}
Douglas Gregor322328b2009-11-18 22:32:06 +00006058
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00006059void Sema::CodeCompleteObjCPropertyDefinition(Scope *S) {
Douglas Gregor218937c2011-02-01 19:23:04 +00006060 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00006061 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00006062 CodeCompletionContext::CCC_Other);
Douglas Gregor322328b2009-11-18 22:32:06 +00006063
6064 // Figure out where this @synthesize lives.
6065 ObjCContainerDecl *Container
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00006066 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor322328b2009-11-18 22:32:06 +00006067 if (!Container ||
6068 (!isa<ObjCImplementationDecl>(Container) &&
6069 !isa<ObjCCategoryImplDecl>(Container)))
6070 return;
6071
6072 // Ignore any properties that have already been implemented.
Douglas Gregorb92a4082012-06-12 13:44:08 +00006073 Container = getContainerDef(Container);
Stephen Hines651f13c2014-04-23 16:59:28 -07006074 for (const auto *D : Container->decls())
6075 if (const auto *PropertyImpl = dyn_cast<ObjCPropertyImplDecl>(D))
Douglas Gregor322328b2009-11-18 22:32:06 +00006076 Results.Ignore(PropertyImpl->getPropertyDecl());
6077
6078 // Add any properties that we find.
Douglas Gregor73449212010-12-09 23:01:55 +00006079 AddedPropertiesSet AddedProperties;
Douglas Gregor322328b2009-11-18 22:32:06 +00006080 Results.EnterNewScope();
6081 if (ObjCImplementationDecl *ClassImpl
6082 = dyn_cast<ObjCImplementationDecl>(Container))
Douglas Gregor4b81cde2011-05-05 15:50:42 +00006083 AddObjCProperties(ClassImpl->getClassInterface(), false,
6084 /*AllowNullaryMethods=*/false, CurContext,
Douglas Gregor73449212010-12-09 23:01:55 +00006085 AddedProperties, Results);
Douglas Gregor322328b2009-11-18 22:32:06 +00006086 else
6087 AddObjCProperties(cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl(),
Douglas Gregor4b81cde2011-05-05 15:50:42 +00006088 false, /*AllowNullaryMethods=*/false, CurContext,
6089 AddedProperties, Results);
Douglas Gregor322328b2009-11-18 22:32:06 +00006090 Results.ExitScope();
6091
Douglas Gregore6b1bb62010-08-11 21:23:17 +00006092 HandleCodeCompleteResults(this, CodeCompleter,
6093 CodeCompletionContext::CCC_Other,
6094 Results.data(),Results.size());
Douglas Gregor322328b2009-11-18 22:32:06 +00006095}
6096
6097void Sema::CodeCompleteObjCPropertySynthesizeIvar(Scope *S,
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00006098 IdentifierInfo *PropertyName) {
John McCall0a2c5e22010-08-25 06:19:51 +00006099 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00006100 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00006101 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00006102 CodeCompletionContext::CCC_Other);
Douglas Gregor322328b2009-11-18 22:32:06 +00006103
6104 // Figure out where this @synthesize lives.
6105 ObjCContainerDecl *Container
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00006106 = dyn_cast_or_null<ObjCContainerDecl>(CurContext);
Douglas Gregor322328b2009-11-18 22:32:06 +00006107 if (!Container ||
6108 (!isa<ObjCImplementationDecl>(Container) &&
6109 !isa<ObjCCategoryImplDecl>(Container)))
6110 return;
6111
6112 // Figure out which interface we're looking into.
Stephen Hines6bcf27b2014-05-29 04:14:42 -07006113 ObjCInterfaceDecl *Class = nullptr;
Douglas Gregor322328b2009-11-18 22:32:06 +00006114 if (ObjCImplementationDecl *ClassImpl
6115 = dyn_cast<ObjCImplementationDecl>(Container))
6116 Class = ClassImpl->getClassInterface();
6117 else
6118 Class = cast<ObjCCategoryImplDecl>(Container)->getCategoryDecl()
6119 ->getClassInterface();
6120
Douglas Gregore8426052011-04-18 14:40:46 +00006121 // Determine the type of the property we're synthesizing.
6122 QualType PropertyType = Context.getObjCIdType();
6123 if (Class) {
6124 if (ObjCPropertyDecl *Property
6125 = Class->FindPropertyDeclaration(PropertyName)) {
6126 PropertyType
6127 = Property->getType().getNonReferenceType().getUnqualifiedType();
6128
6129 // Give preference to ivars
6130 Results.setPreferredType(PropertyType);
6131 }
6132 }
6133
Douglas Gregor322328b2009-11-18 22:32:06 +00006134 // Add all of the instance variables in this class and its superclasses.
6135 Results.EnterNewScope();
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006136 bool SawSimilarlyNamedIvar = false;
6137 std::string NameWithPrefix;
6138 NameWithPrefix += '_';
Benjamin Kramera0651c52011-07-26 16:59:25 +00006139 NameWithPrefix += PropertyName->getName();
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006140 std::string NameWithSuffix = PropertyName->getName().str();
6141 NameWithSuffix += '_';
Douglas Gregor322328b2009-11-18 22:32:06 +00006142 for(; Class; Class = Class->getSuperClass()) {
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006143 for (ObjCIvarDecl *Ivar = Class->all_declared_ivar_begin(); Ivar;
6144 Ivar = Ivar->getNextIvar()) {
Stephen Hines6bcf27b2014-05-29 04:14:42 -07006145 Results.AddResult(Result(Ivar, Results.getBasePriority(Ivar), nullptr),
6146 CurContext, nullptr, false);
6147
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006148 // Determine whether we've seen an ivar with a name similar to the
6149 // property.
Douglas Gregore8426052011-04-18 14:40:46 +00006150 if ((PropertyName == Ivar->getIdentifier() ||
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006151 NameWithPrefix == Ivar->getName() ||
Douglas Gregore8426052011-04-18 14:40:46 +00006152 NameWithSuffix == Ivar->getName())) {
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006153 SawSimilarlyNamedIvar = true;
Douglas Gregore8426052011-04-18 14:40:46 +00006154
6155 // Reduce the priority of this result by one, to give it a slight
6156 // advantage over other results whose names don't match so closely.
6157 if (Results.size() &&
6158 Results.data()[Results.size() - 1].Kind
6159 == CodeCompletionResult::RK_Declaration &&
6160 Results.data()[Results.size() - 1].Declaration == Ivar)
6161 Results.data()[Results.size() - 1].Priority--;
6162 }
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006163 }
Douglas Gregor322328b2009-11-18 22:32:06 +00006164 }
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006165
6166 if (!SawSimilarlyNamedIvar) {
6167 // Create ivar result _propName, that the user can use to synthesize
Douglas Gregore8426052011-04-18 14:40:46 +00006168 // an ivar of the appropriate type.
6169 unsigned Priority = CCP_MemberDeclaration + 1;
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006170 typedef CodeCompletionResult Result;
6171 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00006172 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo(),
6173 Priority,CXAvailability_Available);
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006174
Douglas Gregor8987b232011-09-27 23:30:47 +00006175 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregore8426052011-04-18 14:40:46 +00006176 Builder.AddResultTypeChunk(GetCompletionTypeString(PropertyType, Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00006177 Policy, Allocator));
Douglas Gregoraa490cb2011-04-18 14:13:53 +00006178 Builder.AddTypedTextChunk(Allocator.CopyString(NameWithPrefix));
6179 Results.AddResult(Result(Builder.TakeString(), Priority,
6180 CXCursor_ObjCIvarDecl));
6181 }
6182
Douglas Gregor322328b2009-11-18 22:32:06 +00006183 Results.ExitScope();
6184
Douglas Gregore6b1bb62010-08-11 21:23:17 +00006185 HandleCodeCompleteResults(this, CodeCompleter,
6186 CodeCompletionContext::CCC_Other,
6187 Results.data(),Results.size());
Douglas Gregor322328b2009-11-18 22:32:06 +00006188}
Douglas Gregore8f5a172010-04-07 00:21:17 +00006189
Douglas Gregor408be5a2010-08-25 01:08:01 +00006190// Mapping from selectors to the methods that implement that selector, along
6191// with the "in original class" flag.
Benjamin Kramere1039792013-06-29 17:52:13 +00006192typedef llvm::DenseMap<
6193 Selector, llvm::PointerIntPair<ObjCMethodDecl *, 1, bool> > KnownMethodsMap;
Douglas Gregore8f5a172010-04-07 00:21:17 +00006194
6195/// \brief Find all of the methods that reside in the given container
6196/// (and its superclasses, protocols, etc.) that meet the given
6197/// criteria. Insert those methods into the map of known methods,
6198/// indexed by selector so they can be easily found.
6199static void FindImplementableMethods(ASTContext &Context,
6200 ObjCContainerDecl *Container,
6201 bool WantInstanceMethods,
6202 QualType ReturnType,
Douglas Gregor408be5a2010-08-25 01:08:01 +00006203 KnownMethodsMap &KnownMethods,
6204 bool InOriginalClass = true) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006205 if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Container)) {
Douglas Gregorb92a4082012-06-12 13:44:08 +00006206 // Make sure we have a definition; that's what we'll walk.
Douglas Gregor2e5c15b2011-12-15 05:27:12 +00006207 if (!IFace->hasDefinition())
6208 return;
Douglas Gregorb92a4082012-06-12 13:44:08 +00006209
6210 IFace = IFace->getDefinition();
6211 Container = IFace;
Douglas Gregor2e5c15b2011-12-15 05:27:12 +00006212
Douglas Gregore8f5a172010-04-07 00:21:17 +00006213 const ObjCList<ObjCProtocolDecl> &Protocols
6214 = IFace->getReferencedProtocols();
6215 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregorea766182010-10-18 18:21:28 +00006216 E = Protocols.end();
Douglas Gregore8f5a172010-04-07 00:21:17 +00006217 I != E; ++I)
6218 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregorea766182010-10-18 18:21:28 +00006219 KnownMethods, InOriginalClass);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006220
Douglas Gregorea766182010-10-18 18:21:28 +00006221 // Add methods from any class extensions and categories.
Stephen Hines651f13c2014-04-23 16:59:28 -07006222 for (auto *Cat : IFace->visible_categories()) {
6223 FindImplementableMethods(Context, Cat, WantInstanceMethods, ReturnType,
Douglas Gregorea766182010-10-18 18:21:28 +00006224 KnownMethods, false);
Douglas Gregord3297242013-01-16 23:00:23 +00006225 }
6226
Douglas Gregorea766182010-10-18 18:21:28 +00006227 // Visit the superclass.
6228 if (IFace->getSuperClass())
6229 FindImplementableMethods(Context, IFace->getSuperClass(),
6230 WantInstanceMethods, ReturnType,
6231 KnownMethods, false);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006232 }
6233
6234 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(Container)) {
6235 // Recurse into protocols.
6236 const ObjCList<ObjCProtocolDecl> &Protocols
6237 = Category->getReferencedProtocols();
6238 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
Douglas Gregorea766182010-10-18 18:21:28 +00006239 E = Protocols.end();
Douglas Gregore8f5a172010-04-07 00:21:17 +00006240 I != E; ++I)
6241 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
Douglas Gregorea766182010-10-18 18:21:28 +00006242 KnownMethods, InOriginalClass);
6243
6244 // If this category is the original class, jump to the interface.
6245 if (InOriginalClass && Category->getClassInterface())
6246 FindImplementableMethods(Context, Category->getClassInterface(),
6247 WantInstanceMethods, ReturnType, KnownMethods,
6248 false);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006249 }
6250
6251 if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Container)) {
Douglas Gregorb92a4082012-06-12 13:44:08 +00006252 // Make sure we have a definition; that's what we'll walk.
6253 if (!Protocol->hasDefinition())
6254 return;
6255 Protocol = Protocol->getDefinition();
6256 Container = Protocol;
6257
6258 // Recurse into protocols.
6259 const ObjCList<ObjCProtocolDecl> &Protocols
6260 = Protocol->getReferencedProtocols();
6261 for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(),
6262 E = Protocols.end();
6263 I != E; ++I)
6264 FindImplementableMethods(Context, *I, WantInstanceMethods, ReturnType,
6265 KnownMethods, false);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006266 }
6267
6268 // Add methods in this container. This operation occurs last because
6269 // we want the methods from this container to override any methods
6270 // we've previously seen with the same selector.
Stephen Hines651f13c2014-04-23 16:59:28 -07006271 for (auto *M : Container->methods()) {
David Blaikie262bc182012-04-30 02:36:29 +00006272 if (M->isInstanceMethod() == WantInstanceMethods) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006273 if (!ReturnType.isNull() &&
Stephen Hines651f13c2014-04-23 16:59:28 -07006274 !Context.hasSameUnqualifiedType(ReturnType, M->getReturnType()))
Douglas Gregore8f5a172010-04-07 00:21:17 +00006275 continue;
6276
Benjamin Kramere1039792013-06-29 17:52:13 +00006277 KnownMethods[M->getSelector()] =
Stephen Hines651f13c2014-04-23 16:59:28 -07006278 KnownMethodsMap::mapped_type(M, InOriginalClass);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006279 }
6280 }
6281}
6282
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006283/// \brief Add the parenthesized return or parameter type chunk to a code
6284/// completion string.
6285static void AddObjCPassingTypeChunk(QualType Type,
Douglas Gregor90f5f472012-04-10 18:35:07 +00006286 unsigned ObjCDeclQuals,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006287 ASTContext &Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00006288 const PrintingPolicy &Policy,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006289 CodeCompletionBuilder &Builder) {
6290 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
Douglas Gregor90f5f472012-04-10 18:35:07 +00006291 std::string Quals = formatObjCParamQualifiers(ObjCDeclQuals);
6292 if (!Quals.empty())
6293 Builder.AddTextChunk(Builder.getAllocator().CopyString(Quals));
Douglas Gregor8987b232011-09-27 23:30:47 +00006294 Builder.AddTextChunk(GetCompletionTypeString(Type, Context, Policy,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006295 Builder.getAllocator()));
6296 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6297}
6298
6299/// \brief Determine whether the given class is or inherits from a class by
6300/// the given name.
6301static bool InheritsFromClassNamed(ObjCInterfaceDecl *Class,
Chris Lattner5f9e2722011-07-23 10:55:15 +00006302 StringRef Name) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006303 if (!Class)
6304 return false;
6305
6306 if (Class->getIdentifier() && Class->getIdentifier()->getName() == Name)
6307 return true;
6308
6309 return InheritsFromClassNamed(Class->getSuperClass(), Name);
6310}
6311
6312/// \brief Add code completions for Objective-C Key-Value Coding (KVC) and
6313/// Key-Value Observing (KVO).
6314static void AddObjCKeyValueCompletions(ObjCPropertyDecl *Property,
6315 bool IsInstanceMethod,
6316 QualType ReturnType,
6317 ASTContext &Context,
Douglas Gregore74c25c2011-05-04 23:50:46 +00006318 VisitedSelectorSet &KnownSelectors,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006319 ResultBuilder &Results) {
6320 IdentifierInfo *PropName = Property->getIdentifier();
6321 if (!PropName || PropName->getLength() == 0)
6322 return;
6323
Douglas Gregor8987b232011-09-27 23:30:47 +00006324 PrintingPolicy Policy = getCompletionPrintingPolicy(Results.getSema());
6325
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006326 // Builder that will create each code completion.
6327 typedef CodeCompletionResult Result;
6328 CodeCompletionAllocator &Allocator = Results.getAllocator();
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00006329 CodeCompletionBuilder Builder(Allocator, Results.getCodeCompletionTUInfo());
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006330
6331 // The selector table.
6332 SelectorTable &Selectors = Context.Selectors;
6333
6334 // The property name, copied into the code completion allocation region
6335 // on demand.
6336 struct KeyHolder {
6337 CodeCompletionAllocator &Allocator;
Chris Lattner5f9e2722011-07-23 10:55:15 +00006338 StringRef Key;
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006339 const char *CopiedKey;
Stephen Hines6bcf27b2014-05-29 04:14:42 -07006340
Chris Lattner5f9e2722011-07-23 10:55:15 +00006341 KeyHolder(CodeCompletionAllocator &Allocator, StringRef Key)
Stephen Hines6bcf27b2014-05-29 04:14:42 -07006342 : Allocator(Allocator), Key(Key), CopiedKey(nullptr) {}
6343
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006344 operator const char *() {
6345 if (CopiedKey)
6346 return CopiedKey;
6347
6348 return CopiedKey = Allocator.CopyString(Key);
6349 }
6350 } Key(Allocator, PropName->getName());
6351
6352 // The uppercased name of the property name.
6353 std::string UpperKey = PropName->getName();
6354 if (!UpperKey.empty())
Jordan Rose223f0ff2013-02-09 10:09:43 +00006355 UpperKey[0] = toUppercase(UpperKey[0]);
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006356
6357 bool ReturnTypeMatchesProperty = ReturnType.isNull() ||
6358 Context.hasSameUnqualifiedType(ReturnType.getNonReferenceType(),
6359 Property->getType());
6360 bool ReturnTypeMatchesVoid
6361 = ReturnType.isNull() || ReturnType->isVoidType();
6362
6363 // Add the normal accessor -(type)key.
6364 if (IsInstanceMethod &&
Stephen Hines176edba2014-12-01 14:53:08 -08006365 KnownSelectors.insert(Selectors.getNullarySelector(PropName)).second &&
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006366 ReturnTypeMatchesProperty && !Property->getGetterMethodDecl()) {
6367 if (ReturnType.isNull())
Douglas Gregor90f5f472012-04-10 18:35:07 +00006368 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6369 Context, Policy, Builder);
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006370
6371 Builder.AddTypedTextChunk(Key);
6372 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6373 CXCursor_ObjCInstanceMethodDecl));
6374 }
6375
6376 // If we have an integral or boolean property (or the user has provided
6377 // an integral or boolean return type), add the accessor -(type)isKey.
6378 if (IsInstanceMethod &&
6379 ((!ReturnType.isNull() &&
6380 (ReturnType->isIntegerType() || ReturnType->isBooleanType())) ||
6381 (ReturnType.isNull() &&
6382 (Property->getType()->isIntegerType() ||
6383 Property->getType()->isBooleanType())))) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006384 std::string SelectorName = (Twine("is") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006385 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Stephen Hines176edba2014-12-01 14:53:08 -08006386 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))
6387 .second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006388 if (ReturnType.isNull()) {
6389 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6390 Builder.AddTextChunk("BOOL");
6391 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6392 }
6393
6394 Builder.AddTypedTextChunk(
6395 Allocator.CopyString(SelectorId->getName()));
6396 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6397 CXCursor_ObjCInstanceMethodDecl));
6398 }
6399 }
6400
6401 // Add the normal mutator.
6402 if (IsInstanceMethod && ReturnTypeMatchesVoid &&
6403 !Property->getSetterMethodDecl()) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006404 std::string SelectorName = (Twine("set") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006405 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Stephen Hines176edba2014-12-01 14:53:08 -08006406 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId)).second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006407 if (ReturnType.isNull()) {
6408 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6409 Builder.AddTextChunk("void");
6410 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6411 }
6412
6413 Builder.AddTypedTextChunk(
6414 Allocator.CopyString(SelectorId->getName()));
6415 Builder.AddTypedTextChunk(":");
Douglas Gregor90f5f472012-04-10 18:35:07 +00006416 AddObjCPassingTypeChunk(Property->getType(), /*Quals=*/0,
6417 Context, Policy, Builder);
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006418 Builder.AddTextChunk(Key);
6419 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6420 CXCursor_ObjCInstanceMethodDecl));
6421 }
6422 }
6423
6424 // Indexed and unordered accessors
6425 unsigned IndexedGetterPriority = CCP_CodePattern;
6426 unsigned IndexedSetterPriority = CCP_CodePattern;
6427 unsigned UnorderedGetterPriority = CCP_CodePattern;
6428 unsigned UnorderedSetterPriority = CCP_CodePattern;
6429 if (const ObjCObjectPointerType *ObjCPointer
6430 = Property->getType()->getAs<ObjCObjectPointerType>()) {
6431 if (ObjCInterfaceDecl *IFace = ObjCPointer->getInterfaceDecl()) {
6432 // If this interface type is not provably derived from a known
6433 // collection, penalize the corresponding completions.
6434 if (!InheritsFromClassNamed(IFace, "NSMutableArray")) {
6435 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6436 if (!InheritsFromClassNamed(IFace, "NSArray"))
6437 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6438 }
6439
6440 if (!InheritsFromClassNamed(IFace, "NSMutableSet")) {
6441 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6442 if (!InheritsFromClassNamed(IFace, "NSSet"))
6443 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6444 }
6445 }
6446 } else {
6447 IndexedGetterPriority += CCD_ProbablyNotObjCCollection;
6448 IndexedSetterPriority += CCD_ProbablyNotObjCCollection;
6449 UnorderedGetterPriority += CCD_ProbablyNotObjCCollection;
6450 UnorderedSetterPriority += CCD_ProbablyNotObjCCollection;
6451 }
6452
6453 // Add -(NSUInteger)countOf<key>
6454 if (IsInstanceMethod &&
6455 (ReturnType.isNull() || ReturnType->isIntegerType())) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006456 std::string SelectorName = (Twine("countOf") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006457 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Stephen Hines176edba2014-12-01 14:53:08 -08006458 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))
6459 .second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006460 if (ReturnType.isNull()) {
6461 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6462 Builder.AddTextChunk("NSUInteger");
6463 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6464 }
6465
6466 Builder.AddTypedTextChunk(
6467 Allocator.CopyString(SelectorId->getName()));
6468 Results.AddResult(Result(Builder.TakeString(),
6469 std::min(IndexedGetterPriority,
6470 UnorderedGetterPriority),
6471 CXCursor_ObjCInstanceMethodDecl));
6472 }
6473 }
6474
6475 // Indexed getters
6476 // Add -(id)objectInKeyAtIndex:(NSUInteger)index
6477 if (IsInstanceMethod &&
6478 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Douglas Gregor62041592011-02-17 03:19:26 +00006479 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006480 = (Twine("objectIn") + UpperKey + "AtIndex").str();
Douglas Gregor62041592011-02-17 03:19:26 +00006481 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Stephen Hines176edba2014-12-01 14:53:08 -08006482 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId)).second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006483 if (ReturnType.isNull()) {
6484 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6485 Builder.AddTextChunk("id");
6486 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6487 }
6488
6489 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6490 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6491 Builder.AddTextChunk("NSUInteger");
6492 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6493 Builder.AddTextChunk("index");
6494 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6495 CXCursor_ObjCInstanceMethodDecl));
6496 }
6497 }
6498
6499 // Add -(NSArray *)keyAtIndexes:(NSIndexSet *)indexes
6500 if (IsInstanceMethod &&
6501 (ReturnType.isNull() ||
6502 (ReturnType->isObjCObjectPointerType() &&
6503 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6504 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6505 ->getName() == "NSArray"))) {
Douglas Gregor62041592011-02-17 03:19:26 +00006506 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006507 = (Twine(Property->getName()) + "AtIndexes").str();
Douglas Gregor62041592011-02-17 03:19:26 +00006508 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Stephen Hines176edba2014-12-01 14:53:08 -08006509 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId)).second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006510 if (ReturnType.isNull()) {
6511 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6512 Builder.AddTextChunk("NSArray *");
6513 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6514 }
6515
6516 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6517 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6518 Builder.AddTextChunk("NSIndexSet *");
6519 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6520 Builder.AddTextChunk("indexes");
6521 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6522 CXCursor_ObjCInstanceMethodDecl));
6523 }
6524 }
6525
6526 // Add -(void)getKey:(type **)buffer range:(NSRange)inRange
6527 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006528 std::string SelectorName = (Twine("get") + UpperKey).str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006529 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor62041592011-02-17 03:19:26 +00006530 &Context.Idents.get(SelectorName),
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006531 &Context.Idents.get("range")
6532 };
6533
Stephen Hines176edba2014-12-01 14:53:08 -08006534 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds)).second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006535 if (ReturnType.isNull()) {
6536 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6537 Builder.AddTextChunk("void");
6538 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6539 }
6540
6541 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6542 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6543 Builder.AddPlaceholderChunk("object-type");
6544 Builder.AddTextChunk(" **");
6545 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6546 Builder.AddTextChunk("buffer");
6547 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6548 Builder.AddTypedTextChunk("range:");
6549 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6550 Builder.AddTextChunk("NSRange");
6551 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6552 Builder.AddTextChunk("inRange");
6553 Results.AddResult(Result(Builder.TakeString(), IndexedGetterPriority,
6554 CXCursor_ObjCInstanceMethodDecl));
6555 }
6556 }
6557
6558 // Mutable indexed accessors
6559
6560 // - (void)insertObject:(type *)object inKeyAtIndex:(NSUInteger)index
6561 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006562 std::string SelectorName = (Twine("in") + UpperKey + "AtIndex").str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006563 IdentifierInfo *SelectorIds[2] = {
6564 &Context.Idents.get("insertObject"),
Douglas Gregor62041592011-02-17 03:19:26 +00006565 &Context.Idents.get(SelectorName)
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006566 };
6567
Stephen Hines176edba2014-12-01 14:53:08 -08006568 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds)).second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006569 if (ReturnType.isNull()) {
6570 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6571 Builder.AddTextChunk("void");
6572 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6573 }
6574
6575 Builder.AddTypedTextChunk("insertObject:");
6576 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6577 Builder.AddPlaceholderChunk("object-type");
6578 Builder.AddTextChunk(" *");
6579 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6580 Builder.AddTextChunk("object");
6581 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6582 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6583 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6584 Builder.AddPlaceholderChunk("NSUInteger");
6585 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6586 Builder.AddTextChunk("index");
6587 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6588 CXCursor_ObjCInstanceMethodDecl));
6589 }
6590 }
6591
6592 // - (void)insertKey:(NSArray *)array atIndexes:(NSIndexSet *)indexes
6593 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006594 std::string SelectorName = (Twine("insert") + UpperKey).str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006595 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor62041592011-02-17 03:19:26 +00006596 &Context.Idents.get(SelectorName),
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006597 &Context.Idents.get("atIndexes")
6598 };
6599
Stephen Hines176edba2014-12-01 14:53:08 -08006600 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds)).second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006601 if (ReturnType.isNull()) {
6602 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6603 Builder.AddTextChunk("void");
6604 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6605 }
6606
6607 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6608 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6609 Builder.AddTextChunk("NSArray *");
6610 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6611 Builder.AddTextChunk("array");
6612 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6613 Builder.AddTypedTextChunk("atIndexes:");
6614 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6615 Builder.AddPlaceholderChunk("NSIndexSet *");
6616 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6617 Builder.AddTextChunk("indexes");
6618 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6619 CXCursor_ObjCInstanceMethodDecl));
6620 }
6621 }
6622
6623 // -(void)removeObjectFromKeyAtIndex:(NSUInteger)index
6624 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006625 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006626 = (Twine("removeObjectFrom") + UpperKey + "AtIndex").str();
Douglas Gregor62041592011-02-17 03:19:26 +00006627 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Stephen Hines176edba2014-12-01 14:53:08 -08006628 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId)).second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006629 if (ReturnType.isNull()) {
6630 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6631 Builder.AddTextChunk("void");
6632 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6633 }
6634
6635 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6636 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6637 Builder.AddTextChunk("NSUInteger");
6638 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6639 Builder.AddTextChunk("index");
6640 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6641 CXCursor_ObjCInstanceMethodDecl));
6642 }
6643 }
6644
6645 // -(void)removeKeyAtIndexes:(NSIndexSet *)indexes
6646 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006647 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006648 = (Twine("remove") + UpperKey + "AtIndexes").str();
Douglas Gregor62041592011-02-17 03:19:26 +00006649 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Stephen Hines176edba2014-12-01 14:53:08 -08006650 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId)).second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006651 if (ReturnType.isNull()) {
6652 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6653 Builder.AddTextChunk("void");
6654 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6655 }
6656
6657 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6658 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6659 Builder.AddTextChunk("NSIndexSet *");
6660 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6661 Builder.AddTextChunk("indexes");
6662 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6663 CXCursor_ObjCInstanceMethodDecl));
6664 }
6665 }
6666
6667 // - (void)replaceObjectInKeyAtIndex:(NSUInteger)index withObject:(id)object
6668 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006669 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006670 = (Twine("replaceObjectIn") + UpperKey + "AtIndex").str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006671 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor62041592011-02-17 03:19:26 +00006672 &Context.Idents.get(SelectorName),
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006673 &Context.Idents.get("withObject")
6674 };
6675
Stephen Hines176edba2014-12-01 14:53:08 -08006676 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds)).second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006677 if (ReturnType.isNull()) {
6678 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6679 Builder.AddTextChunk("void");
6680 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6681 }
6682
6683 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6684 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6685 Builder.AddPlaceholderChunk("NSUInteger");
6686 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6687 Builder.AddTextChunk("index");
6688 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6689 Builder.AddTypedTextChunk("withObject:");
6690 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6691 Builder.AddTextChunk("id");
6692 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6693 Builder.AddTextChunk("object");
6694 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6695 CXCursor_ObjCInstanceMethodDecl));
6696 }
6697 }
6698
6699 // - (void)replaceKeyAtIndexes:(NSIndexSet *)indexes withKey:(NSArray *)array
6700 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006701 std::string SelectorName1
Chris Lattner5f9e2722011-07-23 10:55:15 +00006702 = (Twine("replace") + UpperKey + "AtIndexes").str();
6703 std::string SelectorName2 = (Twine("with") + UpperKey).str();
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006704 IdentifierInfo *SelectorIds[2] = {
Douglas Gregor62041592011-02-17 03:19:26 +00006705 &Context.Idents.get(SelectorName1),
6706 &Context.Idents.get(SelectorName2)
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006707 };
6708
Stephen Hines176edba2014-12-01 14:53:08 -08006709 if (KnownSelectors.insert(Selectors.getSelector(2, SelectorIds)).second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006710 if (ReturnType.isNull()) {
6711 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6712 Builder.AddTextChunk("void");
6713 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6714 }
6715
6716 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName1 + ":"));
6717 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6718 Builder.AddPlaceholderChunk("NSIndexSet *");
6719 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6720 Builder.AddTextChunk("indexes");
6721 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
6722 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName2 + ":"));
6723 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6724 Builder.AddTextChunk("NSArray *");
6725 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6726 Builder.AddTextChunk("array");
6727 Results.AddResult(Result(Builder.TakeString(), IndexedSetterPriority,
6728 CXCursor_ObjCInstanceMethodDecl));
6729 }
6730 }
6731
6732 // Unordered getters
6733 // - (NSEnumerator *)enumeratorOfKey
6734 if (IsInstanceMethod &&
6735 (ReturnType.isNull() ||
6736 (ReturnType->isObjCObjectPointerType() &&
6737 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6738 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6739 ->getName() == "NSEnumerator"))) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006740 std::string SelectorName = (Twine("enumeratorOf") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006741 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Stephen Hines176edba2014-12-01 14:53:08 -08006742 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))
6743 .second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006744 if (ReturnType.isNull()) {
6745 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6746 Builder.AddTextChunk("NSEnumerator *");
6747 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6748 }
6749
6750 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6751 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6752 CXCursor_ObjCInstanceMethodDecl));
6753 }
6754 }
6755
6756 // - (type *)memberOfKey:(type *)object
6757 if (IsInstanceMethod &&
6758 (ReturnType.isNull() || ReturnType->isObjCObjectPointerType())) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006759 std::string SelectorName = (Twine("memberOf") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006760 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Stephen Hines176edba2014-12-01 14:53:08 -08006761 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId)).second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006762 if (ReturnType.isNull()) {
6763 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6764 Builder.AddPlaceholderChunk("object-type");
6765 Builder.AddTextChunk(" *");
6766 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6767 }
6768
6769 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6770 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6771 if (ReturnType.isNull()) {
6772 Builder.AddPlaceholderChunk("object-type");
6773 Builder.AddTextChunk(" *");
6774 } else {
6775 Builder.AddTextChunk(GetCompletionTypeString(ReturnType, Context,
Douglas Gregor8987b232011-09-27 23:30:47 +00006776 Policy,
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006777 Builder.getAllocator()));
6778 }
6779 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6780 Builder.AddTextChunk("object");
6781 Results.AddResult(Result(Builder.TakeString(), UnorderedGetterPriority,
6782 CXCursor_ObjCInstanceMethodDecl));
6783 }
6784 }
6785
6786 // Mutable unordered accessors
6787 // - (void)addKeyObject:(type *)object
6788 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006789 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006790 = (Twine("add") + UpperKey + Twine("Object")).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006791 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Stephen Hines176edba2014-12-01 14:53:08 -08006792 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId)).second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006793 if (ReturnType.isNull()) {
6794 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6795 Builder.AddTextChunk("void");
6796 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6797 }
6798
6799 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6800 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6801 Builder.AddPlaceholderChunk("object-type");
6802 Builder.AddTextChunk(" *");
6803 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6804 Builder.AddTextChunk("object");
6805 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6806 CXCursor_ObjCInstanceMethodDecl));
6807 }
6808 }
6809
6810 // - (void)addKey:(NSSet *)objects
6811 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006812 std::string SelectorName = (Twine("add") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006813 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Stephen Hines176edba2014-12-01 14:53:08 -08006814 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId)).second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006815 if (ReturnType.isNull()) {
6816 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6817 Builder.AddTextChunk("void");
6818 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6819 }
6820
6821 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6822 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6823 Builder.AddTextChunk("NSSet *");
6824 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6825 Builder.AddTextChunk("objects");
6826 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6827 CXCursor_ObjCInstanceMethodDecl));
6828 }
6829 }
6830
6831 // - (void)removeKeyObject:(type *)object
6832 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Douglas Gregor62041592011-02-17 03:19:26 +00006833 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006834 = (Twine("remove") + UpperKey + Twine("Object")).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006835 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Stephen Hines176edba2014-12-01 14:53:08 -08006836 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId)).second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006837 if (ReturnType.isNull()) {
6838 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6839 Builder.AddTextChunk("void");
6840 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6841 }
6842
6843 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6844 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6845 Builder.AddPlaceholderChunk("object-type");
6846 Builder.AddTextChunk(" *");
6847 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6848 Builder.AddTextChunk("object");
6849 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6850 CXCursor_ObjCInstanceMethodDecl));
6851 }
6852 }
6853
6854 // - (void)removeKey:(NSSet *)objects
6855 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006856 std::string SelectorName = (Twine("remove") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006857 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Stephen Hines176edba2014-12-01 14:53:08 -08006858 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId)).second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006859 if (ReturnType.isNull()) {
6860 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6861 Builder.AddTextChunk("void");
6862 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6863 }
6864
6865 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6866 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6867 Builder.AddTextChunk("NSSet *");
6868 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6869 Builder.AddTextChunk("objects");
6870 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6871 CXCursor_ObjCInstanceMethodDecl));
6872 }
6873 }
6874
6875 // - (void)intersectKey:(NSSet *)objects
6876 if (IsInstanceMethod && ReturnTypeMatchesVoid) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00006877 std::string SelectorName = (Twine("intersect") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006878 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Stephen Hines176edba2014-12-01 14:53:08 -08006879 if (KnownSelectors.insert(Selectors.getUnarySelector(SelectorId)).second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006880 if (ReturnType.isNull()) {
6881 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6882 Builder.AddTextChunk("void");
6883 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6884 }
6885
6886 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName + ":"));
6887 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6888 Builder.AddTextChunk("NSSet *");
6889 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6890 Builder.AddTextChunk("objects");
6891 Results.AddResult(Result(Builder.TakeString(), UnorderedSetterPriority,
6892 CXCursor_ObjCInstanceMethodDecl));
6893 }
6894 }
6895
6896 // Key-Value Observing
6897 // + (NSSet *)keyPathsForValuesAffectingKey
6898 if (!IsInstanceMethod &&
6899 (ReturnType.isNull() ||
6900 (ReturnType->isObjCObjectPointerType() &&
6901 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl() &&
6902 ReturnType->getAs<ObjCObjectPointerType>()->getInterfaceDecl()
6903 ->getName() == "NSSet"))) {
Douglas Gregor62041592011-02-17 03:19:26 +00006904 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006905 = (Twine("keyPathsForValuesAffecting") + UpperKey).str();
Douglas Gregor62041592011-02-17 03:19:26 +00006906 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Stephen Hines176edba2014-12-01 14:53:08 -08006907 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))
6908 .second) {
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006909 if (ReturnType.isNull()) {
6910 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6911 Builder.AddTextChunk("NSSet *");
6912 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6913 }
6914
6915 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6916 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
Douglas Gregor3f828d12011-06-02 04:02:27 +00006917 CXCursor_ObjCClassMethodDecl));
6918 }
6919 }
6920
6921 // + (BOOL)automaticallyNotifiesObserversForKey
6922 if (!IsInstanceMethod &&
6923 (ReturnType.isNull() ||
6924 ReturnType->isIntegerType() ||
6925 ReturnType->isBooleanType())) {
6926 std::string SelectorName
Chris Lattner5f9e2722011-07-23 10:55:15 +00006927 = (Twine("automaticallyNotifiesObserversOf") + UpperKey).str();
Douglas Gregor3f828d12011-06-02 04:02:27 +00006928 IdentifierInfo *SelectorId = &Context.Idents.get(SelectorName);
Stephen Hines176edba2014-12-01 14:53:08 -08006929 if (KnownSelectors.insert(Selectors.getNullarySelector(SelectorId))
6930 .second) {
Douglas Gregor3f828d12011-06-02 04:02:27 +00006931 if (ReturnType.isNull()) {
6932 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
6933 Builder.AddTextChunk("BOOL");
6934 Builder.AddChunk(CodeCompletionString::CK_RightParen);
6935 }
6936
6937 Builder.AddTypedTextChunk(Allocator.CopyString(SelectorName));
6938 Results.AddResult(Result(Builder.TakeString(), CCP_CodePattern,
6939 CXCursor_ObjCClassMethodDecl));
Douglas Gregor577cdfd2011-02-17 00:22:45 +00006940 }
6941 }
6942}
6943
Douglas Gregore8f5a172010-04-07 00:21:17 +00006944void Sema::CodeCompleteObjCMethodDecl(Scope *S,
6945 bool IsInstanceMethod,
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00006946 ParsedType ReturnTy) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006947 // Determine the return type of the method we're declaring, if
6948 // provided.
6949 QualType ReturnType = GetTypeFromParser(ReturnTy);
Stephen Hines6bcf27b2014-05-29 04:14:42 -07006950 Decl *IDecl = nullptr;
Fariborz Jahaniana28948f2011-08-22 15:54:49 +00006951 if (CurContext->isObjCContainer()) {
6952 ObjCContainerDecl *OCD = dyn_cast<ObjCContainerDecl>(CurContext);
6953 IDecl = cast<Decl>(OCD);
6954 }
Douglas Gregorea766182010-10-18 18:21:28 +00006955 // Determine where we should start searching for methods.
Stephen Hines6bcf27b2014-05-29 04:14:42 -07006956 ObjCContainerDecl *SearchDecl = nullptr;
Douglas Gregore8f5a172010-04-07 00:21:17 +00006957 bool IsInImplementation = false;
John McCalld226f652010-08-21 09:40:31 +00006958 if (Decl *D = IDecl) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006959 if (ObjCImplementationDecl *Impl = dyn_cast<ObjCImplementationDecl>(D)) {
6960 SearchDecl = Impl->getClassInterface();
Douglas Gregore8f5a172010-04-07 00:21:17 +00006961 IsInImplementation = true;
6962 } else if (ObjCCategoryImplDecl *CatImpl
Douglas Gregorea766182010-10-18 18:21:28 +00006963 = dyn_cast<ObjCCategoryImplDecl>(D)) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00006964 SearchDecl = CatImpl->getCategoryDecl();
Douglas Gregore8f5a172010-04-07 00:21:17 +00006965 IsInImplementation = true;
Douglas Gregorea766182010-10-18 18:21:28 +00006966 } else
Douglas Gregore8f5a172010-04-07 00:21:17 +00006967 SearchDecl = dyn_cast<ObjCContainerDecl>(D);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006968 }
6969
6970 if (!SearchDecl && S) {
Ted Kremenekf0d58612013-10-08 17:08:03 +00006971 if (DeclContext *DC = S->getEntity())
Douglas Gregore8f5a172010-04-07 00:21:17 +00006972 SearchDecl = dyn_cast<ObjCContainerDecl>(DC);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006973 }
6974
Douglas Gregorea766182010-10-18 18:21:28 +00006975 if (!SearchDecl) {
Douglas Gregore6b1bb62010-08-11 21:23:17 +00006976 HandleCodeCompleteResults(this, CodeCompleter,
6977 CodeCompletionContext::CCC_Other,
Stephen Hines6bcf27b2014-05-29 04:14:42 -07006978 nullptr, 0);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006979 return;
6980 }
6981
6982 // Find all of the methods that we could declare/implement here.
6983 KnownMethodsMap KnownMethods;
6984 FindImplementableMethods(Context, SearchDecl, IsInstanceMethod,
Douglas Gregorea766182010-10-18 18:21:28 +00006985 ReturnType, KnownMethods);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006986
Douglas Gregore8f5a172010-04-07 00:21:17 +00006987 // Add declarations or definitions for each of the known methods.
John McCall0a2c5e22010-08-25 06:19:51 +00006988 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00006989 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00006990 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00006991 CodeCompletionContext::CCC_Other);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006992 Results.EnterNewScope();
Douglas Gregor8987b232011-09-27 23:30:47 +00006993 PrintingPolicy Policy = getCompletionPrintingPolicy(*this);
Douglas Gregore8f5a172010-04-07 00:21:17 +00006994 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
6995 MEnd = KnownMethods.end();
6996 M != MEnd; ++M) {
Benjamin Kramere1039792013-06-29 17:52:13 +00006997 ObjCMethodDecl *Method = M->second.getPointer();
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00006998 CodeCompletionBuilder Builder(Results.getAllocator(),
6999 Results.getCodeCompletionTUInfo());
Douglas Gregore8f5a172010-04-07 00:21:17 +00007000
7001 // If the result type was not already provided, add it to the
7002 // pattern as (type).
Douglas Gregor577cdfd2011-02-17 00:22:45 +00007003 if (ReturnType.isNull())
Stephen Hines651f13c2014-04-23 16:59:28 -07007004 AddObjCPassingTypeChunk(Method->getReturnType(),
7005 Method->getObjCDeclQualifier(), Context, Policy,
7006 Builder);
Douglas Gregore8f5a172010-04-07 00:21:17 +00007007
7008 Selector Sel = Method->getSelector();
7009
7010 // Add the first part of the selector to the pattern.
Douglas Gregordae68752011-02-01 22:57:45 +00007011 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor813d8342011-02-18 22:29:55 +00007012 Sel.getNameForSlot(0)));
Douglas Gregore8f5a172010-04-07 00:21:17 +00007013
7014 // Add parameters to the pattern.
7015 unsigned I = 0;
7016 for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
7017 PEnd = Method->param_end();
7018 P != PEnd; (void)++P, ++I) {
7019 // Add the part of the selector name.
7020 if (I == 0)
Douglas Gregor218937c2011-02-01 19:23:04 +00007021 Builder.AddTypedTextChunk(":");
Douglas Gregore8f5a172010-04-07 00:21:17 +00007022 else if (I < Sel.getNumArgs()) {
Douglas Gregor218937c2011-02-01 19:23:04 +00007023 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7024 Builder.AddTypedTextChunk(
Douglas Gregor813d8342011-02-18 22:29:55 +00007025 Builder.getAllocator().CopyString(Sel.getNameForSlot(I) + ":"));
Douglas Gregore8f5a172010-04-07 00:21:17 +00007026 } else
7027 break;
7028
7029 // Add the parameter type.
Douglas Gregor90f5f472012-04-10 18:35:07 +00007030 AddObjCPassingTypeChunk((*P)->getOriginalType(),
7031 (*P)->getObjCDeclQualifier(),
7032 Context, Policy,
Douglas Gregor8987b232011-09-27 23:30:47 +00007033 Builder);
Douglas Gregore8f5a172010-04-07 00:21:17 +00007034
7035 if (IdentifierInfo *Id = (*P)->getIdentifier())
Douglas Gregordae68752011-02-01 22:57:45 +00007036 Builder.AddTextChunk(Builder.getAllocator().CopyString( Id->getName()));
Douglas Gregore8f5a172010-04-07 00:21:17 +00007037 }
7038
7039 if (Method->isVariadic()) {
7040 if (Method->param_size() > 0)
Douglas Gregor218937c2011-02-01 19:23:04 +00007041 Builder.AddChunk(CodeCompletionString::CK_Comma);
7042 Builder.AddTextChunk("...");
Douglas Gregore17794f2010-08-31 05:13:43 +00007043 }
Douglas Gregore8f5a172010-04-07 00:21:17 +00007044
Douglas Gregor447107d2010-05-28 00:57:46 +00007045 if (IsInImplementation && Results.includeCodePatterns()) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00007046 // We will be defining the method here, so add a compound statement.
Douglas Gregor218937c2011-02-01 19:23:04 +00007047 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7048 Builder.AddChunk(CodeCompletionString::CK_LeftBrace);
7049 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
Stephen Hines651f13c2014-04-23 16:59:28 -07007050 if (!Method->getReturnType()->isVoidType()) {
Douglas Gregore8f5a172010-04-07 00:21:17 +00007051 // If the result type is not void, add a return clause.
Douglas Gregor218937c2011-02-01 19:23:04 +00007052 Builder.AddTextChunk("return");
7053 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7054 Builder.AddPlaceholderChunk("expression");
7055 Builder.AddChunk(CodeCompletionString::CK_SemiColon);
Douglas Gregore8f5a172010-04-07 00:21:17 +00007056 } else
Douglas Gregor218937c2011-02-01 19:23:04 +00007057 Builder.AddPlaceholderChunk("statements");
Douglas Gregore8f5a172010-04-07 00:21:17 +00007058
Douglas Gregor218937c2011-02-01 19:23:04 +00007059 Builder.AddChunk(CodeCompletionString::CK_VerticalSpace);
7060 Builder.AddChunk(CodeCompletionString::CK_RightBrace);
Douglas Gregore8f5a172010-04-07 00:21:17 +00007061 }
7062
Douglas Gregor408be5a2010-08-25 01:08:01 +00007063 unsigned Priority = CCP_CodePattern;
Benjamin Kramere1039792013-06-29 17:52:13 +00007064 if (!M->second.getInt())
Douglas Gregor408be5a2010-08-25 01:08:01 +00007065 Priority += CCD_InBaseClass;
7066
Douglas Gregorba103062012-03-27 23:34:16 +00007067 Results.AddResult(Result(Builder.TakeString(), Method, Priority));
Douglas Gregore8f5a172010-04-07 00:21:17 +00007068 }
7069
Douglas Gregor577cdfd2011-02-17 00:22:45 +00007070 // Add Key-Value-Coding and Key-Value-Observing accessor methods for all of
7071 // the properties in this class and its categories.
David Blaikie4e4d0842012-03-11 07:00:24 +00007072 if (Context.getLangOpts().ObjC2) {
Chris Lattner5f9e2722011-07-23 10:55:15 +00007073 SmallVector<ObjCContainerDecl *, 4> Containers;
Douglas Gregor577cdfd2011-02-17 00:22:45 +00007074 Containers.push_back(SearchDecl);
7075
Douglas Gregore74c25c2011-05-04 23:50:46 +00007076 VisitedSelectorSet KnownSelectors;
7077 for (KnownMethodsMap::iterator M = KnownMethods.begin(),
7078 MEnd = KnownMethods.end();
7079 M != MEnd; ++M)
7080 KnownSelectors.insert(M->first);
7081
7082
Douglas Gregor577cdfd2011-02-17 00:22:45 +00007083 ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(SearchDecl);
7084 if (!IFace)
7085 if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(SearchDecl))
7086 IFace = Category->getClassInterface();
7087
Stephen Hines651f13c2014-04-23 16:59:28 -07007088 if (IFace)
7089 for (auto *Cat : IFace->visible_categories())
7090 Containers.push_back(Cat);
Douglas Gregor577cdfd2011-02-17 00:22:45 +00007091
Stephen Hines651f13c2014-04-23 16:59:28 -07007092 for (unsigned I = 0, N = Containers.size(); I != N; ++I)
7093 for (auto *P : Containers[I]->properties())
7094 AddObjCKeyValueCompletions(P, IsInstanceMethod, ReturnType, Context,
Douglas Gregore74c25c2011-05-04 23:50:46 +00007095 KnownSelectors, Results);
Douglas Gregor577cdfd2011-02-17 00:22:45 +00007096 }
7097
Douglas Gregore8f5a172010-04-07 00:21:17 +00007098 Results.ExitScope();
7099
Douglas Gregore6b1bb62010-08-11 21:23:17 +00007100 HandleCodeCompleteResults(this, CodeCompleter,
7101 CodeCompletionContext::CCC_Other,
7102 Results.data(),Results.size());
Douglas Gregore8f5a172010-04-07 00:21:17 +00007103}
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007104
7105void Sema::CodeCompleteObjCMethodDeclSelector(Scope *S,
7106 bool IsInstanceMethod,
Douglas Gregor40ed9a12010-07-08 23:37:41 +00007107 bool AtParameterName,
John McCallb3d87482010-08-24 05:47:05 +00007108 ParsedType ReturnTy,
Dmitri Gribenko050315b2013-06-16 03:47:57 +00007109 ArrayRef<IdentifierInfo *> SelIdents) {
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007110 // If we have an external source, load the entire class method
Sebastian Redl3c7f4132010-08-18 23:57:06 +00007111 // pool from the AST file.
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007112 if (ExternalSource) {
7113 for (uint32_t I = 0, N = ExternalSource->GetNumExternalSelectors();
7114 I != N; ++I) {
7115 Selector Sel = ExternalSource->GetExternalSelector(I);
Sebastian Redldb9d2142010-08-02 23:18:59 +00007116 if (Sel.isNull() || MethodPool.count(Sel))
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007117 continue;
Sebastian Redldb9d2142010-08-02 23:18:59 +00007118
7119 ReadMethodPool(Sel);
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007120 }
7121 }
7122
7123 // Build the set of methods we can see.
John McCall0a2c5e22010-08-25 06:19:51 +00007124 typedef CodeCompletionResult Result;
Douglas Gregor218937c2011-02-01 19:23:04 +00007125 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007126 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor218937c2011-02-01 19:23:04 +00007127 CodeCompletionContext::CCC_Other);
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007128
7129 if (ReturnTy)
7130 Results.setPreferredType(GetTypeFromParser(ReturnTy).getNonReferenceType());
Sebastian Redldb9d2142010-08-02 23:18:59 +00007131
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007132 Results.EnterNewScope();
Sebastian Redldb9d2142010-08-02 23:18:59 +00007133 for (GlobalMethodPool::iterator M = MethodPool.begin(),
7134 MEnd = MethodPool.end();
7135 M != MEnd; ++M) {
7136 for (ObjCMethodList *MethList = IsInstanceMethod ? &M->second.first :
7137 &M->second.second;
Stephen Hines0e2c34f2015-03-23 12:09:02 -07007138 MethList && MethList->getMethod();
Argyrios Kyrtzidis2e3d8c02013-04-17 00:08:58 +00007139 MethList = MethList->getNext()) {
Stephen Hines0e2c34f2015-03-23 12:09:02 -07007140 if (!isAcceptableObjCMethod(MethList->getMethod(), MK_Any, SelIdents))
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007141 continue;
7142
Douglas Gregor40ed9a12010-07-08 23:37:41 +00007143 if (AtParameterName) {
7144 // Suggest parameter names we've seen before.
Dmitri Gribenko050315b2013-06-16 03:47:57 +00007145 unsigned NumSelIdents = SelIdents.size();
Stephen Hines0e2c34f2015-03-23 12:09:02 -07007146 if (NumSelIdents &&
7147 NumSelIdents <= MethList->getMethod()->param_size()) {
7148 ParmVarDecl *Param =
7149 MethList->getMethod()->parameters()[NumSelIdents - 1];
Douglas Gregor40ed9a12010-07-08 23:37:41 +00007150 if (Param->getIdentifier()) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007151 CodeCompletionBuilder Builder(Results.getAllocator(),
7152 Results.getCodeCompletionTUInfo());
Douglas Gregordae68752011-02-01 22:57:45 +00007153 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00007154 Param->getIdentifier()->getName()));
7155 Results.AddResult(Builder.TakeString());
Douglas Gregor40ed9a12010-07-08 23:37:41 +00007156 }
7157 }
7158
7159 continue;
7160 }
Stephen Hines6bcf27b2014-05-29 04:14:42 -07007161
Stephen Hines0e2c34f2015-03-23 12:09:02 -07007162 Result R(MethList->getMethod(),
7163 Results.getBasePriority(MethList->getMethod()), nullptr);
Dmitri Gribenko050315b2013-06-16 03:47:57 +00007164 R.StartParameter = SelIdents.size();
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007165 R.AllParametersAreInformative = false;
7166 R.DeclaringEntity = true;
7167 Results.MaybeAddResult(R, CurContext);
7168 }
7169 }
7170
7171 Results.ExitScope();
Douglas Gregore6b1bb62010-08-11 21:23:17 +00007172 HandleCodeCompleteResults(this, CodeCompleter,
7173 CodeCompletionContext::CCC_Other,
7174 Results.data(),Results.size());
Douglas Gregor1f5537a2010-07-08 23:20:03 +00007175}
Douglas Gregor87c08a52010-08-13 22:48:40 +00007176
Douglas Gregorf29c5232010-08-24 22:20:20 +00007177void Sema::CodeCompletePreprocessorDirective(bool InConditional) {
Douglas Gregor218937c2011-02-01 19:23:04 +00007178 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007179 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00007180 CodeCompletionContext::CCC_PreprocessorDirective);
Douglas Gregorf44e8542010-08-24 19:08:16 +00007181 Results.EnterNewScope();
7182
7183 // #if <condition>
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007184 CodeCompletionBuilder Builder(Results.getAllocator(),
7185 Results.getCodeCompletionTUInfo());
Douglas Gregor218937c2011-02-01 19:23:04 +00007186 Builder.AddTypedTextChunk("if");
7187 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7188 Builder.AddPlaceholderChunk("condition");
7189 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007190
7191 // #ifdef <macro>
Douglas Gregor218937c2011-02-01 19:23:04 +00007192 Builder.AddTypedTextChunk("ifdef");
7193 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7194 Builder.AddPlaceholderChunk("macro");
7195 Results.AddResult(Builder.TakeString());
7196
Douglas Gregorf44e8542010-08-24 19:08:16 +00007197 // #ifndef <macro>
Douglas Gregor218937c2011-02-01 19:23:04 +00007198 Builder.AddTypedTextChunk("ifndef");
7199 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7200 Builder.AddPlaceholderChunk("macro");
7201 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007202
7203 if (InConditional) {
7204 // #elif <condition>
Douglas Gregor218937c2011-02-01 19:23:04 +00007205 Builder.AddTypedTextChunk("elif");
7206 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7207 Builder.AddPlaceholderChunk("condition");
7208 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007209
7210 // #else
Douglas Gregor218937c2011-02-01 19:23:04 +00007211 Builder.AddTypedTextChunk("else");
7212 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007213
7214 // #endif
Douglas Gregor218937c2011-02-01 19:23:04 +00007215 Builder.AddTypedTextChunk("endif");
7216 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007217 }
7218
7219 // #include "header"
Douglas Gregor218937c2011-02-01 19:23:04 +00007220 Builder.AddTypedTextChunk("include");
7221 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7222 Builder.AddTextChunk("\"");
7223 Builder.AddPlaceholderChunk("header");
7224 Builder.AddTextChunk("\"");
7225 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007226
7227 // #include <header>
Douglas Gregor218937c2011-02-01 19:23:04 +00007228 Builder.AddTypedTextChunk("include");
7229 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7230 Builder.AddTextChunk("<");
7231 Builder.AddPlaceholderChunk("header");
7232 Builder.AddTextChunk(">");
7233 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007234
7235 // #define <macro>
Douglas Gregor218937c2011-02-01 19:23:04 +00007236 Builder.AddTypedTextChunk("define");
7237 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7238 Builder.AddPlaceholderChunk("macro");
7239 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007240
7241 // #define <macro>(<args>)
Douglas Gregor218937c2011-02-01 19:23:04 +00007242 Builder.AddTypedTextChunk("define");
7243 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7244 Builder.AddPlaceholderChunk("macro");
7245 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7246 Builder.AddPlaceholderChunk("args");
7247 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7248 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007249
7250 // #undef <macro>
Douglas Gregor218937c2011-02-01 19:23:04 +00007251 Builder.AddTypedTextChunk("undef");
7252 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7253 Builder.AddPlaceholderChunk("macro");
7254 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007255
7256 // #line <number>
Douglas Gregor218937c2011-02-01 19:23:04 +00007257 Builder.AddTypedTextChunk("line");
7258 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7259 Builder.AddPlaceholderChunk("number");
7260 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007261
7262 // #line <number> "filename"
Douglas Gregor218937c2011-02-01 19:23:04 +00007263 Builder.AddTypedTextChunk("line");
7264 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7265 Builder.AddPlaceholderChunk("number");
7266 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7267 Builder.AddTextChunk("\"");
7268 Builder.AddPlaceholderChunk("filename");
7269 Builder.AddTextChunk("\"");
7270 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007271
7272 // #error <message>
Douglas Gregor218937c2011-02-01 19:23:04 +00007273 Builder.AddTypedTextChunk("error");
7274 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7275 Builder.AddPlaceholderChunk("message");
7276 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007277
7278 // #pragma <arguments>
Douglas Gregor218937c2011-02-01 19:23:04 +00007279 Builder.AddTypedTextChunk("pragma");
7280 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7281 Builder.AddPlaceholderChunk("arguments");
7282 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007283
David Blaikie4e4d0842012-03-11 07:00:24 +00007284 if (getLangOpts().ObjC1) {
Douglas Gregorf44e8542010-08-24 19:08:16 +00007285 // #import "header"
Douglas Gregor218937c2011-02-01 19:23:04 +00007286 Builder.AddTypedTextChunk("import");
7287 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7288 Builder.AddTextChunk("\"");
7289 Builder.AddPlaceholderChunk("header");
7290 Builder.AddTextChunk("\"");
7291 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007292
7293 // #import <header>
Douglas Gregor218937c2011-02-01 19:23:04 +00007294 Builder.AddTypedTextChunk("import");
7295 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7296 Builder.AddTextChunk("<");
7297 Builder.AddPlaceholderChunk("header");
7298 Builder.AddTextChunk(">");
7299 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007300 }
7301
7302 // #include_next "header"
Douglas Gregor218937c2011-02-01 19:23:04 +00007303 Builder.AddTypedTextChunk("include_next");
7304 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7305 Builder.AddTextChunk("\"");
7306 Builder.AddPlaceholderChunk("header");
7307 Builder.AddTextChunk("\"");
7308 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007309
7310 // #include_next <header>
Douglas Gregor218937c2011-02-01 19:23:04 +00007311 Builder.AddTypedTextChunk("include_next");
7312 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7313 Builder.AddTextChunk("<");
7314 Builder.AddPlaceholderChunk("header");
7315 Builder.AddTextChunk(">");
7316 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007317
7318 // #warning <message>
Douglas Gregor218937c2011-02-01 19:23:04 +00007319 Builder.AddTypedTextChunk("warning");
7320 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7321 Builder.AddPlaceholderChunk("message");
7322 Results.AddResult(Builder.TakeString());
Douglas Gregorf44e8542010-08-24 19:08:16 +00007323
7324 // Note: #ident and #sccs are such crazy anachronisms that we don't provide
7325 // completions for them. And __include_macros is a Clang-internal extension
7326 // that we don't want to encourage anyone to use.
7327
7328 // FIXME: we don't support #assert or #unassert, so don't suggest them.
7329 Results.ExitScope();
7330
Douglas Gregorf44e8542010-08-24 19:08:16 +00007331 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregor721f3592010-08-25 18:41:16 +00007332 CodeCompletionContext::CCC_PreprocessorDirective,
Douglas Gregorf44e8542010-08-24 19:08:16 +00007333 Results.data(), Results.size());
7334}
7335
7336void Sema::CodeCompleteInPreprocessorConditionalExclusion(Scope *S) {
Douglas Gregorf29c5232010-08-24 22:20:20 +00007337 CodeCompleteOrdinaryName(S,
John McCallf312b1e2010-08-26 23:41:50 +00007338 S->getFnParent()? Sema::PCC_RecoveryInFunction
7339 : Sema::PCC_Namespace);
Douglas Gregorf44e8542010-08-24 19:08:16 +00007340}
7341
Douglas Gregorf29c5232010-08-24 22:20:20 +00007342void Sema::CodeCompletePreprocessorMacroName(bool IsDefinition) {
Douglas Gregor218937c2011-02-01 19:23:04 +00007343 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007344 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00007345 IsDefinition? CodeCompletionContext::CCC_MacroName
7346 : CodeCompletionContext::CCC_MacroNameUse);
Douglas Gregor1fbb4472010-08-24 20:21:13 +00007347 if (!IsDefinition && (!CodeCompleter || CodeCompleter->includeMacros())) {
7348 // Add just the names of macros, not their arguments.
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007349 CodeCompletionBuilder Builder(Results.getAllocator(),
7350 Results.getCodeCompletionTUInfo());
Douglas Gregor1fbb4472010-08-24 20:21:13 +00007351 Results.EnterNewScope();
7352 for (Preprocessor::macro_iterator M = PP.macro_begin(),
7353 MEnd = PP.macro_end();
7354 M != MEnd; ++M) {
Douglas Gregordae68752011-02-01 22:57:45 +00007355 Builder.AddTypedTextChunk(Builder.getAllocator().CopyString(
Douglas Gregor218937c2011-02-01 19:23:04 +00007356 M->first->getName()));
Argyrios Kyrtzidisc04bb922012-09-27 00:24:09 +00007357 Results.AddResult(CodeCompletionResult(Builder.TakeString(),
7358 CCP_CodePattern,
7359 CXCursor_MacroDefinition));
Douglas Gregor1fbb4472010-08-24 20:21:13 +00007360 }
7361 Results.ExitScope();
7362 } else if (IsDefinition) {
7363 // FIXME: Can we detect when the user just wrote an include guard above?
7364 }
7365
Douglas Gregor52779fb2010-09-23 23:01:17 +00007366 HandleCodeCompleteResults(this, CodeCompleter, Results.getCompletionContext(),
Douglas Gregor1fbb4472010-08-24 20:21:13 +00007367 Results.data(), Results.size());
7368}
7369
Douglas Gregorf29c5232010-08-24 22:20:20 +00007370void Sema::CodeCompletePreprocessorExpression() {
Douglas Gregor218937c2011-02-01 19:23:04 +00007371 ResultBuilder Results(*this, CodeCompleter->getAllocator(),
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007372 CodeCompleter->getCodeCompletionTUInfo(),
Douglas Gregor52779fb2010-09-23 23:01:17 +00007373 CodeCompletionContext::CCC_PreprocessorExpression);
Douglas Gregorf29c5232010-08-24 22:20:20 +00007374
7375 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00007376 AddMacroResults(PP, Results, true);
Douglas Gregorf29c5232010-08-24 22:20:20 +00007377
7378 // defined (<macro>)
7379 Results.EnterNewScope();
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007380 CodeCompletionBuilder Builder(Results.getAllocator(),
7381 Results.getCodeCompletionTUInfo());
Douglas Gregor218937c2011-02-01 19:23:04 +00007382 Builder.AddTypedTextChunk("defined");
7383 Builder.AddChunk(CodeCompletionString::CK_HorizontalSpace);
7384 Builder.AddChunk(CodeCompletionString::CK_LeftParen);
7385 Builder.AddPlaceholderChunk("macro");
7386 Builder.AddChunk(CodeCompletionString::CK_RightParen);
7387 Results.AddResult(Builder.TakeString());
Douglas Gregorf29c5232010-08-24 22:20:20 +00007388 Results.ExitScope();
7389
7390 HandleCodeCompleteResults(this, CodeCompleter,
7391 CodeCompletionContext::CCC_PreprocessorExpression,
7392 Results.data(), Results.size());
7393}
7394
7395void Sema::CodeCompletePreprocessorMacroArgument(Scope *S,
7396 IdentifierInfo *Macro,
7397 MacroInfo *MacroInfo,
7398 unsigned Argument) {
7399 // FIXME: In the future, we could provide "overload" results, much like we
7400 // do for function calls.
7401
Argyrios Kyrtzidis5c5f03e2011-08-18 19:41:28 +00007402 // Now just ignore this. There will be another code-completion callback
7403 // for the expanded tokens.
Douglas Gregorf29c5232010-08-24 22:20:20 +00007404}
7405
Douglas Gregor55817af2010-08-25 17:04:25 +00007406void Sema::CodeCompleteNaturalLanguage() {
Douglas Gregor55817af2010-08-25 17:04:25 +00007407 HandleCodeCompleteResults(this, CodeCompleter,
Douglas Gregoraf1c6b52010-08-25 17:10:00 +00007408 CodeCompletionContext::CCC_NaturalLanguage,
Stephen Hines6bcf27b2014-05-29 04:14:42 -07007409 nullptr, 0);
Douglas Gregor55817af2010-08-25 17:04:25 +00007410}
7411
Douglas Gregordae68752011-02-01 22:57:45 +00007412void Sema::GatherGlobalCodeCompletions(CodeCompletionAllocator &Allocator,
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007413 CodeCompletionTUInfo &CCTUInfo,
Chris Lattner5f9e2722011-07-23 10:55:15 +00007414 SmallVectorImpl<CodeCompletionResult> &Results) {
Argyrios Kyrtzidis28a83f52012-04-10 17:23:48 +00007415 ResultBuilder Builder(*this, Allocator, CCTUInfo,
7416 CodeCompletionContext::CCC_Recovery);
Douglas Gregor8071e422010-08-15 06:18:01 +00007417 if (!CodeCompleter || CodeCompleter->includeGlobals()) {
7418 CodeCompletionDeclConsumer Consumer(Builder,
7419 Context.getTranslationUnitDecl());
7420 LookupVisibleDecls(Context.getTranslationUnitDecl(), LookupAnyName,
7421 Consumer);
7422 }
Douglas Gregor87c08a52010-08-13 22:48:40 +00007423
7424 if (!CodeCompleter || CodeCompleter->includeMacros())
Douglas Gregor3644d972012-10-09 16:01:50 +00007425 AddMacroResults(PP, Builder, true);
Douglas Gregor87c08a52010-08-13 22:48:40 +00007426
7427 Results.clear();
7428 Results.insert(Results.end(),
7429 Builder.data(), Builder.data() + Builder.size());
7430}