Fix directive processing

Author: GerardSmit

src/Zomp.SyncMethodGenerator/AsyncToSyncRewriter.cs

@@ -2161,11 +2161,14 @@ BinaryExpressionSyntax be
             if (trivia.IsKind(SyntaxKind.DisabledTextTrivia) && directiveStack.IsSyncOnly == true)
             {
                 var statementsText = trivia.ToString();
-                var compilation = ParseCompilationUnit(statementsText);
+                var syntaxTree = CSharpSyntaxTree.ParseText(statementsText, options: (CSharpParseOptions)semanticModel.SyntaxTree.Options);
+                var compilation = semanticModel.Compilation.AddSyntaxTrees(syntaxTree);
+                var model = compilation.GetSemanticModel(syntaxTree);
+                var asyncToSyncRewriter = new AsyncToSyncRewriter(model, disableNullable, preserveProgress);
 
-                foreach (var m in compilation.Members)
+                foreach (var m in syntaxTree.GetCompilationUnitRoot().Members)
                 {
-                    process(m);
+                    process((MemberDeclarationSyntax)asyncToSyncRewriter.Visit(m));
                 }
 
                 continue;
@@ -2195,8 +2198,6 @@ BinaryExpressionSyntax be
                 triviaList.Clear();
             }
 
-            //Cannot to the following because syntax node is not within syntax tree
-            //var statement = (StatementSyntax)Visit(gs.Statement)!;
             statements.Add(statement);
         });
 

tests/Generator.Tests/Snapshots/SyncOnlyTests.InsideEmptyIf#g.verified.cs

@@ -1,5 +1,5 @@
 //HintName: Test.Class.MethodAsync.g.cs
 if (true)
 {
-    throw new InvalidOperationException("Some exception");
+    throw new global::System.InvalidOperationException("Some exception");
 }

tests/Generator.Tests/Snapshots/SyncOnlyTests.SimpleMacro#ExecAsync.g.verified.cs

@@ -1,5 +1,5 @@
 //HintName: Test.Class.ExecAsync.g.cs
 public void Exec()
 {
-    throw new InvalidOperationException("Some exception");
+    throw new global::System.InvalidOperationException("Some exception");
 }

tests/Generator.Tests/Snapshots/SyncOnlyTests.SyncOnlyBeforeMethodBody#MethodWithObsoleteSyncAsync.g.verified.cs

@@ -1,6 +1,6 @@
 //HintName: Test.Class.MethodWithObsoleteSyncAsync.g.cs
-[global::System.Obsolete]
-[global::System.Diagnostics.CodeAnalysis.ExcludeFromCodeCoverage]
+[global::System.ObsoleteAttribute]
+[global::System.Diagnostics.CodeAnalysis.ExcludeFromCodeCoverageAttribute]
 public void MethodWithObsoleteSync()
 {
     global::System.Threading.Thread.Sleep(1000);

tests/Generator.Tests/Snapshots/SyncOnlyTests.SyncOnlyFullNamespace#Test.DisposeTest.DisposeAsync.g.verified.cs

@@ -0,0 +1,21 @@
+//HintName: Test.DisposeTest.DisposeAsync.g.cs
+// <auto-generated/>
+#nullable enable
+namespace Test;
+public partial class DisposeTest
+{
+    public void Dispose(object? obj)
+    {
+
+            if (entity is global::System.IDisposable disposable)
+            {
+                disposable.Dispose();
+                return;
+            }
+
+            if (entity is global::System.IAsyncDisposable)
+            {
+                throw new global::System.InvalidOperationException("Cannot dispose an async disposable in sync-only mode.");
+            }
+}
+}

tests/Generator.Tests/SyncOnlyTests.cs

@@ -290,4 +290,36 @@ public Task SyncOnlyArgument() => """
 #endif
             );
 """.Verify(sourceType: SourceType.MethodBody);
+
+    [Fact]
+    public Task SyncOnlyFullNamespace() => """
+using System;
+namespace Test;
+public partial class DisposeTest
+{
+    [CreateSyncVersion]
+    public async Task DisposeAsync(object? obj)
+    {
+#if !SYNC_ONLY
+            if (obj is IAsyncDisposable asyncDisposable)
+            {
+                await asyncDisposable.DisposeAsync();
+                return;
+            }
+#endif
+
+            if (obj is IDisposable disposable)
+            {
+                disposable.Dispose();
+                return;
+            }
+
+#if SYNC_ONLY
+            if (obj is IAsyncDisposable)
+            {
+                throw new InvalidOperationException("Cannot dispose an async disposable in sync-only mode.");
+            }
+#endif
+}
+""".Verify(sourceType: SourceType.Full);
 }