Merge pull request #61809 from zoecarver/astgen-return-stmt

[astgen] Add ReturnStmt visitor.

Author: zoecarver

include/swift/AST/CASTBridging.h

@@ -105,6 +105,9 @@ void *SwiftTopLevelCodeDecl_createStmt(void *ctx, void *DC, void *startLoc,
 void *SwiftTopLevelCodeDecl_createExpr(void *ctx, void *DC, void *startLoc,
                                        void *element, void *endLoc);
 
+void *ReturnStmt_create(void *ctx, void *loc, void *_Nullable expr);
+
+
 void *SwiftSequenceExpr_create(void *ctx, BridgedArrayRef exprs);
 
 void *SwiftTupleExpr_create(void *ctx, void *lparen, BridgedArrayRef subs,

lib/AST/CASTBridging.cpp

@@ -160,6 +160,11 @@ void *IfStmt_create(void *ctx, void *ifLoc, void *cond, void *_Nullable then, vo
       getSourceLocFromPointer(elseLoc), (Stmt *)elseStmt, None, Context);
 }
 
+void *ReturnStmt_create(void *ctx, void *loc, void *_Nullable expr) {
+  ASTContext &Context = *static_cast<ASTContext *>(ctx);
+  return new (Context) ReturnStmt(getSourceLocFromPointer(loc), (Expr *)expr);
+}
+
 void *BraceStmt_create(void *ctx, void *lbloc, BridgedArrayRef elements, void *rbloc) {
   llvm::SmallVector<ASTNode, 6> nodes;
   for (auto node : getArrayRef<ASTNodeBridged>(elements)) {

lib/ASTGen/Sources/ASTGen/Stmts.swift

@@ -26,4 +26,17 @@ extension ASTGenVisitor {
 
     return .stmt(IfStmt_create(ctx, loc, conditions.first!, body, nil, nil))
   }
+
+  public func visit(_ node: ReturnStmtSyntax) -> ASTNode {
+    let loc = self.base.advanced(by: node.position.utf8Offset).raw
+
+    let expr: ASTNode?
+    if let expression = node.expression {
+      expr = visit(expression)
+    } else {
+      expr = nil
+    }
+
+    return .stmt(ReturnStmt_create(ctx, loc, expr?.rawValue))
+  }
 }