Merge pull request #21 from Cabazure/aot

Add support for AOT

Author: rickykaare

src/Cabazure.Client/ClientInitializationGenerator.cs

@@ -14,7 +14,7 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
         context.RegisterPostInitializationOutput(c =>
             c.AddSource(
             "ClientInitialization.g.cs",
-            """
+            $$"""
             // <auto-generated/>
             #nullable enable
             using System;
@@ -51,15 +51,21 @@ internal static partial IServiceCollection AddCabazureClient(
             .Where(e => e != null)
             .Collect();
 
+        var aotSupport = context.CompilationProvider
+             .Select((options, _) => options.GetTypeByMetadataName(
+                 TypeConstants.DynamicallyAccessedMembersAttribute) is { });
+
         context.RegisterSourceOutput(
-            endpoints,
+            endpoints.Combine(aotSupport),
             GenerateInitialization);
     }
 
     public static void GenerateInitialization(
-        SourceProductionContext context, 
-        ImmutableArray<EndpointReferenceDescriptor> endpoints)
+        SourceProductionContext context,
+        (ImmutableArray<EndpointReferenceDescriptor>, bool) args)
     {
+        var (endpoints, aotSupport) = args;
+
         var source = new StringBuilder();
         source.AppendLine("// <auto-generated/>");
         source.AppendLine("#nullable enable");
@@ -68,6 +74,7 @@ public static void GenerateInitialization(
             .Select(e => e.Namespace)
             .OfType<string>()
             .Append("System")
+            .Append("System.Diagnostics.CodeAnalysis")
             .Append("System.Net.Http")
             .Append("System.Text.Json")
             .Append("System.Collections.Generic")
@@ -90,12 +97,16 @@ public static void GenerateInitialization(
             source.AppendLine();
         }
 
+        var aotAttribute = aotSupport
+            ? "[DynamicallyAccessedMembers(DynamicallyAccessedMemberTypes.PublicConstructors)] "
+            : string.Empty;
+
         source.AppendLine($$"""
             namespace Microsoft.Extensions.DependencyInjection
             {
                 internal static partial class ClientInitialization
                 {
-                    internal static partial IServiceCollection AddCabazureClient<TOptions>(
+                    internal static partial IServiceCollection AddCabazureClient<{{aotAttribute}}TOptions>(
                         this IServiceCollection services,
                         string clientName,
                         Action<JsonSerializerOptions>? jsonOptions,

src/Cabazure.Client/TypeConstants.cs

@@ -20,4 +20,5 @@ public static class TypeConstants
     public const string Task = "System.Threading.Tasks.Task";
     public const string EndpointResponse = "Cabazure.Client.EndpointResponse";
     public const string PagedResponse = "Cabazure.Client.PagedResponse";
+    public const string DynamicallyAccessedMembersAttribute = "System.Diagnostics.CodeAnalysis.DynamicallyAccessedMembersAttribute";
 }

test/Cabazure.Client.IntegrationTests/Cabazure.Client.IntegrationTests.csproj

@@ -9,6 +9,7 @@
     <IsTestProject>true</IsTestProject>
     <EnableNETAnalyzers>true</EnableNETAnalyzers>
     <EnforceExtendedAnalyzerRules>true</EnforceExtendedAnalyzerRules>
+    <IsAotCompatible>true</IsAotCompatible>
   </PropertyGroup>
 
   <ItemGroup>

test/Cabazure.Client.Tests/ClientInitializationGeneratorTests.CanGenerate_Initialization#ClientInitialization.Implementation.g.verified.cs

@@ -3,6 +3,7 @@
 #nullable enable
 using System;
 using System.Collections.Generic;
+using System.Diagnostics.CodeAnalysis;
 using System.Net.Http;
 using System.Text.Json;
 using Azure.Core;
@@ -16,7 +17,7 @@ namespace Microsoft.Extensions.DependencyInjection
 {
     internal static partial class ClientInitialization
     {
-        internal static partial IServiceCollection AddCabazureClient<TOptions>(
+        internal static partial IServiceCollection AddCabazureClient<[DynamicallyAccessedMembers(DynamicallyAccessedMemberTypes.PublicConstructors)] TOptions>(
             this IServiceCollection services,
             string clientName,
             Action<JsonSerializerOptions>? jsonOptions,