initiating a live casting

Yavsc/Controllers/Communicating/LiveController.cs

@@ -0,0 +1,83 @@
+using System;
+using System.Collections.Concurrent;
+using System.Collections.Generic;
+using System.Net.WebSockets;
+using System.Security.Claims;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.AspNet.Mvc;
+using Microsoft.Extensions.Logging;
+using Yavsc.ViewModels.Streaming;
+
+namespace Yavsc.Controllers.Communicating
+{
+    public class LiveController : Controller
+    {
+        ILogger _logger;
+        public static ConcurrentDictionary<string, LiveCastMeta> Casters = new ConcurrentDictionary<string, LiveCastMeta>();
+        public LiveController(LoggerFactory loggerFactory)
+        {
+            _logger = loggerFactory.CreateLogger<LiveController>();
+        }
+
+        public async Task<IActionResult> Cast()
+        {
+            var uname = User.GetUserName();
+            // ensure this request is for a websocket
+            if (!HttpContext.WebSockets.IsWebSocketRequest) return new BadRequestResult();
+            // ensure uniqueness of casting stream from this user
+            var existent = Casters[uname];
+            if (existent != null) return new BadRequestObjectResult("not supported, you already casting, there's support for one live streaming only");
+            var meta = new LiveCastMeta { Socket = await HttpContext.WebSockets.AcceptWebSocketAsync() };
+
+            using (meta.Socket)
+            {
+                if (meta.Socket != null && meta.Socket.State == WebSocketState.Open)
+                {
+                    Casters[uname] = meta;
+                    // TODO: Handle the socket here.
+                    // Find receivers: others in the chat room
+                    // send them the flow
+
+                    byte[] buffer = new byte[1024];
+                    WebSocketReceiveResult received = await meta.Socket.ReceiveAsync
+                    (new ArraySegment<byte>(buffer), CancellationToken.None);
+
+                    // FIXME do we really need to close those one in invalid state ?
+                    Stack<string> ToClose = new Stack<string>();
+
+                    while (received.MessageType != WebSocketMessageType.Close)
+                    {
+                        _logger.LogInformation($"Echoing {received.Count} bytes received in a {received.MessageType} message; Fin={received.EndOfMessage}");
+                        // Echo anything we receive
+                        // and send to all listner found
+                        foreach (var cliItem in meta.Listeners)
+                        {
+                            var listenningSocket = cliItem.Value;
+                            if (listenningSocket.State == WebSocketState.Open)
+                                await listenningSocket.SendAsync(new ArraySegment<byte>
+                                (buffer, 0, received.Count), received.MessageType, received.EndOfMessage, CancellationToken.None);
+                            else ToClose.Push(cliItem.Key);
+                        }
+                        received = await meta.Socket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);
+
+                        string no;
+                        do
+                        {
+                            no = ToClose.Pop();
+                            WebSocket listenningSocket;
+                            if (meta.Listeners.TryRemove(no, out listenningSocket))
+                                await listenningSocket.CloseAsync(WebSocketCloseStatus.EndpointUnavailable, "State != WebSocketState.Open", CancellationToken.None);
+
+                        } while (no != null);
+                    }
+                    await meta.Socket.CloseAsync(received.CloseStatus.Value, received.CloseStatusDescription, CancellationToken.None);
+                    Casters[uname] = null;
+                }
+                else _logger.LogInformation($"failed (meta.Socket != null && meta.Socket.State == WebSocketState.Open)");
+            }
+
+            return Ok();
+        }
+    }
+}