From cdb20e8d40c4282aeec2fcfa9a2ba714f5f4a85d Mon Sep 17 00:00:00 2001
From: calzoneman <calzoneman@gmail.com>
Date: Fri, 1 Jan 2016 18:25:12 -0800
Subject: [PATCH] Handle when a frontend disconnects

---
 src/io/backend/frontendmanager.js | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/io/backend/frontendmanager.js b/src/io/backend/frontendmanager.js
index b04f7d34..601c3a1e 100644
--- a/src/io/backend/frontendmanager.js
+++ b/src/io/backend/frontendmanager.js
@@ -15,10 +15,25 @@ export default class FrontendManager {
         }
 
         this.frontendConnections[socket.endpoint] = socket;
+        socket.on('close', this.onFrontendDisconnect.bind(this, socket));
         socket.on('SocketConnectEvent', this.onSocketConnect.bind(this, socket));
         socket.on('SocketFrameEvent', this.onSocketFrame.bind(this, socket));
     }
 
+    onFrontendDisconnect(socket) {
+        const endpoint = socket.endpoint;
+        if (this.frontendConnections.hasOwnProperty(endpoint)) {
+            if (this.frontendProxiedSockets.hasOwnProperty(endpoint)) {
+                logger.warn(`Frontend ${endpoint} disconnected`);
+                this.frontendProxiedSockets[endpoint].forEach(proxySocket => {
+                    proxySocket.onProxiedEventReceived('disconnect');
+                });
+                delete this.frontendProxiedSockets[endpoint];
+            }
+            delete this.frontendConnections[endpoint];
+        }
+    }
+
     onSocketConnect(frontendConnection, socketID, socketIP, socketUser) {
         const mapKey = frontendConnection.endpoint;
         const proxiedSocket = new ProxiedSocket(