Merge pull request #22825 from slavapestov/remove-old-type-reconstruction

Remove TypeReconstruction.cpp

Author: slavapestov

include/swift/AST/ASTDemangler.h

@@ -30,11 +30,19 @@
 
 namespace swift {
 
+class TypeDecl;
+
 namespace Demangle {
 
 Type getTypeForMangling(ASTContext &ctx,
                         llvm::StringRef mangling);
 
+TypeDecl *getTypeDeclForMangling(ASTContext &ctx,
+                                 llvm::StringRef mangling);
+
+TypeDecl *getTypeDeclForUSR(ASTContext &ctx,
+                            llvm::StringRef usr);
+
 /// An implementation of MetadataReader's BuilderType concept that
 /// just finds and builds things in the AST.
 class ASTBuilder {
@@ -58,6 +66,8 @@ class ASTBuilder {
 
   Type createBuiltinType(StringRef builtinName, StringRef mangledName);
 
+  TypeDecl *createTypeDecl(NodePointer node);
+
   GenericTypeDecl *createTypeDecl(StringRef mangledName, bool &typeAlias);
 
   GenericTypeDecl *createTypeDecl(NodePointer node,

include/swift/IDE/Utils.h

@@ -130,10 +130,6 @@ void getLocationInfoForClangNode(ClangNode ClangNode,
 
 Optional<std::pair<unsigned, unsigned>> parseLineCol(StringRef LineCol);
 
-Decl *getDeclFromUSR(ASTContext &context, StringRef USR, std::string &error);
-Decl *getDeclFromMangledSymbolName(ASTContext &context, StringRef mangledName,
-                                   std::string &error);
-
 class XMLEscapingPrinter : public StreamPrinter {
   public:
   XMLEscapingPrinter(raw_ostream &OS) : StreamPrinter(OS){};

lib/AST/ASTDemangler.cpp

@@ -31,6 +31,7 @@
 #include "swift/AST/Types.h"
 #include "swift/ClangImporter/ClangImporter.h"
 #include "swift/Demangling/Demangler.h"
+#include "swift/Demangling/ManglingMacros.h"
 
 using namespace swift;
 
@@ -45,6 +46,53 @@ Type swift::Demangle::getTypeForMangling(ASTContext &ctx,
   return swift::Demangle::decodeMangledType(builder, node);
 }
 
+TypeDecl *swift::Demangle::getTypeDeclForMangling(ASTContext &ctx,
+                                                  StringRef mangling) {
+  Demangle::Context Dem;
+  auto node = Dem.demangleSymbolAsNode(mangling);
+  if (!node)
+    return nullptr;
+
+  ASTBuilder builder(ctx);
+  return builder.createTypeDecl(node);
+}
+
+TypeDecl *swift::Demangle::getTypeDeclForUSR(ASTContext &ctx,
+                                             StringRef usr) {
+  if (!usr.startswith("s:"))
+    return nullptr;
+
+  std::string mangling(usr);
+  mangling.replace(0, 2, MANGLING_PREFIX_STR);
+
+  return getTypeDeclForMangling(ctx, mangling);
+}
+
+TypeDecl *ASTBuilder::createTypeDecl(NodePointer node) {
+  if (node->getKind() == Node::Kind::Global)
+    return createTypeDecl(node->getChild(0));
+
+  // Special case: associated types are not DeclContexts.
+  if (node->getKind() == Node::Kind::AssociatedTypeRef) {
+    if (node->getNumChildren() != 2)
+      return nullptr;
+
+    auto *DC = findDeclContext(node->getChild(0));
+    auto *proto = dyn_cast_or_null<ProtocolDecl>(DC);
+    if (proto == nullptr)
+      return nullptr;
+
+    auto name = Ctx.getIdentifier(node->getChild(1)->getText());
+    auto results = proto->lookupDirect(name);
+    if (results.size() != 1)
+      return nullptr;
+
+    return dyn_cast<AssociatedTypeDecl>(results[0]);
+  }
+
+  auto *DC = findDeclContext(node);
+  return dyn_cast_or_null<GenericTypeDecl>(DC);
+}
 
 Type
 ASTBuilder::createBuiltinType(StringRef builtinName,

lib/IDE/CMakeLists.txt

@@ -7,7 +7,6 @@ add_swift_host_library(swiftIDE STATIC
   Formatting.cpp
   Refactoring.cpp
   ModuleInterfacePrinting.cpp
-  TypeReconstruction.cpp
   REPLCodeCompletion.cpp
   SwiftSourceDocInfo.cpp
   SyntaxModel.cpp