MediaBrowser.Server.Implementations/WebSocket/FleckServer.cs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

using Fleck;
using MediaBrowser.Common.Net;
using System;
using IWebSocketServer = MediaBrowser.Common.Net.IWebSocketServer;

namespace MediaBrowser.Server.Implementations.WebSocket
{
    public class FleckServer : IWebSocketServer
    {
        private WebSocketServer _server;

        public void Start(int portNumber)
        {
            var server = new WebSocketServer("ws://localhost:" + portNumber);

            server.Start(socket =>
            {
                socket.OnOpen = () => OnClientConnected(socket);
            });

            _server = server;
        }

        public void Stop()
        {
            _server.Dispose();
        }

        private void OnClientConnected(Fleck.IWebSocketConnection context)
        {
            if (WebSocketConnected != null)
            {
                var socket = new FleckWebSocket(context);

                WebSocketConnected(this, new WebSocketConnectEventArgs
                {
                    WebSocket = socket,
                    Endpoint = context.ConnectionInfo.ClientIpAddress + ":" + context.ConnectionInfo.ClientPort
                });
            }
        }
        public event EventHandler<WebSocketConnectEventArgs> WebSocketConnected;

        public int Port
        {
            get { return _server.Port; }
        }

        public void Dispose()
        {
            _server.Dispose();
        }
    }
}