diff options
Diffstat (limited to 'MediaBrowser.Networking/WebSocket')
| -rw-r--r-- | MediaBrowser.Networking/WebSocket/AlchemyServer.cs | 105 | ||||
| -rw-r--r-- | MediaBrowser.Networking/WebSocket/AlchemyWebSocket.cs | 133 |
2 files changed, 238 insertions, 0 deletions
diff --git a/MediaBrowser.Networking/WebSocket/AlchemyServer.cs b/MediaBrowser.Networking/WebSocket/AlchemyServer.cs new file mode 100644 index 000000000..1a3971c7f --- /dev/null +++ b/MediaBrowser.Networking/WebSocket/AlchemyServer.cs @@ -0,0 +1,105 @@ +using Alchemy; +using Alchemy.Classes; +using MediaBrowser.Common.Net; +using MediaBrowser.Model.Logging; +using System; +using System.Net; + +namespace MediaBrowser.Networking.WebSocket +{ + /// <summary> + /// Class AlchemyServer + /// </summary> + public class AlchemyServer : IWebSocketServer + { + /// <summary> + /// Occurs when [web socket connected]. + /// </summary> + public event EventHandler<WebSocketConnectEventArgs> WebSocketConnected; + + /// <summary> + /// Gets or sets the web socket server. + /// </summary> + /// <value>The web socket server.</value> + private WebSocketServer WebSocketServer { get; set; } + + /// <summary> + /// The _logger + /// </summary> + private readonly ILogger _logger; + + /// <summary> + /// Initializes a new instance of the <see cref="AlchemyServer" /> class. + /// </summary> + /// <param name="logger">The logger.</param> + /// <exception cref="System.ArgumentNullException">logger</exception> + public AlchemyServer(ILogger logger) + { + if (logger == null) + { + throw new ArgumentNullException("logger"); + } + _logger = logger; + } + + /// <summary> + /// Starts the specified port number. + /// </summary> + /// <param name="portNumber">The port number.</param> + public void Start(int portNumber) + { + WebSocketServer = new WebSocketServer(portNumber, IPAddress.Any) + { + OnConnected = OnAlchemyWebSocketClientConnected, + TimeOut = TimeSpan.FromMinutes(60) + }; + + WebSocketServer.Start(); + + _logger.Info("Alchemy Web Socket Server started"); + } + + /// <summary> + /// Called when [alchemy web socket client connected]. + /// </summary> + /// <param name="context">The context.</param> + private void OnAlchemyWebSocketClientConnected(UserContext context) + { + if (WebSocketConnected != null) + { + var socket = new AlchemyWebSocket(context, _logger); + + WebSocketConnected(this, new WebSocketConnectEventArgs + { + WebSocket = socket, + Endpoint = context.ClientAddress.ToString() + }); + } + } + + /// <summary> + /// Stops this instance. + /// </summary> + public void Stop() + { + } + + /// <summary> + /// Performs application-defined tasks associated with freeing, releasing, or resetting unmanaged resources. + /// </summary> + public void Dispose() + { + Dispose(true); + GC.SuppressFinalize(this); + } + + /// <summary> + /// Releases unmanaged and - optionally - managed resources. + /// </summary> + /// <param name="dispose"><c>true</c> to release both managed and unmanaged resources; <c>false</c> to release only unmanaged resources.</param> + protected virtual void Dispose(bool dispose) + { + + } + } +} diff --git a/MediaBrowser.Networking/WebSocket/AlchemyWebSocket.cs b/MediaBrowser.Networking/WebSocket/AlchemyWebSocket.cs new file mode 100644 index 000000000..fbdb3186f --- /dev/null +++ b/MediaBrowser.Networking/WebSocket/AlchemyWebSocket.cs @@ -0,0 +1,133 @@ +using Alchemy.Classes; +using MediaBrowser.Common.Net; +using MediaBrowser.Common.Serialization; +using MediaBrowser.Model.Logging; +using System; +using System.Net.WebSockets; +using System.Threading; +using System.Threading.Tasks; + +namespace MediaBrowser.Networking.WebSocket +{ + /// <summary> + /// Class AlchemyWebSocket + /// </summary> + public class AlchemyWebSocket : IWebSocket + { + /// <summary> + /// The logger + /// </summary> + private readonly ILogger _logger; + + /// <summary> + /// Gets or sets the web socket. + /// </summary> + /// <value>The web socket.</value> + private UserContext UserContext { get; set; } + + /// <summary> + /// Initializes a new instance of the <see cref="AlchemyWebSocket" /> class. + /// </summary> + /// <param name="context">The context.</param> + /// <param name="logger">The logger.</param> + /// <exception cref="System.ArgumentNullException">context</exception> + public AlchemyWebSocket(UserContext context, ILogger logger) + { + if (context == null) + { + throw new ArgumentNullException("context"); + } + + _logger = logger; + UserContext = context; + + context.SetOnDisconnect(OnDisconnected); + context.SetOnReceive(OnReceive); + + _logger.Info("Client connected from {0}", context.ClientAddress); + } + + /// <summary> + /// The _disconnected + /// </summary> + private bool _disconnected = false; + /// <summary> + /// Gets or sets the state. + /// </summary> + /// <value>The state.</value> + public WebSocketState State + { + get { return _disconnected ? WebSocketState.Closed : WebSocketState.Open; } + } + + /// <summary> + /// Called when [disconnected]. + /// </summary> + /// <param name="context">The context.</param> + private void OnDisconnected(UserContext context) + { + _disconnected = true; + } + + /// <summary> + /// Called when [receive]. + /// </summary> + /// <param name="context">The context.</param> + private void OnReceive(UserContext context) + { + if (OnReceiveDelegate != null) + { + var json = context.DataFrame.ToString(); + + if (!string.IsNullOrWhiteSpace(json)) + { + try + { + var messageResult = JsonSerializer.DeserializeFromString<WebSocketMessageInfo>(json); + + OnReceiveDelegate(messageResult); + } + catch (Exception ex) + { + _logger.ErrorException("Error processing web socket message", ex); + } + } + } + } + + /// <summary> + /// Sends the async. + /// </summary> + /// <param name="bytes">The bytes.</param> + /// <param name="type">The type.</param> + /// <param name="endOfMessage">if set to <c>true</c> [end of message].</param> + /// <param name="cancellationToken">The cancellation token.</param> + /// <returns>Task.</returns> + public Task SendAsync(byte[] bytes, WebSocketMessageType type, bool endOfMessage, CancellationToken cancellationToken) + { + return Task.Run(() => UserContext.Send(bytes)); + } + + /// <summary> + /// Performs application-defined tasks associated with freeing, releasing, or resetting unmanaged resources. + /// </summary> + public void Dispose() + { + Dispose(true); + } + + /// <summary> + /// Releases unmanaged and - optionally - managed resources. + /// </summary> + /// <param name="dispose"><c>true</c> to release both managed and unmanaged resources; <c>false</c> to release only unmanaged resources.</param> + protected virtual void Dispose(bool dispose) + { + } + + /// <summary> + /// Gets or sets the receive action. + /// </summary> + /// <value>The receive action.</value> + public Action<WebSocketMessageInfo> OnReceiveDelegate { get; set; } + } +} |
