Support for server behaviors

packages/repositories.config

@@ -1,5 +1,7 @@
 <?xml version="1.0" encoding="utf-8"?>
 <repositories>
+  <repository path="../test/Integration/Alchemy/packages.config" />
+  <repository path="../test/Unit/Alchemy/packages.config" />
   <repository path="..\test\Integration\Alchemy\packages.config" />
   <repository path="..\test\Unit\Alchemy\packages.config" />
 </repositories>

src/Alchemy/Alchemy.csproj

@@ -71,6 +71,7 @@
     <Compile Include="Classes\Header.cs" />
     <Compile Include="Classes\Response.cs" />
     <Compile Include="Classes\UserContext.cs" />
+    <Compile Include="Classes\ServerBehaviour.cs" />
     <Compile Include="Handlers\Handler.cs" />
     <Compile Include="Handlers\WebSocket\hybi00\Handler.cs" />
     <Compile Include="Handlers\WebSocket\hybi00\Handshakes.cs" />
@@ -80,7 +81,7 @@
     <Compile Include="Handlers\WebSocket\rfc6455\Handshakes.cs" />
     <Compile Include="Handlers\WebSocket\rfc6455\Handler.cs" />
     <Compile Include="Handlers\WebSocket\WebSocketHandler.cs" />
-    <Compile Include="TcpServer.cs" />
+    <Compile Include="TCPServer.cs" />
     <Compile Include="WebSocketClient.cs" />
     <Compile Include="WebSocketServer.cs" />
     <Compile Include="Properties\AssemblyInfo.cs" />

src/Alchemy/Classes/ServerBehaviour.cs

@@ -0,0 +1,36 @@
+namespace Alchemy.Classes
+{
+    /// <summary>
+    /// The server behaviour base class.
+    /// </summary>
+    public abstract class ServerBehaviour
+    {
+        /// <summary>
+        /// The event triggered when the user is connected.
+        /// </summary>
+        /// <param name="context">The current user context.</param>
+        public virtual void OnConnected(UserContext context)
+        { }
+
+        /// <summary>
+        /// The event triggered when the server receive data.
+        /// </summary>
+        /// <param name="context">The current user context.</param>
+        public virtual void OnReceive(UserContext context)
+        { }
+
+        /// <summary>
+        /// The event triggered when the server send data.
+        /// </summary>
+        /// <param name="context">The current user context.</param>
+        public virtual void OnSend(UserContext context)
+        { }
+
+        /// <summary>
+        /// The event triggered when the user is disconnected.
+        /// </summary>
+        /// <param name="context">The current user context.</param>
+        public virtual void OnDisconnect(UserContext context)
+        { }
+    }
+}

src/Alchemy/WebSocketClient.cs

@@ -60,7 +60,8 @@ static WebSocketClient()
             NewClients = new Queue<Context>();
             ContextMapping = new Dictionary<Context, WebSocketClient>();
 
-            for(int i = 0; i < ClientThreads.Length; i++){
+            for (int i = 0; i < ClientThreads.Length; i++)
+            {
                 ClientThreads[i] = new Thread(HandleClientThread);
                 ClientThreads[i].Start();
             }
@@ -97,9 +98,12 @@ private static void HandleClientThread()
         }
         public WebSocketClient(string path)
         {
-            var r = new Regex("^(wss?)://(.*)\\:([0-9]*)/(.*)$");
+            var r = new Regex("^(\\w+)://(.*)\\:([0-9]*)/(.*)$");
             var matches = r.Match(path);
 
+            if (matches.Groups[1].Value != "ws" || matches.Groups[1].Value != "wss")
+                SubProtocols = new string[] {matches.Groups[1].Value};
+
             _host = matches.Groups[2].Value;
             _port = Int32.Parse(matches.Groups[3].Value);
             _path = matches.Groups[4].Value;
@@ -108,7 +112,7 @@ public WebSocketClient(string path)
         public void Connect()
         {
             if (_client != null) return;
-            
+
             try
             {
                 ReadyState = ReadyStates.CONNECTING;
@@ -142,7 +146,7 @@ protected void OnRunClient(IAsyncResult result)
             {
                 _client.EndConnect(result);
             }
-            catch (Exception ex)
+            catch (Exception)
             {
                 Disconnect();
                 connectError = true;
@@ -192,7 +196,7 @@ private void SetupContext(Context context)
                 {
                     ReceiveEventArgs_Completed(_context.Connection.Client, _context.ReceiveEventArgs);
                 }
- 
+
 
                 if (!IsAuthenticated)
                 {
@@ -236,7 +240,7 @@ void ReceiveEventArgs_Completed(object sender, SocketAsyncEventArgs e)
 
         private void Authenticate()
         {
-            _handshake = new ClientHandshake { Version = "8", Origin = Origin, Host = _host, Key = GenerateKey(), ResourcePath = _path, SubProtocols = SubProtocols};
+            _handshake = new ClientHandshake { Version = "8", Origin = Origin, Host = _host, Key = GenerateKey(), ResourcePath = _path, SubProtocols = SubProtocols };
 
             _client.Client.Send(Encoding.UTF8.GetBytes(_handshake.ToString()));
         }
@@ -264,7 +268,7 @@ private bool CheckAuthenticationResponse(Context context)
                     }
 
                 }
-                if(String.IsNullOrEmpty(CurrentProtocol))
+                if (String.IsNullOrEmpty(CurrentProtocol))
                 {
                     return false;
                 }
@@ -305,7 +309,7 @@ private void ReceiveData(Context context)
 
         private void DoReceive(IAsyncResult result)
         {
-            var context = (Context) result.AsyncState;
+            var context = (Context)result.AsyncState;
             context.Reset();
 
             try
@@ -335,7 +339,7 @@ private static String GenerateKey()
 
             for (var index = 0; index < bytes.Length; index++)
             {
-                bytes[index] = (byte) random.Next(0, 255);
+                bytes[index] = (byte)random.Next(0, 255);
             }
 
             return Convert.ToBase64String(bytes);
@@ -369,12 +373,12 @@ public void Send(byte[] data)
         {
             _context.UserContext.Send(data);
         }
-        
+
         public void Dispose()
         {
             cancellation.Cancel();
             Handler.Instance.Dispose();
         }
-        
+
     }
-}
+}