refactor: rename GetParameterValueForWatchCall to GetParameterValueForWatch and update templates

Author: lqlive

src/LibKubernetesGenerator/ParamHelper.cs

@@ -3,6 +3,7 @@
 using Scriban.Runtime;
 using System;
 using System.Linq;
+using System.Collections.Generic;
 
 namespace LibKubernetesGenerator
 {
@@ -22,7 +23,7 @@ public void RegisterHelper(ScriptObject scriptObject)
             scriptObject.Import(nameof(GetModelCtorParam), new Func<JsonSchema, string>(GetModelCtorParam));
             scriptObject.Import(nameof(IfParamContains), IfParamContains);
             scriptObject.Import(nameof(FilterParameters), FilterParameters);
-            scriptObject.Import(nameof(GetParameterValueForWatchCall), new Func<OpenApiParameter, string>(GetParameterValueForWatchCall));
+            scriptObject.Import(nameof(GetParameterValueForWatch), new Func<OpenApiParameter, string>(GetParameterValueForWatch));
         }
 
         public static bool IfParamContains(OpenApiOperation operation, string name)
@@ -41,12 +42,12 @@ public static bool IfParamContains(OpenApiOperation operation, string name)
             return found;
         }
 
-        public static System.Collections.Generic.IEnumerable<OpenApiParameter> FilterParameters(OpenApiOperation operation, string excludeParam)
+        public static IEnumerable<OpenApiParameter> FilterParameters(OpenApiOperation operation, string excludeParam)
         {
             return operation.Parameters.Where(p => p.Name != excludeParam);
         }
 
-        public string GetParameterValueForWatchCall(OpenApiParameter parameter)
+        public string GetParameterValueForWatch(OpenApiParameter parameter)
         {
             if (parameter.Name == "watch")
             {

src/LibKubernetesGenerator/templates/Client.cs.template

@@ -96,5 +96,69 @@ public partial class {{name}}Client : ResourceClient
         }
     }
     {{end}}
+
+    {{if IfParamContains api.operation "watch"}}
+    {{~ $filteredParams = FilterParameters api.operation "watch" ~}}
+    /// <summary>
+    /// Watch {{ToXmlDoc api.operation.description}}
+    /// </summary>
+    {{ for parameter in $filteredParams}}
+    /// <param name="{{GetDotNetNameOpenApiParameter parameter "false"}}">
+    /// {{ToXmlDoc parameter.description}}
+    /// </param>
+    {{ end }}
+    /// <param name="onEvent">Callback when any event raised from api server</param>
+    /// <param name="onError">Callback when any exception was caught during watching</param>
+    /// <param name="onClosed">Callback when the server closes the connection</param>
+    public Watcher<{{GetReturnType api.operation "T"}}> Watch{{GetActionName api.operation name ""}}(
+        {{ for parameter in $filteredParams}}
+        {{GetDotNetTypeOpenApiParameter parameter}} {{GetDotNetNameOpenApiParameter parameter "true"}},
+        {{ end }}
+        Action<WatchEventType, {{GetReturnType api.operation "T"}}> onEvent = null,
+        Action<Exception> onError = null,
+        Action onClosed = null)
+    {
+        if (onEvent == null) throw new ArgumentNullException(nameof(onEvent));
+
+        var responseTask = Client.{{group}}.{{GetOperationId api.operation "WithHttpMessagesAsync"}}(
+            {{ for parameter in api.operation.parameters}}
+            {{GetParameterValueForWatch parameter}},
+            {{ end }}
+            null,
+            CancellationToken.None);
+
+        return responseTask.Watch<{{GetReturnType api.operation "T"}}, {{GetReturnType api.operation "TList"}}>(
+            onEvent, onError, onClosed);
+    }
+
+    /// <summary>
+    /// Watch {{ToXmlDoc api.operation.description}} as async enumerable
+    /// </summary>
+    {{ for parameter in $filteredParams}}
+    /// <param name="{{GetDotNetNameOpenApiParameter parameter "false"}}">
+    /// {{ToXmlDoc parameter.description}}
+    /// </param>
+    {{ end }}
+    /// <param name="onError">Callback when any exception was caught during watching</param>
+    /// <param name="cancellationToken">Cancellation token</param>
+    public IAsyncEnumerable<(WatchEventType, {{GetReturnType api.operation "T"}})> Watch{{GetActionName api.operation name "Async"}}(
+        {{ for parameter in $filteredParams}}
+        {{GetDotNetTypeOpenApiParameter parameter}} {{GetDotNetNameOpenApiParameter parameter "true"}},
+        {{ end }}
+        Action<Exception> onError = null,
+        CancellationToken cancellationToken = default)
+    {
+        var responseTask = Client.{{group}}.{{GetOperationId api.operation "WithHttpMessagesAsync"}}(
+            {{ for parameter in api.operation.parameters}}
+            {{GetParameterValueForWatch parameter}},
+            {{ end }}
+            null,
+            cancellationToken);
+
+        return responseTask.WatchAsync<{{GetReturnType api.operation "T"}}, {{GetReturnType api.operation "TList"}}>(
+            onError, cancellationToken);
+    }
     {{end}}
+
+    {{end}} 
 }

src/LibKubernetesGenerator/templates/OperationsExtensions.cs.template

@@ -183,7 +183,7 @@ public static Watcher<{{GetReturnType api.operation "T"}}> Watch{{GetOperationId
 
     var responseTask = operations.{{GetOperationId api.operation "WithHttpMessagesAsync"}}(
 {{ for parameter in api.operation.parameters}}
-        {{GetParameterValueForWatchCall parameter}},
+        {{GetParameterValueForWatch parameter}},
 {{end}}
         null,
         CancellationToken.None);
@@ -215,7 +215,7 @@ public static IAsyncEnumerable<(WatchEventType, {{GetReturnType api.operation "T
 {
     var responseTask = operations.{{GetOperationId api.operation "WithHttpMessagesAsync"}}(
 {{ for parameter in api.operation.parameters}}
-        {{GetParameterValueForWatchCall parameter}},
+        {{GetParameterValueForWatch parameter}},
 {{end}}
         null,
         cancellationToken);