[ASTPrinter] TypeWrappers: print inferred attributes

Author: xedin

lib/AST/ASTPrinter.cpp

@@ -1179,6 +1179,16 @@ void PrintAST::printAttributes(const Decl *D) {
 
   D->getAttrs().print(Printer, Options, D);
 
+  // We need to check whether this is a type with an inferred
+  // type wrapper attribute and if so print it explicitly.
+  if (auto *NTD = dyn_cast<NominalTypeDecl>(D)) {
+    auto typeWrapperInfo = NTD->getTypeWrapper();
+    // The attribute has been inferred and we have to print it.
+    if (typeWrapperInfo && typeWrapperInfo->IsInferred) {
+      typeWrapperInfo->Attr->print(Printer, Options, D);
+    }
+  }
+
   // Print the implicit 'final' attribute.
   if (auto VD = dyn_cast<ValueDecl>(D)) {
     auto VarD = dyn_cast<VarDecl>(D);

test/ModuleInterface/type_wrappers.swift

@@ -46,7 +46,7 @@ public protocol Wrapped {
 
 public protocol OuterWrapped : Wrapped {}
 
-// CHECK: public struct WithProtocol : TypeWrappers.Wrapped {
+// CHECK:  @TypeWrappers.Wrapper public struct WithProtocol : TypeWrappers.Wrapped {
 // CHECK:   public var a: Swift.Int {
 // CHECK:     get
 // CHECK:     set