Files
swift-mirror/lib/SymbolGraphGen/Edge.cpp
Nathan Hawes 388052b6ab [SymbolGraph][CursorInfo] Add option to SourceKit's CursorInfo request to include the SymbolGraph JSON
Adds a new 'key.retrieve_symbol_graph' option to the request. When set to 1 it
includes the JSON for a SymbolGraph containing a single node for the symbol at
the requested position.

This also extends the SymbolGraph library with a new entry point to get a graph
for a single symbol, and to additionally support type substitution to match the
existing CursorInfo behavior (e.g. so that when invoked on `first` in
`Array<Int>().first`, the type is given as `Int?` rather than `Element?`).

Resolves rdar://problem/70551509
2020-12-12 14:38:41 +10:00

62 lines
2.0 KiB
C++

//===--- Edge.cpp - Symbol Graph Edge -------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
#include "swift/AST/Module.h"
#include "Edge.h"
#include "Symbol.h"
#include "SymbolGraphASTWalker.h"
using namespace swift;
using namespace symbolgraphgen;
void Edge::serialize(llvm::json::OStream &OS) const {
OS.object([&](){
OS.attribute("kind", Kind.Name);
SmallString<256> SourceUSR, TargetUSR;
Source.getUSR(SourceUSR);
OS.attribute("source", SourceUSR.str());
Target.getUSR(TargetUSR);
OS.attribute("target", TargetUSR.str());
// In case a dependent module isn't available, serialize a fallback name.
auto TargetModuleName = Target.getSymbolDecl()
->getModuleContext()->getName().str();
if (TargetModuleName != Graph->M.getName().str()) {
SmallString<128> Scratch(TargetModuleName);
llvm::raw_svector_ostream PathOS(Scratch);
PathOS << '.';
Target.printPath(PathOS);
OS.attribute("targetFallback", Scratch.str());
}
if (ConformanceExtension) {
SmallVector<Requirement, 4> FilteredRequirements;
filterGenericRequirements(
ConformanceExtension->getGenericRequirements(),
ConformanceExtension->getExtendedNominal()
->getDeclContext()->getSelfNominalTypeDecl(),
FilteredRequirements);
if (!FilteredRequirements.empty()) {
OS.attributeArray("swiftConstraints", [&](){
for (const auto &Req :
ConformanceExtension->getGenericRequirements()) {
::serialize(Req, OS);
}
});
}
}
});
}