hokein updated this revision to Diff 423892.
hokein added a comment.
implement a pseudo-VisitTemplateName locally.
Repository:
rG LLVM Github Monorepo
CHANGES SINCE LAST ACTION
https://reviews.llvm.org/D123212/new/
https://reviews.llvm.org/D123212
Files:
clang-tools-extra/clangd/FindTarget.cpp
clang-tools-extra/clangd/IncludeCleaner.cpp
clang-tools-extra/clangd/unittests/FindTargetTests.cpp
clang-tools-extra/clangd/unittests/IncludeCleanerTests.cpp
Index: clang-tools-extra/clangd/unittests/IncludeCleanerTests.cpp
===================================================================
--- clang-tools-extra/clangd/unittests/IncludeCleanerTests.cpp
+++ clang-tools-extra/clangd/unittests/IncludeCleanerTests.cpp
@@ -79,9 +79,22 @@
"using namespace ns;",
},
{
+ // Refs from UsingTypeLoc and implicit constructor!
"struct ^A {}; using B = A; using ^C = B;",
"C a;",
},
+ {"namespace ns { template<typename T> class A {}; } using ns::^A;",
+ "A<int>* a;"},
+ {"namespace ns { template<typename T> class A {}; } using ns::^A;",
+ R"cpp(
+ template <template <typename> class T> class X {};
+ X<A> x;
+ )cpp"},
+ {R"cpp(
+ namespace ns { template<typename T> struct ^A { ^A(T); }; }
+ using ns::^A;
+ )cpp",
+ "A CATD(123);"},
{
"typedef bool ^Y; template <typename T> struct ^X {};",
"X<Y> x;",
@@ -227,6 +240,7 @@
TU.Code = T.MainCode;
Annotations Header(T.HeaderCode);
TU.HeaderCode = Header.code().str();
+ TU.ExtraArgs.push_back("-std=c++17");
auto AST = TU.build();
std::vector<Position> Points;
Index: clang-tools-extra/clangd/unittests/FindTargetTests.cpp
===================================================================
--- clang-tools-extra/clangd/unittests/FindTargetTests.cpp
+++ clang-tools-extra/clangd/unittests/FindTargetTests.cpp
@@ -229,6 +229,45 @@
)cpp";
EXPECT_DECLS("UnresolvedUsingValueDecl", {"using Base<T>::waldo", Rel::Alias},
{"void waldo()"});
+
+ Code = R"cpp(
+ namespace ns {
+ template<typename T> class S {};
+ }
+
+ using ns::S;
+
+ template<typename T>
+ using A = [[S]]<T>;
+ )cpp";
+ EXPECT_DECLS("TemplateSpecializationTypeLoc", {"using ns::S", Rel::Alias},
+ {"template <typename T> class S"},
+ {"class S", Rel::TemplatePattern});
+
+ Code = R"cpp(
+ namespace ns {
+ template<typename T> class S {};
+ }
+
+ using ns::S;
+ template <template <typename> class T> class X {};
+ using B = X<[[S]]>;
+ )cpp";
+ EXPECT_DECLS("TemplateArgumentLoc", {"using ns::S", Rel::Alias},
+ {"template <typename T> class S"});
+
+ Code = R"cpp(
+ namespace ns {
+ template<typename T> class S { public: S(T); };
+ }
+
+ using ns::S;
+ [[S]] s(123);
+ )cpp";
+ Flags.push_back("-std=c++17"); // For CTAD feature.
+ EXPECT_DECLS("DeducedTemplateSpecializationTypeLoc",
+ {"using ns::S", Rel::Alias}, {"template <typename T> class S"},
+ {"class S", Rel::TemplatePattern});
}
TEST_F(TargetDeclTest, BaseSpecifier) {
Index: clang-tools-extra/clangd/IncludeCleaner.cpp
===================================================================
--- clang-tools-extra/clangd/IncludeCleaner.cpp
+++ clang-tools-extra/clangd/IncludeCleaner.cpp
@@ -79,11 +79,29 @@
}
bool VisitTemplateSpecializationType(TemplateSpecializationType *TST) {
- add(TST->getTemplateName().getAsTemplateDecl()); // Primary template.
+ // Using templateName case is handled by the override TraverseTemplateName.
+ if (TST->getTemplateName().getKind() == TemplateName::UsingTemplate)
+ return true;
add(TST->getAsCXXRecordDecl()); // Specialization
return true;
}
+ // There is no VisitTemplateName in RAV, thus we override the Traverse version
+ // to handle the Using TemplateName case.
+ bool TraverseTemplateName(TemplateName TN) {
+ VisitTemplateName(TN);
+ return Base::TraverseTemplateName(TN);
+ }
+ // A pseudo VisitTemplateName, dispatched by the above TraverseTemplateName!
+ bool VisitTemplateName(TemplateName TN) {
+ if (const auto *USD = TN.getAsUsingShadowDecl()) {
+ add(USD);
+ return true;
+ }
+ add(TN.getAsTemplateDecl()); // Primary template.
+ return true;
+ }
+
bool VisitUsingType(UsingType *UT) {
add(UT->getFoundDecl());
return true;
Index: clang-tools-extra/clangd/FindTarget.cpp
===================================================================
--- clang-tools-extra/clangd/FindTarget.cpp
+++ clang-tools-extra/clangd/FindTarget.cpp
@@ -384,11 +384,14 @@
}
void VisitDeducedTemplateSpecializationType(
const DeducedTemplateSpecializationType *DTST) {
+ if (const auto *USD = DTST->getTemplateName().getAsUsingShadowDecl())
+ Outer.add(USD, Flags);
+
// FIXME: This is a workaround for https://llvm.org/PR42914,
// which is causing DTST->getDeducedType() to be empty. We
// fall back to the template pattern and miss the instantiation
// even when it's known in principle. Once that bug is fixed,
- // this method can be removed (the existing handling in
+ // the following code can be removed (the existing handling in
// VisitDeducedType() is sufficient).
if (auto *TD = DTST->getTemplateName().getAsTemplateDecl())
Outer.add(TD->getTemplatedDecl(), Flags | Rel::TemplatePattern);
@@ -419,6 +422,9 @@
VisitTemplateSpecializationType(const TemplateSpecializationType *TST) {
// Have to handle these case-by-case.
+ if (const auto *UTN = TST->getTemplateName().getAsUsingShadowDecl())
+ Outer.add(UTN, Flags);
+
// templated type aliases: there's no specialized/instantiated using
// decl to point to. So try to find a decl for the underlying type
// (after substitution), and failing that point to the (templated) using
@@ -508,6 +514,9 @@
Arg.getAsTemplateOrTemplatePattern().getAsTemplateDecl()) {
report(TD, Flags);
}
+ if (const auto *USD =
+ Arg.getAsTemplateOrTemplatePattern().getAsUsingShadowDecl())
+ add(USD, Flags);
}
}
};
_______________________________________________
cfe-commits mailing list
cfe-commits@lists.llvm.org
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
