averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-21 12:14:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

AnnotatingPrinter: For members from a synthesized extension, we append the target's USR to their original USR to differentiate them.

Browse Source 
This is required to jumping to definition correctly, NFC for now.
This commit is contained in:
Xi Ge
2016-02-16 11:33:26 -08:00
parent b7a415bade
commit b480ab1f5f
1 changed files with 30 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
tools/SourceKit/lib/SwiftLang/SwiftEditorInterfaceGen.cpp
View File

@@ -112,6 +112,13 @@ class AnnotatingPrinter : public StreamPrinter {
};
std::vector<DeclUSR> DeclUSRs;
// For members of a synthesized extension, we should append the USR of the
// synthesize target to the original USR.
std::string TargetUSR;
// A separator between the target usr and the real usr.
std::string Separator = "::SYNTHESIZED::";
public:
AnnotatingPrinter(SourceTextInfo &Info, llvm::raw_ostream &OS)
: StreamPrinter(OS), Info(Info) { }
@@ -120,6 +127,23 @@ public:
assert(DeclUSRs.empty() && "unmatched printDeclLoc call ?");
}
void printSynthesizedExtensionPre(const ExtensionDecl *ED,
const NominalTypeDecl *Target) override {
// When we start print a synthesized extension, record the target's USR.
llvm::SmallString<64> Buf;
llvm::raw_svector_ostream OS(Buf);
if (!SwiftLangSupport::printUSR(Target, OS)) {
TargetUSR = OS.str();
}
}
void printSynthesizedExtensionPost(const ExtensionDecl *ED,
const NominalTypeDecl *Target) override {
// When we leave a synthesized extension, clear target's USR.
TargetUSR = "";
}
void printDeclLoc(const Decl *D) override {
unsigned LocOffset = OS.tell();
TextDecl Entry(D, TextRange{LocOffset, 0});
@@ -133,6 +157,12 @@ public:
llvm::SmallString<64> Buf;
llvm::raw_svector_ostream OS(Buf);
if (!SwiftLangSupport::printUSR(VD, OS)) {
// Append target's USR if this is a member of a synthesized extension.
if (!TargetUSR.empty()) {
OS << Separator;
OS << TargetUSR;
}
StringRef USR = OS.str();
Info.USRMap[USR] = Entry;
DeclUSRs.emplace_back(VD, USR);