[clang][modules] Don't prevent translation of FW_Private includes when explicitly...
[llvm-project.git] / clang-tools-extra / clangd / InlayHints.cpp
blob3da047f9542138565fe204103967f9dd978a5e00
1 //===--- InlayHints.cpp ------------------------------------------*- C++-*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 #include "InlayHints.h"
9 #include "AST.h"
10 #include "Config.h"
11 #include "HeuristicResolver.h"
12 #include "ParsedAST.h"
13 #include "SourceCode.h"
14 #include "clang/AST/ASTDiagnostic.h"
15 #include "clang/AST/Decl.h"
16 #include "clang/AST/DeclarationName.h"
17 #include "clang/AST/Expr.h"
18 #include "clang/AST/ExprCXX.h"
19 #include "clang/AST/RecursiveASTVisitor.h"
20 #include "clang/AST/Stmt.h"
21 #include "clang/AST/StmtVisitor.h"
22 #include "clang/AST/Type.h"
23 #include "clang/Basic/Builtins.h"
24 #include "clang/Basic/OperatorKinds.h"
25 #include "clang/Basic/SourceManager.h"
26 #include "llvm/ADT/DenseSet.h"
27 #include "llvm/ADT/ScopeExit.h"
28 #include "llvm/ADT/StringExtras.h"
29 #include "llvm/ADT/StringRef.h"
30 #include "llvm/ADT/Twine.h"
31 #include "llvm/Support/Casting.h"
32 #include "llvm/Support/SaveAndRestore.h"
33 #include "llvm/Support/ScopedPrinter.h"
34 #include "llvm/Support/raw_ostream.h"
35 #include <optional>
36 #include <string>
38 namespace clang {
39 namespace clangd {
40 namespace {
42 // For now, inlay hints are always anchored at the left or right of their range.
43 enum class HintSide { Left, Right };
45 // Helper class to iterate over the designator names of an aggregate type.
47 // For an array type, yields [0], [1], [2]...
48 // For aggregate classes, yields null for each base, then .field1, .field2, ...
49 class AggregateDesignatorNames {
50 public:
51 AggregateDesignatorNames(QualType T) {
52 if (!T.isNull()) {
53 T = T.getCanonicalType();
54 if (T->isArrayType()) {
55 IsArray = true;
56 Valid = true;
57 return;
59 if (const RecordDecl *RD = T->getAsRecordDecl()) {
60 Valid = true;
61 FieldsIt = RD->field_begin();
62 FieldsEnd = RD->field_end();
63 if (const auto *CRD = llvm::dyn_cast<CXXRecordDecl>(RD)) {
64 BasesIt = CRD->bases_begin();
65 BasesEnd = CRD->bases_end();
66 Valid = CRD->isAggregate();
68 OneField = Valid && BasesIt == BasesEnd && FieldsIt != FieldsEnd &&
69 std::next(FieldsIt) == FieldsEnd;
73 // Returns false if the type was not an aggregate.
74 operator bool() { return Valid; }
75 // Advance to the next element in the aggregate.
76 void next() {
77 if (IsArray)
78 ++Index;
79 else if (BasesIt != BasesEnd)
80 ++BasesIt;
81 else if (FieldsIt != FieldsEnd)
82 ++FieldsIt;
84 // Print the designator to Out.
85 // Returns false if we could not produce a designator for this element.
86 bool append(std::string &Out, bool ForSubobject) {
87 if (IsArray) {
88 Out.push_back('[');
89 Out.append(std::to_string(Index));
90 Out.push_back(']');
91 return true;
93 if (BasesIt != BasesEnd)
94 return false; // Bases can't be designated. Should we make one up?
95 if (FieldsIt != FieldsEnd) {
96 llvm::StringRef FieldName;
97 if (const IdentifierInfo *II = FieldsIt->getIdentifier())
98 FieldName = II->getName();
100 // For certain objects, their subobjects may be named directly.
101 if (ForSubobject &&
102 (FieldsIt->isAnonymousStructOrUnion() ||
103 // std::array<int,3> x = {1,2,3}. Designators not strictly valid!
104 (OneField && isReservedName(FieldName))))
105 return true;
107 if (!FieldName.empty() && !isReservedName(FieldName)) {
108 Out.push_back('.');
109 Out.append(FieldName.begin(), FieldName.end());
110 return true;
112 return false;
114 return false;
117 private:
118 bool Valid = false;
119 bool IsArray = false;
120 bool OneField = false; // e.g. std::array { T __elements[N]; }
121 unsigned Index = 0;
122 CXXRecordDecl::base_class_const_iterator BasesIt;
123 CXXRecordDecl::base_class_const_iterator BasesEnd;
124 RecordDecl::field_iterator FieldsIt;
125 RecordDecl::field_iterator FieldsEnd;
128 // Collect designator labels describing the elements of an init list.
130 // This function contributes the designators of some (sub)object, which is
131 // represented by the semantic InitListExpr Sem.
132 // This includes any nested subobjects, but *only* if they are part of the same
133 // original syntactic init list (due to brace elision).
134 // In other words, it may descend into subobjects but not written init-lists.
136 // For example: struct Outer { Inner a,b; }; struct Inner { int x, y; }
137 // Outer o{{1, 2}, 3};
138 // This function will be called with Sem = { {1, 2}, {3, ImplicitValue} }
139 // It should generate designators '.a:' and '.b.x:'.
140 // '.a:' is produced directly without recursing into the written sublist.
141 // (The written sublist will have a separate collectDesignators() call later).
142 // Recursion with Prefix='.b' and Sem = {3, ImplicitValue} produces '.b.x:'.
143 void collectDesignators(const InitListExpr *Sem,
144 llvm::DenseMap<SourceLocation, std::string> &Out,
145 const llvm::DenseSet<SourceLocation> &NestedBraces,
146 std::string &Prefix) {
147 if (!Sem || Sem->isTransparent())
148 return;
149 assert(Sem->isSemanticForm());
151 // The elements of the semantic form all correspond to direct subobjects of
152 // the aggregate type. `Fields` iterates over these subobject names.
153 AggregateDesignatorNames Fields(Sem->getType());
154 if (!Fields)
155 return;
156 for (const Expr *Init : Sem->inits()) {
157 auto Next = llvm::make_scope_exit([&, Size(Prefix.size())] {
158 Fields.next(); // Always advance to the next subobject name.
159 Prefix.resize(Size); // Erase any designator we appended.
161 // Skip for a broken initializer or if it is a "hole" in a subobject that
162 // was not explicitly initialized.
163 if (!Init || llvm::isa<ImplicitValueInitExpr>(Init))
164 continue;
166 const auto *BraceElidedSubobject = llvm::dyn_cast<InitListExpr>(Init);
167 if (BraceElidedSubobject &&
168 NestedBraces.contains(BraceElidedSubobject->getLBraceLoc()))
169 BraceElidedSubobject = nullptr; // there were braces!
171 if (!Fields.append(Prefix, BraceElidedSubobject != nullptr))
172 continue; // no designator available for this subobject
173 if (BraceElidedSubobject) {
174 // If the braces were elided, this aggregate subobject is initialized
175 // inline in the same syntactic list.
176 // Descend into the semantic list describing the subobject.
177 // (NestedBraces are still correct, they're from the same syntactic list).
178 collectDesignators(BraceElidedSubobject, Out, NestedBraces, Prefix);
179 continue;
181 Out.try_emplace(Init->getBeginLoc(), Prefix);
185 // Get designators describing the elements of a (syntactic) init list.
186 // This does not produce designators for any explicitly-written nested lists.
187 llvm::DenseMap<SourceLocation, std::string>
188 getDesignators(const InitListExpr *Syn) {
189 assert(Syn->isSyntacticForm());
191 // collectDesignators needs to know which InitListExprs in the semantic tree
192 // were actually written, but InitListExpr::isExplicit() lies.
193 // Instead, record where braces of sub-init-lists occur in the syntactic form.
194 llvm::DenseSet<SourceLocation> NestedBraces;
195 for (const Expr *Init : Syn->inits())
196 if (auto *Nested = llvm::dyn_cast<InitListExpr>(Init))
197 NestedBraces.insert(Nested->getLBraceLoc());
199 // Traverse the semantic form to find the designators.
200 // We use their SourceLocation to correlate with the syntactic form later.
201 llvm::DenseMap<SourceLocation, std::string> Designators;
202 std::string EmptyPrefix;
203 collectDesignators(Syn->isSemanticForm() ? Syn : Syn->getSemanticForm(),
204 Designators, NestedBraces, EmptyPrefix);
205 return Designators;
208 void stripLeadingUnderscores(StringRef &Name) { Name = Name.ltrim('_'); }
210 // getDeclForType() returns the decl responsible for Type's spelling.
211 // This is the inverse of ASTContext::getTypeDeclType().
212 template <typename Ty, typename = decltype(((Ty *)nullptr)->getDecl())>
213 const NamedDecl *getDeclForTypeImpl(const Ty *T) {
214 return T->getDecl();
216 const NamedDecl *getDeclForTypeImpl(const void *T) { return nullptr; }
217 const NamedDecl *getDeclForType(const Type *T) {
218 switch (T->getTypeClass()) {
219 #define ABSTRACT_TYPE(TY, BASE)
220 #define TYPE(TY, BASE) \
221 case Type::TY: \
222 return getDeclForTypeImpl(llvm::cast<TY##Type>(T));
223 #include "clang/AST/TypeNodes.inc"
225 llvm_unreachable("Unknown TypeClass enum");
228 // getSimpleName() returns the plain identifier for an entity, if any.
229 llvm::StringRef getSimpleName(const DeclarationName &DN) {
230 if (IdentifierInfo *Ident = DN.getAsIdentifierInfo())
231 return Ident->getName();
232 return "";
234 llvm::StringRef getSimpleName(const NamedDecl &D) {
235 return getSimpleName(D.getDeclName());
237 llvm::StringRef getSimpleName(QualType T) {
238 if (const auto *ET = llvm::dyn_cast<ElaboratedType>(T))
239 return getSimpleName(ET->getNamedType());
240 if (const auto *BT = llvm::dyn_cast<BuiltinType>(T)) {
241 PrintingPolicy PP(LangOptions{});
242 PP.adjustForCPlusPlus();
243 return BT->getName(PP);
245 if (const auto *D = getDeclForType(T.getTypePtr()))
246 return getSimpleName(D->getDeclName());
247 return "";
250 // Returns a very abbreviated form of an expression, or "" if it's too complex.
251 // For example: `foo->bar()` would produce "bar".
252 // This is used to summarize e.g. the condition of a while loop.
253 std::string summarizeExpr(const Expr *E) {
254 struct Namer : ConstStmtVisitor<Namer, std::string> {
255 std::string Visit(const Expr *E) {
256 if (E == nullptr)
257 return "";
258 return ConstStmtVisitor::Visit(E->IgnoreImplicit());
261 // Any sort of decl reference, we just use the unqualified name.
262 std::string VisitMemberExpr(const MemberExpr *E) {
263 return getSimpleName(*E->getMemberDecl()).str();
265 std::string VisitDeclRefExpr(const DeclRefExpr *E) {
266 return getSimpleName(*E->getFoundDecl()).str();
268 std::string VisitCallExpr(const CallExpr *E) {
269 return Visit(E->getCallee());
271 std::string
272 VisitCXXDependentScopeMemberExpr(const CXXDependentScopeMemberExpr *E) {
273 return getSimpleName(E->getMember()).str();
275 std::string
276 VisitDependentScopeDeclRefExpr(const DependentScopeDeclRefExpr *E) {
277 return getSimpleName(E->getDeclName()).str();
279 std::string VisitCXXFunctionalCastExpr(const CXXFunctionalCastExpr *E) {
280 return getSimpleName(E->getType()).str();
282 std::string VisitCXXTemporaryObjectExpr(const CXXTemporaryObjectExpr *E) {
283 return getSimpleName(E->getType()).str();
286 // Step through implicit nodes that clang doesn't classify as such.
287 std::string VisitCXXMemberCallExpr(const CXXMemberCallExpr *E) {
288 // Call to operator bool() inside if (X): dispatch to X.
289 if (E->getNumArgs() == 0 &&
290 E->getMethodDecl()->getDeclName().getNameKind() ==
291 DeclarationName::CXXConversionFunctionName &&
292 E->getSourceRange() ==
293 E->getImplicitObjectArgument()->getSourceRange())
294 return Visit(E->getImplicitObjectArgument());
295 return ConstStmtVisitor::VisitCXXMemberCallExpr(E);
297 std::string VisitCXXConstructExpr(const CXXConstructExpr *E) {
298 if (E->getNumArgs() == 1)
299 return Visit(E->getArg(0));
300 return "";
303 // Literals are just printed
304 std::string VisitCXXBoolLiteralExpr(const CXXBoolLiteralExpr *E) {
305 return E->getValue() ? "true" : "false";
307 std::string VisitIntegerLiteral(const IntegerLiteral *E) {
308 return llvm::to_string(E->getValue());
310 std::string VisitFloatingLiteral(const FloatingLiteral *E) {
311 std::string Result;
312 llvm::raw_string_ostream OS(Result);
313 E->getValue().print(OS);
314 // Printer adds newlines?!
315 Result.resize(llvm::StringRef(Result).rtrim().size());
316 return Result;
318 std::string VisitStringLiteral(const StringLiteral *E) {
319 std::string Result = "\"";
320 if (E->containsNonAscii()) {
321 Result += "...";
322 } else if (E->getLength() > 10) {
323 Result += E->getString().take_front(7);
324 Result += "...";
325 } else {
326 llvm::raw_string_ostream OS(Result);
327 llvm::printEscapedString(E->getString(), OS);
329 Result.push_back('"');
330 return Result;
333 // Simple operators. Motivating cases are `!x` and `I < Length`.
334 std::string printUnary(llvm::StringRef Spelling, const Expr *Operand,
335 bool Prefix) {
336 std::string Sub = Visit(Operand);
337 if (Sub.empty())
338 return "";
339 if (Prefix)
340 return (Spelling + Sub).str();
341 Sub += Spelling;
342 return Sub;
344 bool InsideBinary = false; // No recursing into binary expressions.
345 std::string printBinary(llvm::StringRef Spelling, const Expr *LHSOp,
346 const Expr *RHSOp) {
347 if (InsideBinary)
348 return "";
349 llvm::SaveAndRestore InBinary(InsideBinary, true);
351 std::string LHS = Visit(LHSOp);
352 std::string RHS = Visit(RHSOp);
353 if (LHS.empty() && RHS.empty())
354 return "";
356 if (LHS.empty())
357 LHS = "...";
358 LHS.push_back(' ');
359 LHS += Spelling;
360 LHS.push_back(' ');
361 if (RHS.empty())
362 LHS += "...";
363 else
364 LHS += RHS;
365 return LHS;
367 std::string VisitUnaryOperator(const UnaryOperator *E) {
368 return printUnary(E->getOpcodeStr(E->getOpcode()), E->getSubExpr(),
369 !E->isPostfix());
371 std::string VisitBinaryOperator(const BinaryOperator *E) {
372 return printBinary(E->getOpcodeStr(E->getOpcode()), E->getLHS(),
373 E->getRHS());
375 std::string VisitCXXOperatorCallExpr(const CXXOperatorCallExpr *E) {
376 const char *Spelling = getOperatorSpelling(E->getOperator());
377 // Handle weird unary-that-look-like-binary postfix operators.
378 if ((E->getOperator() == OO_PlusPlus ||
379 E->getOperator() == OO_MinusMinus) &&
380 E->getNumArgs() == 2)
381 return printUnary(Spelling, E->getArg(0), false);
382 if (E->isInfixBinaryOp())
383 return printBinary(Spelling, E->getArg(0), E->getArg(1));
384 if (E->getNumArgs() == 1) {
385 switch (E->getOperator()) {
386 case OO_Plus:
387 case OO_Minus:
388 case OO_Star:
389 case OO_Amp:
390 case OO_Tilde:
391 case OO_Exclaim:
392 case OO_PlusPlus:
393 case OO_MinusMinus:
394 return printUnary(Spelling, E->getArg(0), true);
395 default:
396 break;
399 return "";
402 return Namer{}.Visit(E);
405 // Determines if any intermediate type in desugaring QualType QT is of
406 // substituted template parameter type. Ignore pointer or reference wrappers.
407 bool isSugaredTemplateParameter(QualType QT) {
408 static auto PeelWrapper = [](QualType QT) {
409 // Neither `PointerType` nor `ReferenceType` is considered as sugared
410 // type. Peel it.
411 QualType Peeled = QT->getPointeeType();
412 return Peeled.isNull() ? QT : Peeled;
415 // This is a bit tricky: we traverse the type structure and find whether or
416 // not a type in the desugaring process is of SubstTemplateTypeParmType.
417 // During the process, we may encounter pointer or reference types that are
418 // not marked as sugared; therefore, the desugar function won't apply. To
419 // move forward the traversal, we retrieve the pointees using
420 // QualType::getPointeeType().
422 // However, getPointeeType could leap over our interests: The QT::getAs<T>()
423 // invoked would implicitly desugar the type. Consequently, if the
424 // SubstTemplateTypeParmType is encompassed within a TypedefType, we may lose
425 // the chance to visit it.
426 // For example, given a QT that represents `std::vector<int *>::value_type`:
427 // `-ElaboratedType 'value_type' sugar
428 // `-TypedefType 'vector<int *>::value_type' sugar
429 // |-Typedef 'value_type'
430 // `-SubstTemplateTypeParmType 'int *' sugar class depth 0 index 0 T
431 // |-ClassTemplateSpecialization 'vector'
432 // `-PointerType 'int *'
433 // `-BuiltinType 'int'
434 // Applying `getPointeeType` to QT results in 'int', a child of our target
435 // node SubstTemplateTypeParmType.
437 // As such, we always prefer the desugared over the pointee for next type
438 // in the iteration. It could avoid the getPointeeType's implicit desugaring.
439 while (true) {
440 if (QT->getAs<SubstTemplateTypeParmType>())
441 return true;
442 QualType Desugared = QT->getLocallyUnqualifiedSingleStepDesugaredType();
443 if (Desugared != QT)
444 QT = Desugared;
445 else if (auto Peeled = PeelWrapper(Desugared); Peeled != QT)
446 QT = Peeled;
447 else
448 break;
450 return false;
453 // A simple wrapper for `clang::desugarForDiagnostic` that provides optional
454 // semantic.
455 std::optional<QualType> desugar(ASTContext &AST, QualType QT) {
456 bool ShouldAKA = false;
457 auto Desugared = clang::desugarForDiagnostic(AST, QT, ShouldAKA);
458 if (!ShouldAKA)
459 return std::nullopt;
460 return Desugared;
463 // Apply a series of heuristic methods to determine whether or not a QualType QT
464 // is suitable for desugaring (e.g. getting the real name behind the using-alias
465 // name). If so, return the desugared type. Otherwise, return the unchanged
466 // parameter QT.
468 // This could be refined further. See
469 // https://github.com/clangd/clangd/issues/1298.
470 QualType maybeDesugar(ASTContext &AST, QualType QT) {
471 // Prefer desugared type for name that aliases the template parameters.
472 // This can prevent things like printing opaque `: type` when accessing std
473 // containers.
474 if (isSugaredTemplateParameter(QT))
475 return desugar(AST, QT).value_or(QT);
477 // Prefer desugared type for `decltype(expr)` specifiers.
478 if (QT->isDecltypeType())
479 return QT.getCanonicalType();
480 if (const AutoType *AT = QT->getContainedAutoType())
481 if (!AT->getDeducedType().isNull() &&
482 AT->getDeducedType()->isDecltypeType())
483 return QT.getCanonicalType();
485 return QT;
488 // Given a callee expression `Fn`, if the call is through a function pointer,
489 // try to find the declaration of the corresponding function pointer type,
490 // so that we can recover argument names from it.
491 // FIXME: This function is mostly duplicated in SemaCodeComplete.cpp; unify.
492 static FunctionProtoTypeLoc getPrototypeLoc(Expr *Fn) {
493 TypeLoc Target;
494 Expr *NakedFn = Fn->IgnoreParenCasts();
495 if (const auto *T = NakedFn->getType().getTypePtr()->getAs<TypedefType>()) {
496 Target = T->getDecl()->getTypeSourceInfo()->getTypeLoc();
497 } else if (const auto *DR = dyn_cast<DeclRefExpr>(NakedFn)) {
498 const auto *D = DR->getDecl();
499 if (const auto *const VD = dyn_cast<VarDecl>(D)) {
500 Target = VD->getTypeSourceInfo()->getTypeLoc();
504 if (!Target)
505 return {};
507 // Unwrap types that may be wrapping the function type
508 while (true) {
509 if (auto P = Target.getAs<PointerTypeLoc>()) {
510 Target = P.getPointeeLoc();
511 continue;
513 if (auto A = Target.getAs<AttributedTypeLoc>()) {
514 Target = A.getModifiedLoc();
515 continue;
517 if (auto P = Target.getAs<ParenTypeLoc>()) {
518 Target = P.getInnerLoc();
519 continue;
521 break;
524 if (auto F = Target.getAs<FunctionProtoTypeLoc>()) {
525 return F;
528 return {};
531 ArrayRef<const ParmVarDecl *>
532 maybeDropCxxExplicitObjectParameters(ArrayRef<const ParmVarDecl *> Params) {
533 if (!Params.empty() && Params.front()->isExplicitObjectParameter())
534 Params = Params.drop_front(1);
535 return Params;
538 struct Callee {
539 // Only one of Decl or Loc is set.
540 // Loc is for calls through function pointers.
541 const FunctionDecl *Decl = nullptr;
542 FunctionProtoTypeLoc Loc;
545 class InlayHintVisitor : public RecursiveASTVisitor<InlayHintVisitor> {
546 public:
547 InlayHintVisitor(std::vector<InlayHint> &Results, ParsedAST &AST,
548 const Config &Cfg, std::optional<Range> RestrictRange)
549 : Results(Results), AST(AST.getASTContext()), Tokens(AST.getTokens()),
550 Cfg(Cfg), RestrictRange(std::move(RestrictRange)),
551 MainFileID(AST.getSourceManager().getMainFileID()),
552 Resolver(AST.getHeuristicResolver()),
553 TypeHintPolicy(this->AST.getPrintingPolicy()) {
554 bool Invalid = false;
555 llvm::StringRef Buf =
556 AST.getSourceManager().getBufferData(MainFileID, &Invalid);
557 MainFileBuf = Invalid ? StringRef{} : Buf;
559 TypeHintPolicy.SuppressScope = true; // keep type names short
560 TypeHintPolicy.AnonymousTagLocations =
561 false; // do not print lambda locations
563 // Not setting PrintCanonicalTypes for "auto" allows
564 // SuppressDefaultTemplateArgs (set by default) to have an effect.
567 bool VisitTypeLoc(TypeLoc TL) {
568 if (const auto *DT = llvm::dyn_cast<DecltypeType>(TL.getType()))
569 if (QualType UT = DT->getUnderlyingType(); !UT->isDependentType())
570 addTypeHint(TL.getSourceRange(), UT, ": ");
571 return true;
574 bool VisitCXXConstructExpr(CXXConstructExpr *E) {
575 // Weed out constructor calls that don't look like a function call with
576 // an argument list, by checking the validity of getParenOrBraceRange().
577 // Also weed out std::initializer_list constructors as there are no names
578 // for the individual arguments.
579 if (!E->getParenOrBraceRange().isValid() ||
580 E->isStdInitListInitialization()) {
581 return true;
584 Callee Callee;
585 Callee.Decl = E->getConstructor();
586 if (!Callee.Decl)
587 return true;
588 processCall(Callee, {E->getArgs(), E->getNumArgs()});
589 return true;
592 bool VisitCallExpr(CallExpr *E) {
593 if (!Cfg.InlayHints.Parameters)
594 return true;
596 bool IsFunctor = isFunctionObjectCallExpr(E);
597 // Do not show parameter hints for user-defined literals or
598 // operator calls except for operator(). (Among other reasons, the resulting
599 // hints can look awkward, e.g. the expression can itself be a function
600 // argument and then we'd get two hints side by side).
601 if ((isa<CXXOperatorCallExpr>(E) && !IsFunctor) ||
602 isa<UserDefinedLiteral>(E))
603 return true;
605 auto CalleeDecls = Resolver->resolveCalleeOfCallExpr(E);
606 if (CalleeDecls.size() != 1)
607 return true;
609 Callee Callee;
610 if (const auto *FD = dyn_cast<FunctionDecl>(CalleeDecls[0]))
611 Callee.Decl = FD;
612 else if (const auto *FTD = dyn_cast<FunctionTemplateDecl>(CalleeDecls[0]))
613 Callee.Decl = FTD->getTemplatedDecl();
614 else if (FunctionProtoTypeLoc Loc = getPrototypeLoc(E->getCallee()))
615 Callee.Loc = Loc;
616 else
617 return true;
619 // N4868 [over.call.object]p3 says,
620 // The argument list submitted to overload resolution consists of the
621 // argument expressions present in the function call syntax preceded by the
622 // implied object argument (E).
624 // As well as the provision from P0847R7 Deducing This [expr.call]p7:
625 // ...If the function is an explicit object member function and there is an
626 // implied object argument ([over.call.func]), the list of provided
627 // arguments is preceded by the implied object argument for the purposes of
628 // this correspondence...
630 // However, we don't have the implied object argument
631 // for static operator() per clang::Sema::BuildCallToObjectOfClassType.
632 llvm::ArrayRef<const Expr *> Args = {E->getArgs(), E->getNumArgs()};
633 // We don't have the implied object argument through a function pointer
634 // either.
635 if (const CXXMethodDecl *Method =
636 dyn_cast_or_null<CXXMethodDecl>(Callee.Decl))
637 if (Method->isInstance() &&
638 (IsFunctor || Method->hasCXXExplicitFunctionObjectParameter()))
639 Args = Args.drop_front(1);
640 processCall(Callee, Args);
641 return true;
644 bool VisitFunctionDecl(FunctionDecl *D) {
645 if (auto *FPT =
646 llvm::dyn_cast<FunctionProtoType>(D->getType().getTypePtr())) {
647 if (!FPT->hasTrailingReturn()) {
648 if (auto FTL = D->getFunctionTypeLoc())
649 addReturnTypeHint(D, FTL.getRParenLoc());
652 if (Cfg.InlayHints.BlockEnd && D->isThisDeclarationADefinition()) {
653 // We use `printName` here to properly print name of ctor/dtor/operator
654 // overload.
655 if (const Stmt *Body = D->getBody())
656 addBlockEndHint(Body->getSourceRange(), "", printName(AST, *D), "");
658 return true;
661 bool VisitForStmt(ForStmt *S) {
662 if (Cfg.InlayHints.BlockEnd) {
663 std::string Name;
664 // Common case: for (int I = 0; I < N; I++). Use "I" as the name.
665 if (auto *DS = llvm::dyn_cast_or_null<DeclStmt>(S->getInit());
666 DS && DS->isSingleDecl())
667 Name = getSimpleName(llvm::cast<NamedDecl>(*DS->getSingleDecl()));
668 else
669 Name = summarizeExpr(S->getCond());
670 markBlockEnd(S->getBody(), "for", Name);
672 return true;
675 bool VisitCXXForRangeStmt(CXXForRangeStmt *S) {
676 if (Cfg.InlayHints.BlockEnd)
677 markBlockEnd(S->getBody(), "for", getSimpleName(*S->getLoopVariable()));
678 return true;
681 bool VisitWhileStmt(WhileStmt *S) {
682 if (Cfg.InlayHints.BlockEnd)
683 markBlockEnd(S->getBody(), "while", summarizeExpr(S->getCond()));
684 return true;
687 bool VisitSwitchStmt(SwitchStmt *S) {
688 if (Cfg.InlayHints.BlockEnd)
689 markBlockEnd(S->getBody(), "switch", summarizeExpr(S->getCond()));
690 return true;
693 // If/else chains are tricky.
694 // if (cond1) {
695 // } else if (cond2) {
696 // } // mark as "cond1" or "cond2"?
697 // For now, the answer is neither, just mark as "if".
698 // The ElseIf is a different IfStmt that doesn't know about the outer one.
699 llvm::DenseSet<const IfStmt *> ElseIfs; // not eligible for names
700 bool VisitIfStmt(IfStmt *S) {
701 if (Cfg.InlayHints.BlockEnd) {
702 if (const auto *ElseIf = llvm::dyn_cast_or_null<IfStmt>(S->getElse()))
703 ElseIfs.insert(ElseIf);
704 // Don't use markBlockEnd: the relevant range is [then.begin, else.end].
705 if (const auto *EndCS = llvm::dyn_cast<CompoundStmt>(
706 S->getElse() ? S->getElse() : S->getThen())) {
707 addBlockEndHint(
708 {S->getThen()->getBeginLoc(), EndCS->getRBracLoc()}, "if",
709 ElseIfs.contains(S) ? "" : summarizeExpr(S->getCond()), "");
712 return true;
715 void markBlockEnd(const Stmt *Body, llvm::StringRef Label,
716 llvm::StringRef Name = "") {
717 if (const auto *CS = llvm::dyn_cast_or_null<CompoundStmt>(Body))
718 addBlockEndHint(CS->getSourceRange(), Label, Name, "");
721 bool VisitTagDecl(TagDecl *D) {
722 if (Cfg.InlayHints.BlockEnd && D->isThisDeclarationADefinition()) {
723 std::string DeclPrefix = D->getKindName().str();
724 if (const auto *ED = dyn_cast<EnumDecl>(D)) {
725 if (ED->isScoped())
726 DeclPrefix += ED->isScopedUsingClassTag() ? " class" : " struct";
728 addBlockEndHint(D->getBraceRange(), DeclPrefix, getSimpleName(*D), ";");
730 return true;
733 bool VisitNamespaceDecl(NamespaceDecl *D) {
734 if (Cfg.InlayHints.BlockEnd) {
735 // For namespace, the range actually starts at the namespace keyword. But
736 // it should be fine since it's usually very short.
737 addBlockEndHint(D->getSourceRange(), "namespace", getSimpleName(*D), "");
739 return true;
742 bool VisitLambdaExpr(LambdaExpr *E) {
743 FunctionDecl *D = E->getCallOperator();
744 if (!E->hasExplicitResultType())
745 addReturnTypeHint(D, E->hasExplicitParameters()
746 ? D->getFunctionTypeLoc().getRParenLoc()
747 : E->getIntroducerRange().getEnd());
748 return true;
751 void addReturnTypeHint(FunctionDecl *D, SourceRange Range) {
752 auto *AT = D->getReturnType()->getContainedAutoType();
753 if (!AT || AT->getDeducedType().isNull())
754 return;
755 addTypeHint(Range, D->getReturnType(), /*Prefix=*/"-> ");
758 bool VisitVarDecl(VarDecl *D) {
759 // Do not show hints for the aggregate in a structured binding,
760 // but show hints for the individual bindings.
761 if (auto *DD = dyn_cast<DecompositionDecl>(D)) {
762 for (auto *Binding : DD->bindings()) {
763 // For structured bindings, print canonical types. This is important
764 // because for bindings that use the tuple_element protocol, the
765 // non-canonical types would be "tuple_element<I, A>::type".
766 if (auto Type = Binding->getType();
767 !Type.isNull() && !Type->isDependentType())
768 addTypeHint(Binding->getLocation(), Type.getCanonicalType(),
769 /*Prefix=*/": ");
771 return true;
774 if (auto *AT = D->getType()->getContainedAutoType()) {
775 if (AT->isDeduced() && !D->getType()->isDependentType()) {
776 // Our current approach is to place the hint on the variable
777 // and accordingly print the full type
778 // (e.g. for `const auto& x = 42`, print `const int&`).
779 // Alternatively, we could place the hint on the `auto`
780 // (and then just print the type deduced for the `auto`).
781 addTypeHint(D->getLocation(), D->getType(), /*Prefix=*/": ");
785 // Handle templates like `int foo(auto x)` with exactly one instantiation.
786 if (auto *PVD = llvm::dyn_cast<ParmVarDecl>(D)) {
787 if (D->getIdentifier() && PVD->getType()->isDependentType() &&
788 !getContainedAutoParamType(D->getTypeSourceInfo()->getTypeLoc())
789 .isNull()) {
790 if (auto *IPVD = getOnlyParamInstantiation(PVD))
791 addTypeHint(D->getLocation(), IPVD->getType(), /*Prefix=*/": ");
795 return true;
798 ParmVarDecl *getOnlyParamInstantiation(ParmVarDecl *D) {
799 auto *TemplateFunction = llvm::dyn_cast<FunctionDecl>(D->getDeclContext());
800 if (!TemplateFunction)
801 return nullptr;
802 auto *InstantiatedFunction = llvm::dyn_cast_or_null<FunctionDecl>(
803 getOnlyInstantiation(TemplateFunction));
804 if (!InstantiatedFunction)
805 return nullptr;
807 unsigned ParamIdx = 0;
808 for (auto *Param : TemplateFunction->parameters()) {
809 // Can't reason about param indexes in the presence of preceding packs.
810 // And if this param is a pack, it may expand to multiple params.
811 if (Param->isParameterPack())
812 return nullptr;
813 if (Param == D)
814 break;
815 ++ParamIdx;
817 assert(ParamIdx < TemplateFunction->getNumParams() &&
818 "Couldn't find param in list?");
819 assert(ParamIdx < InstantiatedFunction->getNumParams() &&
820 "Instantiated function has fewer (non-pack) parameters?");
821 return InstantiatedFunction->getParamDecl(ParamIdx);
824 bool VisitInitListExpr(InitListExpr *Syn) {
825 // We receive the syntactic form here (shouldVisitImplicitCode() is false).
826 // This is the one we will ultimately attach designators to.
827 // It may have subobject initializers inlined without braces. The *semantic*
828 // form of the init-list has nested init-lists for these.
829 // getDesignators will look at the semantic form to determine the labels.
830 assert(Syn->isSyntacticForm() && "RAV should not visit implicit code!");
831 if (!Cfg.InlayHints.Designators)
832 return true;
833 if (Syn->isIdiomaticZeroInitializer(AST.getLangOpts()))
834 return true;
835 llvm::DenseMap<SourceLocation, std::string> Designators =
836 getDesignators(Syn);
837 for (const Expr *Init : Syn->inits()) {
838 if (llvm::isa<DesignatedInitExpr>(Init))
839 continue;
840 auto It = Designators.find(Init->getBeginLoc());
841 if (It != Designators.end() &&
842 !isPrecededByParamNameComment(Init, It->second))
843 addDesignatorHint(Init->getSourceRange(), It->second);
845 return true;
848 // FIXME: Handle RecoveryExpr to try to hint some invalid calls.
850 private:
851 using NameVec = SmallVector<StringRef, 8>;
853 void processCall(Callee Callee, llvm::ArrayRef<const Expr *> Args) {
854 assert(Callee.Decl || Callee.Loc);
856 if (!Cfg.InlayHints.Parameters || Args.size() == 0)
857 return;
859 // The parameter name of a move or copy constructor is not very interesting.
860 if (Callee.Decl)
861 if (auto *Ctor = dyn_cast<CXXConstructorDecl>(Callee.Decl))
862 if (Ctor->isCopyOrMoveConstructor())
863 return;
865 ArrayRef<const ParmVarDecl *> Params, ForwardedParams;
866 // Resolve parameter packs to their forwarded parameter
867 SmallVector<const ParmVarDecl *> ForwardedParamsStorage;
868 if (Callee.Decl) {
869 Params = maybeDropCxxExplicitObjectParameters(Callee.Decl->parameters());
870 ForwardedParamsStorage = resolveForwardingParameters(Callee.Decl);
871 ForwardedParams =
872 maybeDropCxxExplicitObjectParameters(ForwardedParamsStorage);
873 } else {
874 Params = maybeDropCxxExplicitObjectParameters(Callee.Loc.getParams());
875 ForwardedParams = {Params.begin(), Params.end()};
878 NameVec ParameterNames = chooseParameterNames(ForwardedParams);
880 // Exclude setters (i.e. functions with one argument whose name begins with
881 // "set"), and builtins like std::move/forward/... as their parameter name
882 // is also not likely to be interesting.
883 if (Callee.Decl &&
884 (isSetter(Callee.Decl, ParameterNames) || isSimpleBuiltin(Callee.Decl)))
885 return;
887 for (size_t I = 0; I < ParameterNames.size() && I < Args.size(); ++I) {
888 // Pack expansion expressions cause the 1:1 mapping between arguments and
889 // parameters to break down, so we don't add further inlay hints if we
890 // encounter one.
891 if (isa<PackExpansionExpr>(Args[I])) {
892 break;
895 StringRef Name = ParameterNames[I];
896 bool NameHint = shouldHintName(Args[I], Name);
897 bool ReferenceHint = shouldHintReference(Params[I], ForwardedParams[I]);
899 if (NameHint || ReferenceHint) {
900 addInlayHint(Args[I]->getSourceRange(), HintSide::Left,
901 InlayHintKind::Parameter, ReferenceHint ? "&" : "",
902 NameHint ? Name : "", ": ");
907 static bool isSetter(const FunctionDecl *Callee, const NameVec &ParamNames) {
908 if (ParamNames.size() != 1)
909 return false;
911 StringRef Name = getSimpleName(*Callee);
912 if (!Name.starts_with_insensitive("set"))
913 return false;
915 // In addition to checking that the function has one parameter and its
916 // name starts with "set", also check that the part after "set" matches
917 // the name of the parameter (ignoring case). The idea here is that if
918 // the parameter name differs, it may contain extra information that
919 // may be useful to show in a hint, as in:
920 // void setTimeout(int timeoutMillis);
921 // This currently doesn't handle cases where params use snake_case
922 // and functions don't, e.g.
923 // void setExceptionHandler(EHFunc exception_handler);
924 // We could improve this by replacing `equals_insensitive` with some
925 // `sloppy_equals` which ignores case and also skips underscores.
926 StringRef WhatItIsSetting = Name.substr(3).ltrim("_");
927 return WhatItIsSetting.equals_insensitive(ParamNames[0]);
930 // Checks if the callee is one of the builtins
931 // addressof, as_const, forward, move(_if_noexcept)
932 static bool isSimpleBuiltin(const FunctionDecl *Callee) {
933 switch (Callee->getBuiltinID()) {
934 case Builtin::BIaddressof:
935 case Builtin::BIas_const:
936 case Builtin::BIforward:
937 case Builtin::BImove:
938 case Builtin::BImove_if_noexcept:
939 return true;
940 default:
941 return false;
945 bool shouldHintName(const Expr *Arg, StringRef ParamName) {
946 if (ParamName.empty())
947 return false;
949 // If the argument expression is a single name and it matches the
950 // parameter name exactly, omit the name hint.
951 if (ParamName == getSpelledIdentifier(Arg))
952 return false;
954 // Exclude argument expressions preceded by a /*paramName*/.
955 if (isPrecededByParamNameComment(Arg, ParamName))
956 return false;
958 return true;
961 bool shouldHintReference(const ParmVarDecl *Param,
962 const ParmVarDecl *ForwardedParam) {
963 // We add a & hint only when the argument is passed as mutable reference.
964 // For parameters that are not part of an expanded pack, this is
965 // straightforward. For expanded pack parameters, it's likely that they will
966 // be forwarded to another function. In this situation, we only want to add
967 // the reference hint if the argument is actually being used via mutable
968 // reference. This means we need to check
969 // 1. whether the value category of the argument is preserved, i.e. each
970 // pack expansion uses std::forward correctly.
971 // 2. whether the argument is ever copied/cast instead of passed
972 // by-reference
973 // Instead of checking this explicitly, we use the following proxy:
974 // 1. the value category can only change from rvalue to lvalue during
975 // forwarding, so checking whether both the parameter of the forwarding
976 // function and the forwarded function are lvalue references detects such
977 // a conversion.
978 // 2. if the argument is copied/cast somewhere in the chain of forwarding
979 // calls, it can only be passed on to an rvalue reference or const lvalue
980 // reference parameter. Thus if the forwarded parameter is a mutable
981 // lvalue reference, it cannot have been copied/cast to on the way.
982 // Additionally, we should not add a reference hint if the forwarded
983 // parameter was only partially resolved, i.e. points to an expanded pack
984 // parameter, since we do not know how it will be used eventually.
985 auto Type = Param->getType();
986 auto ForwardedType = ForwardedParam->getType();
987 return Type->isLValueReferenceType() &&
988 ForwardedType->isLValueReferenceType() &&
989 !ForwardedType.getNonReferenceType().isConstQualified() &&
990 !isExpandedFromParameterPack(ForwardedParam);
993 // Checks if "E" is spelled in the main file and preceded by a C-style comment
994 // whose contents match ParamName (allowing for whitespace and an optional "="
995 // at the end.
996 bool isPrecededByParamNameComment(const Expr *E, StringRef ParamName) {
997 auto &SM = AST.getSourceManager();
998 auto FileLoc = SM.getFileLoc(E->getBeginLoc());
999 auto Decomposed = SM.getDecomposedLoc(FileLoc);
1000 if (Decomposed.first != MainFileID)
1001 return false;
1003 StringRef SourcePrefix = MainFileBuf.substr(0, Decomposed.second);
1004 // Allow whitespace between comment and expression.
1005 SourcePrefix = SourcePrefix.rtrim();
1006 // Check for comment ending.
1007 if (!SourcePrefix.consume_back("*/"))
1008 return false;
1009 // Ignore some punctuation and whitespace around comment.
1010 // In particular this allows designators to match nicely.
1011 llvm::StringLiteral IgnoreChars = " =.";
1012 SourcePrefix = SourcePrefix.rtrim(IgnoreChars);
1013 ParamName = ParamName.trim(IgnoreChars);
1014 // Other than that, the comment must contain exactly ParamName.
1015 if (!SourcePrefix.consume_back(ParamName))
1016 return false;
1017 SourcePrefix = SourcePrefix.rtrim(IgnoreChars);
1018 return SourcePrefix.endswith("/*");
1021 // If "E" spells a single unqualified identifier, return that name.
1022 // Otherwise, return an empty string.
1023 static StringRef getSpelledIdentifier(const Expr *E) {
1024 E = E->IgnoreUnlessSpelledInSource();
1026 if (auto *DRE = dyn_cast<DeclRefExpr>(E))
1027 if (!DRE->getQualifier())
1028 return getSimpleName(*DRE->getDecl());
1030 if (auto *ME = dyn_cast<MemberExpr>(E))
1031 if (!ME->getQualifier() && ME->isImplicitAccess())
1032 return getSimpleName(*ME->getMemberDecl());
1034 return {};
1037 NameVec chooseParameterNames(ArrayRef<const ParmVarDecl *> Parameters) {
1038 NameVec ParameterNames;
1039 for (const auto *P : Parameters) {
1040 if (isExpandedFromParameterPack(P)) {
1041 // If we haven't resolved a pack paramater (e.g. foo(Args... args)) to a
1042 // non-pack parameter, then hinting as foo(args: 1, args: 2, args: 3) is
1043 // unlikely to be useful.
1044 ParameterNames.emplace_back();
1045 } else {
1046 auto SimpleName = getSimpleName(*P);
1047 // If the parameter is unnamed in the declaration:
1048 // attempt to get its name from the definition
1049 if (SimpleName.empty()) {
1050 if (const auto *PD = getParamDefinition(P)) {
1051 SimpleName = getSimpleName(*PD);
1054 ParameterNames.emplace_back(SimpleName);
1058 // Standard library functions often have parameter names that start
1059 // with underscores, which makes the hints noisy, so strip them out.
1060 for (auto &Name : ParameterNames)
1061 stripLeadingUnderscores(Name);
1063 return ParameterNames;
1066 // for a ParmVarDecl from a function declaration, returns the corresponding
1067 // ParmVarDecl from the definition if possible, nullptr otherwise.
1068 static const ParmVarDecl *getParamDefinition(const ParmVarDecl *P) {
1069 if (auto *Callee = dyn_cast<FunctionDecl>(P->getDeclContext())) {
1070 if (auto *Def = Callee->getDefinition()) {
1071 auto I = std::distance(Callee->param_begin(),
1072 llvm::find(Callee->parameters(), P));
1073 if (I < Callee->getNumParams()) {
1074 return Def->getParamDecl(I);
1078 return nullptr;
1081 // We pass HintSide rather than SourceLocation because we want to ensure
1082 // it is in the same file as the common file range.
1083 void addInlayHint(SourceRange R, HintSide Side, InlayHintKind Kind,
1084 llvm::StringRef Prefix, llvm::StringRef Label,
1085 llvm::StringRef Suffix) {
1086 auto LSPRange = getHintRange(R);
1087 if (!LSPRange)
1088 return;
1090 addInlayHint(*LSPRange, Side, Kind, Prefix, Label, Suffix);
1093 void addInlayHint(Range LSPRange, HintSide Side, InlayHintKind Kind,
1094 llvm::StringRef Prefix, llvm::StringRef Label,
1095 llvm::StringRef Suffix) {
1096 // We shouldn't get as far as adding a hint if the category is disabled.
1097 // We'd like to disable as much of the analysis as possible above instead.
1098 // Assert in debug mode but add a dynamic check in production.
1099 assert(Cfg.InlayHints.Enabled && "Shouldn't get here if disabled!");
1100 switch (Kind) {
1101 #define CHECK_KIND(Enumerator, ConfigProperty) \
1102 case InlayHintKind::Enumerator: \
1103 assert(Cfg.InlayHints.ConfigProperty && \
1104 "Shouldn't get here if kind is disabled!"); \
1105 if (!Cfg.InlayHints.ConfigProperty) \
1106 return; \
1107 break
1108 CHECK_KIND(Parameter, Parameters);
1109 CHECK_KIND(Type, DeducedTypes);
1110 CHECK_KIND(Designator, Designators);
1111 CHECK_KIND(BlockEnd, BlockEnd);
1112 #undef CHECK_KIND
1115 Position LSPPos = Side == HintSide::Left ? LSPRange.start : LSPRange.end;
1116 if (RestrictRange &&
1117 (LSPPos < RestrictRange->start || !(LSPPos < RestrictRange->end)))
1118 return;
1119 bool PadLeft = Prefix.consume_front(" ");
1120 bool PadRight = Suffix.consume_back(" ");
1121 Results.push_back(InlayHint{LSPPos, (Prefix + Label + Suffix).str(), Kind,
1122 PadLeft, PadRight, LSPRange});
1125 // Get the range of the main file that *exactly* corresponds to R.
1126 std::optional<Range> getHintRange(SourceRange R) {
1127 const auto &SM = AST.getSourceManager();
1128 auto Spelled = Tokens.spelledForExpanded(Tokens.expandedTokens(R));
1129 // TokenBuffer will return null if e.g. R corresponds to only part of a
1130 // macro expansion.
1131 if (!Spelled || Spelled->empty())
1132 return std::nullopt;
1133 // Hint must be within the main file, not e.g. a non-preamble include.
1134 if (SM.getFileID(Spelled->front().location()) != SM.getMainFileID() ||
1135 SM.getFileID(Spelled->back().location()) != SM.getMainFileID())
1136 return std::nullopt;
1137 return Range{sourceLocToPosition(SM, Spelled->front().location()),
1138 sourceLocToPosition(SM, Spelled->back().endLocation())};
1141 void addTypeHint(SourceRange R, QualType T, llvm::StringRef Prefix) {
1142 if (!Cfg.InlayHints.DeducedTypes || T.isNull())
1143 return;
1145 // The sugared type is more useful in some cases, and the canonical
1146 // type in other cases.
1147 auto Desugared = maybeDesugar(AST, T);
1148 std::string TypeName = Desugared.getAsString(TypeHintPolicy);
1149 if (T != Desugared && !shouldPrintTypeHint(TypeName)) {
1150 // If the desugared type is too long to display, fallback to the sugared
1151 // type.
1152 TypeName = T.getAsString(TypeHintPolicy);
1154 if (shouldPrintTypeHint(TypeName))
1155 addInlayHint(R, HintSide::Right, InlayHintKind::Type, Prefix, TypeName,
1156 /*Suffix=*/"");
1159 void addDesignatorHint(SourceRange R, llvm::StringRef Text) {
1160 addInlayHint(R, HintSide::Left, InlayHintKind::Designator,
1161 /*Prefix=*/"", Text, /*Suffix=*/"=");
1164 bool shouldPrintTypeHint(llvm::StringRef TypeName) const noexcept {
1165 return Cfg.InlayHints.TypeNameLimit == 0 ||
1166 TypeName.size() < Cfg.InlayHints.TypeNameLimit;
1169 void addBlockEndHint(SourceRange BraceRange, StringRef DeclPrefix,
1170 StringRef Name, StringRef OptionalPunctuation) {
1171 auto HintRange = computeBlockEndHintRange(BraceRange, OptionalPunctuation);
1172 if (!HintRange)
1173 return;
1175 std::string Label = DeclPrefix.str();
1176 if (!Label.empty() && !Name.empty())
1177 Label += ' ';
1178 Label += Name;
1180 constexpr unsigned HintMaxLengthLimit = 60;
1181 if (Label.length() > HintMaxLengthLimit)
1182 return;
1184 addInlayHint(*HintRange, HintSide::Right, InlayHintKind::BlockEnd, " // ",
1185 Label, "");
1188 // Compute the LSP range to attach the block end hint to, if any allowed.
1189 // 1. "}" is the last non-whitespace character on the line. The range of "}"
1190 // is returned.
1191 // 2. After "}", if the trimmed trailing text is exactly
1192 // `OptionalPunctuation`, say ";". The range of "} ... ;" is returned.
1193 // Otherwise, the hint shouldn't be shown.
1194 std::optional<Range> computeBlockEndHintRange(SourceRange BraceRange,
1195 StringRef OptionalPunctuation) {
1196 constexpr unsigned HintMinLineLimit = 2;
1198 auto &SM = AST.getSourceManager();
1199 auto [BlockBeginFileId, BlockBeginOffset] =
1200 SM.getDecomposedLoc(SM.getFileLoc(BraceRange.getBegin()));
1201 auto RBraceLoc = SM.getFileLoc(BraceRange.getEnd());
1202 auto [RBraceFileId, RBraceOffset] = SM.getDecomposedLoc(RBraceLoc);
1204 // Because we need to check the block satisfies the minimum line limit, we
1205 // require both source location to be in the main file. This prevents hint
1206 // to be shown in weird cases like '{' is actually in a "#include", but it's
1207 // rare anyway.
1208 if (BlockBeginFileId != MainFileID || RBraceFileId != MainFileID)
1209 return std::nullopt;
1211 StringRef RestOfLine = MainFileBuf.substr(RBraceOffset).split('\n').first;
1212 if (!RestOfLine.starts_with("}"))
1213 return std::nullopt;
1215 StringRef TrimmedTrailingText = RestOfLine.drop_front().trim();
1216 if (!TrimmedTrailingText.empty() &&
1217 TrimmedTrailingText != OptionalPunctuation)
1218 return std::nullopt;
1220 auto BlockBeginLine = SM.getLineNumber(BlockBeginFileId, BlockBeginOffset);
1221 auto RBraceLine = SM.getLineNumber(RBraceFileId, RBraceOffset);
1223 // Don't show hint on trivial blocks like `class X {};`
1224 if (BlockBeginLine + HintMinLineLimit - 1 > RBraceLine)
1225 return std::nullopt;
1227 // This is what we attach the hint to, usually "}" or "};".
1228 StringRef HintRangeText = RestOfLine.take_front(
1229 TrimmedTrailingText.empty()
1231 : TrimmedTrailingText.bytes_end() - RestOfLine.bytes_begin());
1233 Position HintStart = sourceLocToPosition(SM, RBraceLoc);
1234 Position HintEnd = sourceLocToPosition(
1235 SM, RBraceLoc.getLocWithOffset(HintRangeText.size()));
1236 return Range{HintStart, HintEnd};
1239 static bool isFunctionObjectCallExpr(CallExpr *E) noexcept {
1240 if (auto *CallExpr = dyn_cast<CXXOperatorCallExpr>(E))
1241 return CallExpr->getOperator() == OverloadedOperatorKind::OO_Call;
1242 return false;
1245 std::vector<InlayHint> &Results;
1246 ASTContext &AST;
1247 const syntax::TokenBuffer &Tokens;
1248 const Config &Cfg;
1249 std::optional<Range> RestrictRange;
1250 FileID MainFileID;
1251 StringRef MainFileBuf;
1252 const HeuristicResolver *Resolver;
1253 PrintingPolicy TypeHintPolicy;
1256 } // namespace
1258 std::vector<InlayHint> inlayHints(ParsedAST &AST,
1259 std::optional<Range> RestrictRange) {
1260 std::vector<InlayHint> Results;
1261 const auto &Cfg = Config::current();
1262 if (!Cfg.InlayHints.Enabled)
1263 return Results;
1264 InlayHintVisitor Visitor(Results, AST, Cfg, std::move(RestrictRange));
1265 Visitor.TraverseAST(AST.getASTContext());
1267 // De-duplicate hints. Duplicates can sometimes occur due to e.g. explicit
1268 // template instantiations.
1269 llvm::sort(Results);
1270 Results.erase(std::unique(Results.begin(), Results.end()), Results.end());
1272 return Results;
1275 } // namespace clangd
1276 } // namespace clang