handle client (dis)connects of both direct and bridge

Author: InventivetalentDev

KnockServer/InfoForm.Designer.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Reflection;
 using System.Windows.Forms;
 
 namespace KnockServer
@@ -19,6 +20,9 @@ public InfoForm()
             autostartCheckbox.Checked = Properties.Settings.Default.AutoStart;
             gameActivityCheckbox.Checked = Properties.Settings.Default.ShowActivity;
             connectionMethodComboBox.SelectedIndex = connectionMethodComboBox.FindStringExact(Properties.Settings.Default.ConnectionMethod);
+
+            versionLabel.Text = Assembly.GetExecutingAssembly().GetName().Version.ToString();
+            clientCountLabel.Text = CustomApplicationContext.GetClientCount().ToString()+" Client(s)";
         }
 
 

KnockServer/InfoForm.cs

@@ -180,7 +180,9 @@ public class CustomApplicationContext : ApplicationContext
 
         WebServiceHost hostWeb;
         WebSocketServer socketServer;
-         WebSocket socketClient;
+        static SocketServer innerSocketServer;
+         static WebSocket socketClient;
+         static SocketMessageHandler _messageHandler;
 
         public CustomApplicationContext()
         {
@@ -219,9 +221,12 @@ public CustomApplicationContext()
 
                     var port = 16945;
                     socketServer = new WebSocketServer(port);
-                    socketServer.AddWebSocketService<SocketServer>("/");
+                    socketServer.AddWebSocketService<SocketServer>("/", () =>
+                    {
+                        return innerSocketServer=  new SocketServer();
+                    });
                     socketServer.Start();
-
+                    
                     Console.WriteLine("Web Service Running!");
                     Console.WriteLine(socketServer.Address + ":" + port);
 
@@ -246,7 +251,7 @@ public CustomApplicationContext()
                         Origin ="ws://"+Properties.Settings.Default.ServerId+".servers.vrknock.app:16945"
                     };
 
-                    var _messageHandler = new SocketMessageHandler()
+                    _messageHandler = new SocketMessageHandler()
                     {
                         Send = (s,t) =>
                         {
@@ -300,6 +305,18 @@ public CustomApplicationContext()
             trayIcon.ShowBalloonTip(2000, "VRKnock", "Server Running!", ToolTipIcon.Info);
         }
 
+        public static int GetClientCount()
+        {
+            if (Properties.Settings.Default.ConnectionMethod == "DIRECT" && innerSocketServer != null)
+            {
+                return innerSocketServer.clientCount;
+            }else if (Properties.Settings.Default.ConnectionMethod == "BRIDGE" && _messageHandler != null)
+            {
+                return _messageHandler.clientCount;
+            }
+
+            return 0;
+        }
 
         void Info(object sender, EventArgs e)
         {

KnockServer/Program.cs

@@ -7,11 +7,29 @@ namespace KnockServer
 {
     public class SocketMessageHandler
     {
+        public int clientCount = 0;
+        
         public void OnMessage(MessageEventArgs e)
         {
             Console.WriteLine(e.Data);
             Request request = ParseRequest(e.Data);
 
+            if (request._state != null)
+            {
+                if (request._state == "CONNECT")
+                {
+                    clientCount++;
+                }
+
+                if (request._state == "DISCONNECT")
+                {
+                    clientCount--;
+                }
+
+                Console.WriteLine("ClientCount: "+clientCount);
+                return;
+            }
+
             Status status = null;
             switch (request.action)
             {
@@ -149,6 +167,7 @@ protected virtual string SendImpl(string text, string target)
 
     public class Request
     {
+        public string _state { get; set; }
         public string action { get; set; }
         public string code { get; set; }
         public string message { get; set; }

KnockServer/SocketMessageHandler.cs

@@ -11,6 +11,8 @@ public class SocketServer : WebSocketBehavior
     {
         private SocketMessageHandler _messageHandler;
 
+        public int clientCount = 0;
+        
         public SocketServer()
         {
             _messageHandler = new SocketMessageHandler()
@@ -28,12 +30,16 @@ protected override void OnClose(CloseEventArgs e)
         {
             base.OnClose(e);
             Console.WriteLine("A Client Disconnected.");
+            clientCount--;
+            Console.WriteLine("ClientCount: "+clientCount);
         }
 
         protected override void OnOpen()
         {
             base.OnOpen();
             Console.WriteLine("New Client Connected!");
+            clientCount++;
+            Console.WriteLine("ClientCount: "+clientCount);
         }