Implementing a more modular API

examples/clientset/Program.cs

@@ -0,0 +1,24 @@
+// See https://aka.ms/new-console-template for more information
+using k8s;
+using k8s.ClientSets;
+using System.Threading.Tasks;
+
+namespace clientset
+{
+    internal class Program
+    {
+        private static async Task Main(string[] args)
+        {
+            var config = KubernetesClientConfiguration.BuildConfigFromConfigFile();
+            IKubernetes client = new Kubernetes(config);
+
+            ClientSet clientSet = new ClientSet(client);
+            var list = await clientSet.CoreV1.Pod.ListNamespacedPodAsync("default").ConfigureAwait(false);
+            foreach (var item in list)
+            {
+                System.Console.WriteLine(item.Metadata.Name);
+            }
+        }
+    }
+
+}

examples/clientset/clientset.csproj

@@ -0,0 +1,6 @@
+<Project Sdk="Microsoft.NET.Sdk">
+    <PropertyGroup>
+        <OutputType>Exe</OutputType>
+    </PropertyGroup>
+
+</Project>

src/KubernetesClient.Aot/KubernetesClient.Aot.csproj

@@ -46,6 +46,10 @@
     <Compile Include="..\KubernetesClient\Models\V1Status.cs" />
 
   </ItemGroup>
+  <ItemGroup>
+      <Compile Include="..\KubernetesClient\ClientSets\ClientSet.cs" />
+      <Compile Include="..\KubernetesClient\ClientSets\ResourceClient.cs"/>
+  </ItemGroup>
   <ItemGroup>
     <Compile Include="..\KubernetesClient\AbstractKubernetes.cs" />
     <Compile Include="..\KubernetesClient\GeneratedApiVersion.cs" />
@@ -107,4 +111,4 @@
   <ItemGroup>
     <ProjectReference Include="..\LibKubernetesGenerator\generators\LibKubernetesGenerator\LibKubernetesGenerator.csproj" OutputItemType="Analyzer" ReferenceOutputAssembly="false" />
   </ItemGroup>
-</Project>
+</Project>

src/KubernetesClient.Classic/KubernetesClient.Classic.csproj

@@ -75,7 +75,10 @@
     <Compile Include="..\KubernetesClient\Autorest\HttpRequestMessageWrapper.cs" />
     <Compile Include="..\KubernetesClient\Autorest\HttpResponseMessageWrapper.cs" />
   </ItemGroup>
-
+  <ItemGroup>
+    <Compile Include="..\KubernetesClient\ClientSets\ClientSet.cs" />
+    <Compile Include="..\KubernetesClient\ClientSets\ResourceClient.cs"/>
+  </ItemGroup>
   <ItemGroup>
     <Compile Include="..\KubernetesClient\FileSystem.cs" />
     <Compile Include="..\KubernetesClient\IKubernetes.cs" />

src/KubernetesClient/AbstractKubernetes.cs

@@ -5,7 +5,7 @@ namespace k8s;
 
 public abstract partial class AbstractKubernetes
 {
-    private static class HttpMethods
+    internal static class HttpMethods
     {
         public static readonly HttpMethod Delete = HttpMethod.Delete;
         public static readonly HttpMethod Get = HttpMethod.Get;
@@ -23,7 +23,7 @@ private static class HttpMethods
 
     }
 
-    private sealed class QueryBuilder
+    internal sealed class QueryBuilder
     {
         private readonly List<string> parameters = new List<string>();
 
@@ -99,7 +99,7 @@ private MediaTypeHeaderValue GetHeader(V1Patch body)
         }
     }
 
-    protected abstract Task<HttpOperationResponse<T>> CreateResultAsync<T>(HttpRequestMessage httpRequest, HttpResponseMessage httpResponse, bool? watch, CancellationToken cancellationToken);
+    internal abstract Task<HttpOperationResponse<T>> CreateResultAsync<T>(HttpRequestMessage httpRequest, HttpResponseMessage httpResponse, bool? watch, CancellationToken cancellationToken);
 
-    protected abstract Task<HttpResponseMessage> SendRequest<T>(string relativeUri, HttpMethod method, IReadOnlyDictionary<string, IReadOnlyList<string>> customHeaders, T body, CancellationToken cancellationToken);
+    internal abstract Task<HttpResponseMessage> SendRequest<T>(string relativeUri, HttpMethod method, IReadOnlyDictionary<string, IReadOnlyList<string>> customHeaders, T body, CancellationToken cancellationToken);
 }

