Adds onOpenHandler

Author: yawaflua

yawaflua.WebSockets/Core/WebSocketConfig.cs

@@ -0,0 +1,10 @@
+using Microsoft.AspNetCore.Http;
+using yawaflua.WebSockets.Models.Interfaces;
+
+namespace yawaflua.WebSockets.Core;
+
+public class WebSocketConfig
+{
+    public Func<IWebSocket, HttpContext, Task>? OnOpenHandler { get; set; } = null;
+
+}

yawaflua.WebSockets/Core/WebSocketRouter.cs

@@ -17,11 +17,12 @@ public class WebSocketRouter
     internal static readonly List<IWebSocketClient> Clients = new();
     private readonly IServiceProvider _serviceProvider;
     private readonly ILogger<WebSocketRouter> _logger;
-
-    public WebSocketRouter(IServiceProvider serviceProvider, ILogger<WebSocketRouter> logger)
+    private readonly WebSocketConfig WebSocketConfig;
+    public WebSocketRouter(IServiceProvider serviceProvider, ILogger<WebSocketRouter> logger, WebSocketConfig webSocketConfig)
     {
         _serviceProvider = serviceProvider;
         this._logger = logger;
+        WebSocketConfig = webSocketConfig;
         DiscoverHandlers();
         Task.Run(() =>
         {
@@ -146,6 +147,13 @@ await Task.Run(async () =>
                     {
                         var client = new WebSocketClient(context, webSocket, path);
                         Clients.Add(client);
+                        
+                        await Task.Run(async () =>
+                        {
+                            if (WebSocketConfig.OnOpenHandler != null)
+                                await WebSocketConfig.OnOpenHandler((webSocket as IWebSocket)!, context);
+                        }, cts);
+                        
                         var buffer = new byte[1024 * 4];
                         while (webSocket.State == WebSocketState.Open)
                         {

yawaflua.WebSockets/ServiceBindings.cs

@@ -8,11 +8,12 @@ namespace yawaflua.WebSockets;
 
 public static class ServiceBindings
 {
-    public static IServiceCollection SettingUpWebSockets(this IServiceCollection isc)
+    public static IServiceCollection SettingUpWebSockets(this IServiceCollection isc, Action<WebSocketOptions>? socketOptions = null)
     {
         isc.AddSingleton<WebSocketRouter>();
         isc.AddScoped<IWebSocketManager, WebSocketManager>();
         isc.AddSingleton<WebSocketMiddleware>();
+        isc.Configure("WebSocketOptions", socketOptions);
         return isc;
     }