Add Protocol implementation

2019-03-19 21:02:10 +01:00
parent 268e04a710
commit 58fbaf4e35
4 changed files with 143 additions and 0 deletions
--- a/PlantBox.Shared/Communication/CommandStream.cs
+++ b/PlantBox.Shared/Communication/CommandStream.cs
@@ -0,0 +1,80 @@
+using System;
+using System.Collections.Generic;
+using System.Net.Sockets;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace PlantBox.Shared.Communication
+{
+    public class CommandStream
+    {
+        private NetworkStream _stream;
+
+        public CommandStream(NetworkStream networkStream)
+        {
+            _stream = networkStream;
+        }
+
+        public CommandPacket Receive()
+        {
+            // Length
+            byte[] buffer = new byte[4];
+            _stream.Read(buffer, 0, buffer.Length);
+
+            int length = BitConverter.ToInt32(buffer, 0);
+
+            // Data
+            buffer = new byte[length];
+            _stream.Read(buffer, 0, buffer.Length);
+
+            string[] packet = Encoding.UTF8.GetString(buffer).Split('\n');
+
+            if (!Enum.TryParse(packet[0], true, out Command command))
+            {
+                command = Command.Invalid;
+            }
+
+            return new CommandPacket(command, packet[1].Split(';'));
+        }
+
+        public async Task<CommandPacket> ReceiveAsync()
+        {
+            // Length
+            byte[] buffer = new byte[4];
+            await _stream.ReadAsync(buffer, 0, buffer.Length);
+
+            int length = BitConverter.ToInt32(buffer, 0);
+
+            // Data
+            buffer = new byte[length];
+            await _stream.ReadAsync(buffer, 0, buffer.Length);
+
+            string[] packet = Encoding.UTF8.GetString(buffer).Split('\n');
+
+            if (!Enum.TryParse(packet[0], true, out Command command))
+            {
+                command = Command.Invalid;
+            }
+
+            return new CommandPacket(command, packet[1].Split(';'));
+        }
+
+        public void Send(CommandPacket command)
+        {
+            string packet = command.ToString();
+            byte[] data = Encoding.UTF8.GetBytes(packet);
+
+            _stream.Write(BitConverter.GetBytes(data.Length), 0, 4);
+            _stream.Write(data, 0, data.Length);
+        }
+
+        public async Task SendAsync(CommandPacket command)
+        {
+            string packet = command.ToString();
+            byte[] data = Encoding.UTF8.GetBytes(packet);
+
+            await _stream.WriteAsync(BitConverter.GetBytes(data.Length), 0, 4);
+            await _stream.WriteAsync(data, 0, data.Length);
+        }
+    }
+}