fix missing aot model SourceGenerationContext (#1636)

* Add VersionTestAsync to validate Kubernetes client version retrieval

* Add SourceGenerationContextGenerator and update registration in KubernetesClientSourceGenerator

Author: tg123

src/LibKubernetesGenerator/KubernetesClientSourceGenerator.cs

@@ -59,6 +59,7 @@ private static IContainer BuildContainer(OpenApiDocument swagger)
                 ;
 
             builder.RegisterType<ModelExtGenerator>();
+            builder.RegisterType<SourceGenerationContextGenerator>();
             builder.RegisterType<ModelGenerator>();
             builder.RegisterType<ApiGenerator>();
             builder.RegisterType<ClientSetGenerator>();
@@ -79,6 +80,7 @@ public void Initialize(IncrementalGeneratorInitializationContext generatorContex
 
                 container.Resolve<ModelGenerator>().Generate(swagger, ctx);
                 container.Resolve<ModelExtGenerator>().Generate(swagger, ctx);
+                container.Resolve<SourceGenerationContextGenerator>().Generate(swagger, ctx);
                 container.Resolve<VersionConverterStubGenerator>().Generate(swagger, ctx);
                 container.Resolve<ApiGenerator>().Generate(swagger, ctx);
                 container.Resolve<ClientSetGenerator>().Generate(swagger, ctx);

src/LibKubernetesGenerator/SourceGenerationContextGenerator.cs

@@ -0,0 +1,24 @@
+using Microsoft.CodeAnalysis;
+using NSwag;
+
+namespace LibKubernetesGenerator
+{
+    internal class SourceGenerationContextGenerator
+    {
+        private readonly ScriptObjectFactory scriptObjectFactory;
+
+        public SourceGenerationContextGenerator(ScriptObjectFactory scriptObjectFactory)
+        {
+            this.scriptObjectFactory = scriptObjectFactory;
+        }
+
+        public void Generate(OpenApiDocument swagger, IncrementalGeneratorPostInitializationContext context)
+        {
+            var definitions = swagger.Definitions.Values;
+            var sc = scriptObjectFactory.CreateScriptObject();
+            sc.SetValue("definitions", definitions, true);
+
+            context.RenderToContext("SourceGenerationContext.cs.template", sc, "SourceGenerationContext.g.cs");
+        }
+    }
+}

src/LibKubernetesGenerator/templates/ModelExtensions.cs.template

@@ -15,15 +15,3 @@ namespace k8s.Models
     }
 {{ end }}
 }
-
-#if NET8_0_OR_GREATER
-namespace k8s 
-{
-    {{ for definition in definitions }}
-    [JsonSerializable(typeof({{ GetClassName definition }}))]
-    {{ end }}
-    internal partial class SourceGenerationContext : JsonSerializerContext
-    {
-    }
-}
-#endif

src/LibKubernetesGenerator/templates/SourceGenerationContext.cs.template

@@ -0,0 +1,15 @@
+// <auto-generated>
+// Code generated by https://github.com/kubernetes-client/csharp/tree/master/src/LibKubernetesGenerator
+// Changes may cause incorrect behavior and will be lost if the code is regenerated.
+// </auto-generated>
+#if NET8_0_OR_GREATER
+namespace k8s 
+{
+    {{ for definition in definitions }}
+    [JsonSerializable(typeof({{ GetClassName definition }}))]
+    {{ end }}
+    internal partial class SourceGenerationContext : JsonSerializerContext
+    {
+    }
+}
+#endif

tests/E2E.Aot.Tests/MinikubeTests.cs

@@ -370,6 +370,15 @@ async Task AssertMd5sumAsync(string file, byte[] orig)
             }
         }
 
+        
+        [MinikubeFact]
+        public async Task VersionTestAsync()
+        {
+            using var client = CreateClient();
+            var version = await client.Version.GetCodeAsync().ConfigureAwait(false);
+            Assert.NotNull(version);
+        }        
+
         public static IKubernetes CreateClient()
         {
             return new Kubernetes(KubernetesClientConfiguration.BuildDefaultConfig());