src/KubernetesClient/ClientSets/ClientSet.cs

@@ -0,0 +1,6 @@
+namespace k8s.ClientSets;
+
+public partial class ClientSet
+{
+
+}

src/KubernetesClient/ClientSets/ResourceClient.cs

@@ -0,0 +1,11 @@
+namespace k8s.ClientSets;
+
+public abstract class ResourceClient
+{
+    protected Kubernetes Client { get; }
+
+    public ResourceClient(IKubernetes kubernetes)
+    {
+        Client = (Kubernetes)kubernetes;
+    }
+}

src/KubernetesClient/GenericClient.cs

@@ -30,7 +30,8 @@
 
         public async Task<T> CreateAsync<T>(T obj, CancellationToken cancel = default)
         where T : IKubernetesObject
         {
+
             var resp = await kubernetes.CustomObjects.CreateClusterCustomObjectWithHttpMessagesAsync<T>(obj, group, version, plural, cancellationToken: cancel).ConfigureAwait(false);
             return resp.Body;
         }

src/KubernetesClient/Kubernetes.WebSocket.cs

@@ -18,19 +18,19 @@ public partial class Kubernetes
         /// <inheritdoc/>
         public Task<WebSocket> WebSocketNamespacedPodExecAsync(string name, string @namespace = "default",
             string command = null, string container = null, bool stderr = true, bool stdin = true, bool stdout = true,
-            bool tty = true, string webSocketSubProtol = null, Dictionary<string, List<string>> customHeaders = null,
+            bool tty = true, string webSocketSubProtocol = null, Dictionary<string, List<string>> customHeaders = null,
             CancellationToken cancellationToken = default)
         {
             return WebSocketNamespacedPodExecAsync(name, @namespace, new string[] { command }, container, stderr, stdin,
-                stdout, tty, webSocketSubProtol, customHeaders, cancellationToken);
+                stdout, tty, webSocketSubProtocol, customHeaders, cancellationToken);
         }
 
         /// <inheritdoc/>
         public virtual async Task<IStreamDemuxer> MuxedStreamNamespacedPodExecAsync(
             string name,
             string @namespace = "default", IEnumerable<string> command = null, string container = null,
             bool stderr = true, bool stdin = true, bool stdout = true, bool tty = true,
-            string webSocketSubProtol = WebSocketProtocol.V4BinaryWebsocketProtocol,
+            string webSocketSubProtocol = WebSocketProtocol.V4BinaryWebsocketProtocol,
             Dictionary<string, List<string>> customHeaders = null,
             CancellationToken cancellationToken = default)
         {
@@ -45,7 +45,7 @@ public virtual async Task<IStreamDemuxer> MuxedStreamNamespacedPodExecAsync(
         public virtual Task<WebSocket> WebSocketNamespacedPodExecAsync(string name, string @namespace = "default",
             IEnumerable<string> command = null, string container = null, bool stderr = true, bool stdin = true,
             bool stdout = true, bool tty = true,
-            string webSocketSubProtol = WebSocketProtocol.V4BinaryWebsocketProtocol,
+            string webSocketSubProtocol = WebSocketProtocol.V4BinaryWebsocketProtocol,
             Dictionary<string, List<string>> customHeaders = null,
             CancellationToken cancellationToken = default)
         {
@@ -114,7 +114,7 @@ public virtual Task<WebSocket> WebSocketNamespacedPodExecAsync(string name, stri
             uriBuilder.Query =
                 query.ToString(1, query.Length - 1); // UriBuilder.Query doesn't like leading '?' chars, so trim it
 
-            return StreamConnectAsync(uriBuilder.Uri, webSocketSubProtol, customHeaders,
+            return StreamConnectAsync(uriBuilder.Uri, webSocketSubProtocol, customHeaders,
                 cancellationToken);
         }
 
@@ -173,7 +173,7 @@ public Task<WebSocket> WebSocketNamespacedPodPortForwardAsync(string name, strin
         /// <inheritdoc/>
         public Task<WebSocket> WebSocketNamespacedPodAttachAsync(string name, string @namespace,
             string container = default, bool stderr = true, bool stdin = false, bool stdout = true,
-            bool tty = false, string webSocketSubProtol = null, Dictionary<string, List<string>> customHeaders = null,
+            bool tty = false, string webSocketSubProtocol = null, Dictionary<string, List<string>> customHeaders = null,
             CancellationToken cancellationToken = default)
         {
             if (name == null)
@@ -208,7 +208,7 @@ public Task<WebSocket> WebSocketNamespacedPodAttachAsync(string name, string @na
             uriBuilder.Query =
                 query.ToString(1, query.Length - 1); // UriBuilder.Query doesn't like leading '?' chars, so trim it
 
-            return StreamConnectAsync(uriBuilder.Uri, webSocketSubProtol, customHeaders,
+            return StreamConnectAsync(uriBuilder.Uri, webSocketSubProtocol, customHeaders,
                 cancellationToken);
         }
 

src/KubernetesClient/Kubernetes.cs

@@ -56,7 +56,7 @@ private void Initialize()
             BaseUri = new Uri("http://localhost");
         }
 
-        protected override async Task<HttpOperationResponse<T>> CreateResultAsync<T>(HttpRequestMessage httpRequest, HttpResponseMessage httpResponse, bool? watch, CancellationToken cancellationToken)
+        internal override async Task<HttpOperationResponse<T>> CreateResultAsync<T>(HttpRequestMessage httpRequest, HttpResponseMessage httpResponse, bool? watch, CancellationToken cancellationToken)
         {
             if (httpRequest == null)
             {
@@ -96,7 +96,7 @@ protected override async Task<HttpOperationResponse<T>> CreateResultAsync<T>(Htt
             return result;
         }
 
-        protected override Task<HttpResponseMessage> SendRequest<T>(string relativeUri, HttpMethod method, IReadOnlyDictionary<string, IReadOnlyList<string>> customHeaders, T body, CancellationToken cancellationToken)
+        internal override Task<HttpResponseMessage> SendRequest<T>(string relativeUri, HttpMethod method, IReadOnlyDictionary<string, IReadOnlyList<string>> customHeaders, T body, CancellationToken cancellationToken)
         {
             var httpRequest = new HttpRequestMessage
             {

src/LibKubernetesGenerator/ClientSetGenerator.cs

@@ -0,0 +1,104 @@
+using CaseExtensions;
+using Microsoft.CodeAnalysis;
+using NSwag;
+using System.Collections.Generic;
+using System.Linq;
+
+namespace LibKubernetesGenerator
+{
+    internal class ClientSetGenerator
+    {
+        private readonly ScriptObjectFactory _scriptObjectFactory;
+
+        public ClientSetGenerator(ScriptObjectFactory scriptObjectFactory)
+        {
+            _scriptObjectFactory = scriptObjectFactory;
+        }
+
+        public void Generate(OpenApiDocument swagger, IncrementalGeneratorPostInitializationContext context)
+        {
+            var data = swagger.Operations
+                .Where(o => o.Method != OpenApiOperationMethod.Options)
+                .Select(o =>
+                {
+                    var ps = o.Operation.ActualParameters.OrderBy(p => !p.IsRequired).ToArray();
+
+                    o.Operation.Parameters.Clear();
+
+                    var name = new HashSet<string>();
+
+                    var i = 1;
+                    foreach (var p in ps)
+                    {
+                        if (name.Contains(p.Name))
+                        {
+                            p.Name += i++;
+                        }
+
+                        o.Operation.Parameters.Add(p);
+                        name.Add(p.Name);
+                    }
+
+                    return o;
+                })
+                .Select(o =>
+                {
+                    o.Path = o.Path.TrimStart('/');
+                    o.Method = char.ToUpper(o.Method[0]) + o.Method.Substring(1);
+                    return o;
+                })
+                .ToArray();
+
+            var sc = _scriptObjectFactory.CreateScriptObject();
+
+            var groups = new List<string>();
+            var apiGroups = new Dictionary<string, OpenApiOperationDescription[]>();
+
+            foreach (var grouped in data.Where(d => HasKubernetesAction(d.Operation?.ExtensionData))
+                         .GroupBy(d => d.Operation.Tags.First()))
+            {
+                var clients = new List<string>();
+                var name = grouped.Key.ToPascalCase();
+                groups.Add(name);
+                var apis = grouped.Select(x =>
+                {
+                    var groupVersionKindElements = x.Operation?.ExtensionData?["x-kubernetes-group-version-kind"];
+                    var groupVersionKind = (Dictionary<string, object>)groupVersionKindElements;
+
+                    return new { Kind = groupVersionKind?["kind"], Api = x };
+
+                });
+
+                foreach (var item in apis.GroupBy(x => x.Kind))
+                {
+                    var kind = item.Key as string;
+                    apiGroups[kind] = item.Select(x => x.Api).ToArray();
+                    clients.Add(kind);
+                }
+
+                sc.SetValue("clients", clients, true);
+                sc.SetValue("name", name, true);
+                context.RenderToContext("GroupClient.cs.template", sc, $"{name}GroupClient.g.cs");
+            }
+
+            foreach (var apiGroup in apiGroups)
+            {
+                var name = apiGroup.Key;
+                var apis = apiGroup.Value.ToArray();
+
+                sc.SetValue("apis", apis, true);
+                sc.SetValue("name", name, true);
+                context.RenderToContext("Client.cs.template", sc, $"{name}Client.g.cs");
+                context.RenderToContext("ClientExtensions.cs.template", sc, $"{name}ClientExtensions.g.cs");
+            }
+
+            sc = _scriptObjectFactory.CreateScriptObject();
+            sc.SetValue("groups", groups, true);
+
+            context.RenderToContext("ClientSet.cs.template", sc, $"ClientSet.g.cs");
+        }
+
+        private bool HasKubernetesAction(IDictionary<string, object> extensionData) =>
+            extensionData?.ContainsKey("x-kubernetes-action") ?? false;
+    }
+}

src/LibKubernetesGenerator/KubernetesClientSourceGenerator.cs

@@ -21,46 +21,38 @@ private static IContainer BuildContainer(OpenApiDocument swagger)
             builder.RegisterType<ClassNameHelper>()
                 .WithParameter(new NamedParameter(nameof(swagger), swagger))
                 .AsSelf()
-                .AsImplementedInterfaces()
-                ;
+                .AsImplementedInterfaces();
 
             builder.RegisterType<StringHelpers>()
-                .AsImplementedInterfaces()
-                ;
+                .AsImplementedInterfaces();
 
             builder.RegisterType<MetaHelper>()
-                .AsImplementedInterfaces()
-                ;
+                .AsImplementedInterfaces();
 
             builder.RegisterType<PluralHelper>()
                 .WithParameter(new TypedParameter(typeof(OpenApiDocument), swagger))
-                .AsImplementedInterfaces()
-                ;
+                .AsImplementedInterfaces();
 
             builder.RegisterType<GeneralNameHelper>()
                 .AsSelf()
-                .AsImplementedInterfaces()
-                ;
+                .AsImplementedInterfaces();
 
             builder.RegisterType<TypeHelper>()
                 .AsSelf()
-                .AsImplementedInterfaces()
-                ;
+                .AsImplementedInterfaces();
 
             builder.RegisterType<ParamHelper>()
-                .AsImplementedInterfaces()
-                ;
+                .AsImplementedInterfaces();
 
             builder.RegisterType<UtilHelper>()
-                .AsImplementedInterfaces()
-                ;
+                .AsImplementedInterfaces();
 
-            builder.RegisterType<ScriptObjectFactory>()
-                ;
+            builder.RegisterType<ScriptObjectFactory>();
 
             builder.RegisterType<ModelExtGenerator>();
             builder.RegisterType<ModelGenerator>();
             builder.RegisterType<ApiGenerator>();
+            builder.RegisterType<ClientSetGenerator>();
             builder.RegisterType<VersionConverterStubGenerator>();
             builder.RegisterType<VersionGenerator>();
 
@@ -80,6 +72,7 @@ public void Initialize(IncrementalGeneratorInitializationContext generatorContex
                 container.Resolve<ModelExtGenerator>().Generate(swagger, ctx);
                 container.Resolve<VersionConverterStubGenerator>().Generate(swagger, ctx);
                 container.Resolve<ApiGenerator>().Generate(swagger, ctx);
+                container.Resolve<ClientSetGenerator>().Generate(swagger, ctx);
             });
 #endif