[astgen] Set TypeRepr when visiting ParamDecls.

Author: zoecarver

include/swift/AST/CASTBridging.h

@@ -144,7 +144,8 @@ void *BridgedSourceLoc_advanced(void *loc, long len);
 
 void *ParamDecl_create(void *ctx, void *loc, void *_Nullable argLoc,
                        void *_Nullable argName, void *_Nullable paramLoc,
-                       void *_Nullable paramName, void *declContext);
+                       void *_Nullable paramName, void *_Nullable type,
+                       void *declContext);
 
 void *FuncDecl_create(void *ctx, void *staticLoc, _Bool isStatic, void *funcLoc,
                       BridgedIdentifier name, void *nameLoc, _Bool isAsync,

lib/AST/CASTBridging.cpp

@@ -183,13 +183,16 @@ void *ParamDecl_create(void *ctx, void *loc, void *_Nullable argLoc,
                        BridgedIdentifier _Nullable argName,
                        void *_Nullable paramLoc,
                        BridgedIdentifier _Nullable paramName,
+                       void *_Nullable type,
                        void *declContext) {
   ASTContext &Context = *static_cast<ASTContext *>(ctx);
-  return new (Context) ParamDecl(
+  auto paramDecl = new (Context) ParamDecl(
       getSourceLocFromPointer(loc), getSourceLocFromPointer(argLoc),
       Identifier::getFromOpaquePointer(argName),
       getSourceLocFromPointer(paramLoc),
       Identifier::getFromOpaquePointer(paramName), (DeclContext *)declContext);
+  paramDecl->setTypeRepr((TypeRepr *)type);
+  return paramDecl;
 }
 
 void *FuncDecl_create(void *ctx, void *staticLoc, bool isStatic, void *funcLoc,

lib/ASTGen/Sources/ASTGen/Decls.swift

@@ -83,6 +83,7 @@ extension ASTGenVisitor {
 
     let firstName: UnsafeMutableRawPointer?
     let secondName: UnsafeMutableRawPointer?
+    let type: UnsafeMutableRawPointer?
 
     if let nodeFirstName = node.firstName {
       var text = nodeFirstName.text
@@ -101,8 +102,14 @@ extension ASTGenVisitor {
     } else {
       secondName = nil
     }
+    
+    if let typeSyntax = node.type {
+      type = visit(typeSyntax).rawValue
+    } else {
+      type = nil
+    }
 
-    return .decl(ParamDecl_create(ctx, loc, loc, firstName, loc, secondName, declContext))
+    return .decl(ParamDecl_create(ctx, loc, loc, firstName, loc, secondName, type, declContext))
   }
 
   public func visit(_ node: FunctionDeclSyntax) -> ASTNode {