Add connection_init payload

Author: ecrocombe

src/Linq2GraphQL.Client.Subscriptions/GraphSubscriptionExecute.cs

@@ -26,7 +26,7 @@ public async Task<IObservable<TResult>> StartAsync()
             return sseClient.Subscription.Select(e => ConvertResult(queryExecutor.ProcessResponse(e, QueryNode.Name, payload.Query)));
         }
 
-        var wsClient = new WSClient(client.SubscriptionUrl, client.SubscriptionProtocol, payload);
+        var wsClient = new WSClient(client, payload);
         await wsClient.Start();
         return wsClient.Subscription.Select(e => ConvertResult(queryExecutor.ProcessResponse(e, QueryNode.Name, payload.Query)));
     }

src/Linq2GraphQL.Client.Subscriptions/WSClient.cs

@@ -9,19 +9,17 @@ namespace Linq2GraphQL.Client.Subscriptions;
 
 public class WSClient : IAsyncDisposable
 {
+    private readonly GraphClient _graphClient;
     private readonly GraphQLRequest payload;
-    private readonly SubscriptionProtocol subscriptionProtocol;
 
     private readonly Subject<string> subscriptionSubject = new();
-    private readonly string url;
     private readonly WebsocketClient client;
 
     private readonly JsonSerializerOptions jsonOptions;
 
-    public WSClient(string url, SubscriptionProtocol subprotocol, GraphQLRequest payload)
+    public WSClient(GraphClient graphClient, GraphQLRequest payload)
     {
-        this.url = url;
-        subscriptionProtocol = subprotocol;
+        _graphClient = graphClient;
         this.payload = payload;
         jsonOptions = new JsonSerializerOptions
         {
@@ -35,7 +33,7 @@ public WSClient(string url, SubscriptionProtocol subprotocol, GraphQLRequest pay
             return ws;
         });
 
-        client = new WebsocketClient(new Uri(url), factory)
+        client = new WebsocketClient(new Uri(_graphClient.SubscriptionUrl), factory)
         {
             ReconnectTimeout = TimeSpan.FromSeconds(30)
         };
@@ -74,7 +72,18 @@ public async Task Start()
         });
 
         await client.Start();
-        SendRequest(new WebsocketRequest("connection_init"));
+
+        if (_graphClient.WSConnectionInitPayload is not null) 
+        {
+            var initPayload = _graphClient.WSConnectionInitPayload(_graphClient);
+            if (initPayload is not null)
+            {
+                SendRequest(new WebsocketRequest("connection_init")
+                {
+                    Payload = await _graphClient.WSConnectionInitPayload(_graphClient)
+                });
+            }
+        }
 
         var subscriptionRequest = new WebsocketRequest(GetSubscribeCommand())
         {
@@ -87,7 +96,7 @@ public async Task Start()
 
     private string GetSubprotocolString()
     {
-        switch (subscriptionProtocol)
+        switch (_graphClient.SubscriptionProtocol)
         {
             case SubscriptionProtocol.GraphQLWebSocket:
                 return "graphql-transport-ws";
@@ -96,13 +105,13 @@ private string GetSubprotocolString()
                 return "graphql-ws";
 
             default:
-                throw new Exception($"{subscriptionProtocol} is unknown");
+                throw new Exception($"{_graphClient.SubscriptionProtocol} is unknown");
         }
     }
 
     private string GetSubscribeCommand()
     {
-        switch (subscriptionProtocol)
+        switch (_graphClient.SubscriptionProtocol)
         {
             case SubscriptionProtocol.GraphQLWebSocket:
                 return "subscribe";
@@ -111,7 +120,7 @@ private string GetSubscribeCommand()
                 return "start";
 
             default:
-                throw new Exception($"{subscriptionProtocol} is unknown");
+                throw new Exception($"{_graphClient.SubscriptionProtocol} is unknown");
         }
     }
 

src/Linq2GraphQL.Client/GraphClient.cs

@@ -36,6 +36,7 @@ public GraphClient(HttpClient httpClient, IOptions<GraphClientOptions> options,
     public HttpClient HttpClient { get; }
     public JsonSerializerOptions SerializerOptions { get; }
 
+    public Func<GraphClient, Task<GraphQLRequest>> WSConnectionInitPayload => options.Value.WSConnectionInitPayload;
     private string GetSubscriptionUrl()
     {
         var baseUrl = HttpClient?.BaseAddress.ToString();

src/Linq2GraphQL.Client/GraphClientOptions.cs

@@ -1,7 +1,9 @@
+
 namespace Linq2GraphQL.Client;
 
 public class GraphClientOptions
 {
     public bool UseSafeMode { get; set; } = false;
     public SubscriptionProtocol SubscriptionProtocol { get; set; } = default;
+    public Func<GraphClient, Task<GraphQLRequest>> WSConnectionInitPayload { get; internal set; } = opts => null;
 }