WIP stuf

rainforestapp · Sep 16, 2016 · fd30ef8 · fd30ef8
1 parent 0e43900
commit fd30ef8
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/socket/index.js b/src/socket/index.js
@@ -50,6 +50,17 @@ export const startSocket = (store, socketConstructor = Socket) => {
     store.dispatch(startPolling(payload));
   };
 
+  const handleLeave = () => {
+    if (!channel) {
+      throw new Error("Leaving a channel when channel isn't joined");
+    }
+
+    channel.leave();
+    channel = null;
+    console.log("LEFT CHANNEL!");
+    store.dispatch(connectionClosed());
+  };
+
   const connectToSocket = ({ socket: socketState, worker }) => {
     const workerUUID = worker.get('uuid');
     const socketAuth = socketState.get('auth');
@@ -88,6 +99,9 @@ export const startSocket = (store, socketConstructor = Socket) => {
     channel.on('start_polling', payload => {
       handleStartPolling(payload);
     });
+    channel.on('leave', payload => {
+      handleLeave(payload);
+    });
     channel.join()
       .receive('ok', resp => {
         pushWorkerState(workerState());