taking shape

Author: eduard-dumitru

src/UiPath.CoreIpc.BackCompat/ServiceClientBuilder.cs

@@ -4,8 +4,6 @@
 
 namespace UiPath.Ipc.BackCompat;
 
-
-
 public abstract class ServiceClientBuilder<TDerived, TInterface>
     : ServiceClientBuilder<TInterface>
     where TInterface : class

src/UiPath.CoreIpc/Client/ServiceClient.cs

@@ -208,6 +208,7 @@ public override async ValueTask CloseConnection()
             return (LatestConnection, newlyConnected: true);
         }
     }
+
     private async Task<Network> Connect(CancellationToken ct)
     {
         if (ConnectionFactory is not null

src/UiPath.CoreIpc/Extensibility/ClientBase.cs

@@ -15,6 +15,11 @@ public abstract record ClientBase : EndpointConfig
 
     public virtual void Validate() { }
 
+    public TProxy GetProxy<TProxy>() where TProxy : class
+    => new ServiceClientProper<TClient, TClientState>(_client, typeof(TProxy))
+        .CreateProxy<TProxy>();
+
+
     internal void ValidateInternal()
     {
         var haveDeferredInjectedCallbacks = Callbacks?.Any(x => !x.Service.HasServiceProvider() && x.Service.MaybeGetInstance() is null) ?? false;
@@ -55,9 +60,7 @@ internal override RouterConfig CreateCallbackRouterConfig()
 
 public interface IClient<TState, TSelf>
     where TSelf : ClientBase, IClient<TState, TSelf>
-    where TState : class, IClientState<TSelf, TState>, new()
-{
-}
+    where TState : class, IClientState<TSelf, TState>, new() { }
 
 public static class ClientExtensions
 {

src/UiPath.CoreIpc/IpcJsonSerializer.cs

@@ -1,5 +1,4 @@
-using Microsoft.Extensions.Logging.Abstractions;
-using Newtonsoft.Json;
+using Newtonsoft.Json;
 using System.Buffers;
 using System.Globalization;
 using System.Text;
@@ -22,30 +21,39 @@ internal class IpcJsonSerializer : ISerializer, IArrayPool<char>
 #if !NET461
     [AsyncMethodBuilder(typeof(PoolingAsyncValueTaskMethodBuilder<>))]
 #endif
-    public async ValueTask<T> DeserializeAsync<T>(Stream json)
+    public async ValueTask<T?> DeserializeAsync<T>(Stream json)
     {
         using var stream = IOHelpers.GetStream((int)json.Length);
         await json.CopyToAsync(stream);
         stream.Position = 0;
         using var reader = CreateReader(new StreamReader(stream));
         return StringArgsSerializer.Deserialize<T>(reader);
     }
-    public void Serialize(object obj, Stream stream) => Serialize(obj, new StreamWriter(stream), StringArgsSerializer);
+    public void Serialize(object? obj, Stream stream) => Serialize(obj, new StreamWriter(stream), StringArgsSerializer);
     private void Serialize(object obj, TextWriter streamWriter, JsonSerializer serializer)
     {
         using var writer = new JsonTextWriter(streamWriter) { ArrayPool = this, CloseOutput = false };
         serializer.Serialize(writer, obj);
         writer.Flush();
     }
     public char[] Rent(int minimumLength) => ArrayPool<char>.Shared.Rent(minimumLength);
-    public void Return(char[] array) => ArrayPool<char>.Shared.Return(array);
-    public string Serialize(object obj)
+    public void Return(char[]? array)
+    {
+        if (array is null)
+        {
+            return;
+        }
+
+        ArrayPool<char>.Shared.Return(array);
+    }
+
+    public string Serialize(object? obj)
     {
         var stringWriter = new StringWriter(new StringBuilder(capacity: 256), CultureInfo.InvariantCulture);
         Serialize(obj, stringWriter, StringArgsSerializer);
         return stringWriter.ToString();
     }
-    public object Deserialize(string json, Type type)
+    public object? Deserialize(string json, Type type)
     {
         using var reader = CreateReader(new StringReader(json));
         return StringArgsSerializer.Deserialize(reader, type);