Fix return statement in child blocks

src/Zomp.SyncMethodGenerator/AsyncToSyncRewriter.cs

@@ -520,9 +520,11 @@ bool IsValidParameter(ParameterSyntax ps)
                  semanticModel.GetTypeInfo(returnExpression).Type is INamedTypeSymbol { Name: "Task" or "ValueTask", IsGenericType: false } returnType &&
                  returnType.ToString() is TaskType or ValueTaskType)
         {
-            var result = (ExpressionSyntax)Visit(returnExpression);
+            var result = !ShouldRemoveArgument(returnExpression)
+                ? (ExpressionSyntax)Visit(returnExpression)
+                : null;
 
-            if (node.Parent is not BlockSyntax)
+            if (result is not null && node.Parent is not BlockSyntax)
             {
                 // The parent is not a block, for example: if (true) return ReturnAsync();
                 // We need to create a block with the expression and the return statement.
@@ -536,14 +538,26 @@ bool IsValidParameter(ParameterSyntax ps)
             }
 
             // Don't return if the return statement is the last statement in the method.
-            if (node.Parent.Parent is MethodDeclarationSyntax { Body.Statements: [.., var lastStatement] } &&
+            if (node.Parent?.Parent is MethodDeclarationSyntax { Body.Statements: [.., var lastStatement] } &&
                 lastStatement == node)
             {
+                if (result is null)
+                {
+                    return null;
+                }
+
                 return ExpressionStatement(result)
                     .WithLeadingTrivia(node.GetLeadingTrivia())
                     .WithTrailingTrivia(node.GetTrailingTrivia());
             }
 
+            if (result is null)
+            {
+                return ReturnStatement()
+                    .WithTrailingTrivia(node.GetTrailingTrivia())
+                    .WithLeadingTrivia(node.GetLeadingTrivia());
+            }
+
             // Create a block without the braces (eg. Return(); return;)
             return Block(List(new StatementSyntax[]
                 {
@@ -1550,7 +1564,7 @@ private TypeSyntax ProcessSyntaxUsingSymbol(TypeSyntax typeSyntax)
         IfStatementSyntax @if => ShouldRemoveArgument(@if.Condition),
         ExpressionStatementSyntax e => ShouldRemoveArgument(e.Expression),
         LocalDeclarationStatementSyntax l => CanDropDeclaration(l),
-        ReturnStatementSyntax { Expression: { } re } => ShouldRemoveArgument(re),
+        ReturnStatementSyntax { Parent.Parent: MethodDeclarationSyntax, Expression: { } re } => ShouldRemoveArgument(re),
         _ => false,
     };
 

tests/Generator.Tests/Snapshots/UnitTests.ReturnDefaultValueTask#ReturnAsync.g.verified.cs

@@ -0,0 +1,9 @@
+//HintName: Test.Class.ReturnAsync.g.cs
+private void Return(bool input)
+{
+    if (input)
+    {
+        return;
+    }
+    global::System.Console.WriteLine("123");
+}

tests/Generator.Tests/UnitTests.cs

@@ -195,6 +195,20 @@ private Task ReturnAsync(bool input)
 
 private Task ReturnAsync() => Task.CompletedTask;
 private void Return() { }
+""".Verify();
+
+    [Fact]
+    public Task ReturnDefaultValueTask() => """
+[CreateSyncVersion]
+private ValueTask ReturnAsync(bool input)
+{
+    if (input)
+    {
+        return default;
+    }
+    Console.WriteLine("123");
+    return default;
+}
 """.Verify();
 
     [Fact]