Implement LifetimeDependentReturnTypeRepr::printImpl

Author: meg-gupta

include/swift/AST/LifetimeDependence.h

@@ -104,6 +104,21 @@ class LifetimeDependenceSpecifier {
     }
     llvm_unreachable("Invalid LifetimeDependenceSpecifier::SpecifierKind");
   }
+
+  StringRef getLifetimeDependenceKindString() const {
+    switch (lifetimeDependenceKind) {
+    case LifetimeDependenceKind::Borrow:
+      return "_borrow";
+    case LifetimeDependenceKind::Consume:
+      return "_consume";
+    case LifetimeDependenceKind::Copy:
+      return "_copy";
+    case LifetimeDependenceKind::Mutate:
+      return "_mutate";
+    }
+    llvm_unreachable(
+        "Invalid LifetimeDependenceSpecifier::LifetimeDependenceKind");
+  }
 };
 } // namespace swift
 

lib/AST/ASTDumper.cpp

@@ -418,6 +418,7 @@ static StringRef getDumpString(ValueOwnership ownership) {
 
   llvm_unreachable("Unhandled ValueOwnership in switch.");
 }
+
 static StringRef getDumpString(ForeignErrorConvention::IsOwned_t owned) {
   switch (owned) {
   case swift::ForeignErrorConvention::IsNotOwned:
@@ -447,20 +448,6 @@ static size_t getDumpString(size_t value) {
 }
 static void *getDumpString(void *value) { return value; }
 
-static StringRef getDumpString(LifetimeDependenceKind kind) {
-  switch (kind) {
-  case LifetimeDependenceKind::Copy:
-    return "copy";
-  case LifetimeDependenceKind::Consume:
-    return "consume";
-  case LifetimeDependenceKind::Borrow:
-    return "borrow";
-  case LifetimeDependenceKind::Mutate:
-    return "mutate";
-  }
-  llvm_unreachable("Invalid lifetime dependence kind\n");
-}
-
 //===----------------------------------------------------------------------===//
 //  Decl printing.
 //===----------------------------------------------------------------------===//
@@ -3535,7 +3522,7 @@ class PrintTypeRepr : public TypeReprVisitor<PrintTypeRepr, void, StringRef>,
     for (auto &dep : T->getLifetimeDependencies()) {
       printFieldRaw(
           [&](raw_ostream &out) {
-            out << getDumpString(dep.getLifetimeDependenceKind()) << "(";
+            out << dep.getLifetimeDependenceKindString() << "(";
             out << dep.getParamString() << ")";
           },
           "");

lib/AST/TypeRepr.cpp

@@ -469,6 +469,11 @@ SourceLoc LifetimeDependentReturnTypeRepr::getLocImpl() const {
 
 void LifetimeDependentReturnTypeRepr::printImpl(
     ASTPrinter &Printer, const PrintOptions &Opts) const {
+  for (auto &dep : getLifetimeDependencies()) {
+    Printer << dep.getLifetimeDependenceKindString() << "(";
+    Printer << dep.getParamString() << ")";
+  }
+
   printTypeRepr(getBase(), Printer, Opts);
 }