remove onmessage, onconnect, and ondisconnect from websocket

Author: HomelessDinosaur

src/Nitric.Sdk/Resource/WebsocketResource.cs

@@ -86,49 +86,15 @@ public void On(Service.WebsocketEventType eventType, params Middleware<Websocket
         }
 
         /// <summary>
-        /// Registers a handler to be called whenever a connection event is published to this websocket.
+        /// Registers a handler to be called whenever an event is published to this websocket.
         /// </summary>
         /// <param name="handler">The handler to call to process websocket events</param>
-        public void OnConnect(Func<WebsocketContext, Task<WebsocketContext>> handler)
+        public void On(Service.WebsocketEventType eventType, Func<WebsocketContext, Task<WebsocketContext>> handler)
         {
             var registrationRequest = new RegistrationRequest
             {
                 SocketName = Name,
-                EventType = Service.WebsocketEventType.Connected.ToGrpc()
-            };
-
-            var websocketWorker = new WebsocketWorker(registrationRequest, handler);
-
-            Nitric.RegisterWorker(websocketWorker);
-        }
-
-        /// <summary>
-        /// Registers a handler to be called whenever a disconnection event is published to this websocket.
-        /// </summary>
-        /// <param name="handler">The handler to call to process websocket events</param>
-        public void OnDisconnect(Func<WebsocketContext, Task<WebsocketContext>> handler)
-        {
-            var registrationRequest = new RegistrationRequest
-            {
-                SocketName = Name,
-                EventType = Service.WebsocketEventType.Disconnected.ToGrpc()
-            };
-
-            var websocketWorker = new WebsocketWorker(registrationRequest, handler);
-
-            Nitric.RegisterWorker(websocketWorker);
-        }
-
-        /// <summary>
-        /// Registers a handler to be called whenever a message event is published to this websocket.
-        /// </summary>
-        /// <param name="handler">The handler to call to process websocket events</param>
-        public void OnMessage(Func<WebsocketContext, Task<WebsocketContext>> handler)
-        {
-            var registrationRequest = new RegistrationRequest
-            {
-                SocketName = Name,
-                EventType = Service.WebsocketEventType.Message.ToGrpc()
+                EventType = eventType.ToGrpc()
             };
 
             var websocketWorker = new WebsocketWorker(registrationRequest, handler);