Adding binary connection to WampSharp.Windows

Author: Elad Zelingher

src/net45/Default/WampSharp.Windows/WAMP2/V2/Fluent/MessageWebSocketActivator.cs

@@ -35,13 +35,25 @@ private IControlledWampConnection<TMessage> GetConnectionFactory<TMessage>(IWamp
                 return CreateTextConnection(textBinding);
             }
 
+            IWampBinaryBinding<TMessage> binaryBinding = binding as IWampBinaryBinding<TMessage>;
+
+            if (binaryBinding != null)
+            {
+                return CreateBinaryConnection(binaryBinding);
+            }
+
             throw new Exception();
         }
 
         protected IControlledWampConnection<TMessage> CreateTextConnection<TMessage>(IWampTextBinding<TMessage> textBinding)
         {
             return new MessageWebSocketTextConnection<TMessage>(mServerAddress, textBinding);
         }
+
+        protected IControlledWampConnection<TMessage> CreateBinaryConnection<TMessage>(IWampBinaryBinding<TMessage> binaryBinding)
+        {
+            return new MessageWebSocketBinaryConnection<TMessage>(mServerAddress, binaryBinding);
+        }
     }
 }
 

src/net45/Default/WampSharp.Windows/Windows/MessageWebSocketBinaryConnection.cs

@@ -0,0 +1,75 @@
+#if PCL
+using System;
+using System.IO;
+using System.Threading.Tasks;
+using WampSharp.V2.Binding;
+using Windows.Networking.Sockets;
+using WampSharp.Core.Message;
+
+namespace WampSharp.Windows
+{
+    public class MessageWebSocketBinaryConnection<TMessage> : MessageWebSocketConnection<TMessage>
+    {
+        private readonly IWampBinaryBinding<TMessage> mBinaryBinding;
+
+        public MessageWebSocketBinaryConnection(string uri, IWampBinaryBinding<TMessage> binding) :
+            base(uri, binding, SocketMessageType.Binary)
+        {
+            mBinaryBinding = binding;
+        }
+
+        // Regarding the async modifier - see https://github.com/Code-Sharp/WampSharp/issues/122
+        protected override async void OnMessageReceived(MessageWebSocket sender, MessageWebSocketMessageReceivedEventArgs args)
+        {
+            try
+            {
+                Stream stream = args.GetDataStream().AsStreamForRead();
+
+                MemoryStream memoryStream = new MemoryStream();
+
+                await stream.CopyToAsync(memoryStream);
+
+                stream.Position = 0;
+
+                WampMessage<TMessage> message = mBinaryBinding.Parse(stream);
+
+                RaiseMessageArrived(message);
+            }
+            catch (Exception ex)
+            {
+                RaiseConnectionError(ex);
+
+                if (mWebSocket != null)
+                {
+                    mWebSocket.Dispose();
+                }
+            }
+        }
+
+        protected override async Task SendAsync(WampMessage<object> message)
+        {
+            try
+            {
+                Stream stream = mWebSocket.OutputStream.AsStreamForWrite();
+
+                byte[] frame = mBinaryBinding.Format(message);
+
+                await stream.WriteAsync(frame, 0, frame.Length).ConfigureAwait(false);
+
+                await stream.FlushAsync().ConfigureAwait(false);
+            }
+            catch (Exception ex)
+            {
+                RaiseConnectionError(ex);
+
+                if (mWebSocket != null)
+                {
+                    mWebSocket.Dispose();
+                }
+
+                throw;
+            }
+        }
+    }
+}
+#endif

src/pcl/Default/WampSharp.Windows/WampSharp.Windows.csproj

@@ -65,6 +65,9 @@
     <Compile Include="..\..\..\net45\Default\WampSharp.Windows\Windows\MessageWebSocketTextConnection.cs">
       <Link>Windows\MessageWebSocketTextConnection.cs</Link>
     </Compile>
+    <Compile Include="..\..\..\net45\Default\WampSharp.Windows\Windows\MessageWebSocketBinaryConnection.cs">
+      <Link>Windows\MessageWebSocketBinaryConnection.cs</Link>
+    </Compile>
   </ItemGroup>
   <Import Project="$(MSBuildExtensionsPath32)\Microsoft\Portable\$(TargetFrameworkVersion)\Microsoft.Portable.CSharp.targets" />
   <!-- To modify your build process, add your task inside one of the targets below and uncomment it.