7 files changed, 179 insertions, 0 deletions

diff --git a/src/config.js b/src/config.js
index 4423e61e9..05ee07ee5 100644
--- a/src/config.js
+++ b/src/config.js
@@ -13,6 +13,10 @@ export const LOCAL_API = 'http://localhost:3000';
 export const DEV_API = 'https://dev.franzinfra.com';
 export const LIVE_API = 'https://api.franzinfra.com';
 
+export const LOCAL_WS_API = 'ws://localhost:3000';
+export const DEV_WS_API = 'wss://dev.franzinfra.com';
+export const LIVE_WS_API = 'wss://api.franzinfra.com';
+
 export const LOCAL_API_WEBSITE = 'http://localhost:3333';
 export const DEV_API_WEBSITE = 'https://meetfranz.com';
 export const LIVE_API_WEBSITE = 'https://meetfranz.com';
diff --git a/src/environment.js b/src/environment.js
index 68fa45173..e47b373b0 100644
--- a/src/environment.js
+++ b/src/environment.js
@@ -7,6 +7,9 @@ import {
   LOCAL_API_WEBSITE,
   DEV_API_WEBSITE,
   LIVE_API_WEBSITE,
+  LIVE_WS_API,
+  LOCAL_WS_API,
+  DEV_WS_API,
 } from './config';
 
 export const isDevMode = isDev;
@@ -26,17 +29,22 @@ export const ctrlKey = isMac ? '⌘' : 'Ctrl';
 export const cmdKey = isMac ? 'Cmd' : 'Ctrl';
 
 let api;
+let wsApi;
 let web;
 if (!isDevMode || (isDevMode && useLiveAPI)) {
   api = LIVE_API;
+  wsApi = LIVE_WS_API;
   web = LIVE_API_WEBSITE;
 } else if (isDevMode && useLocalAPI) {
   api = LOCAL_API;
+  wsApi = LOCAL_WS_API;
   web = LOCAL_API_WEBSITE;
 } else {
   api = DEV_API;
+  wsApi = DEV_WS_API;
   web = DEV_API_WEBSITE;
 }
 
 export const API = api;
+export const WS_API = wsApi;
 export const WEBSITE = web;
diff --git a/src/features/settingsWS/actions.js b/src/features/settingsWS/actions.js
new file mode 100755
index 000000000..631670c8a
--- /dev/null
+++ b/src/features/settingsWS/actions.js
@@ -0,0 +1,10 @@
+import PropTypes from 'prop-types';
+import { createActionsFromDefinitions } from '../../actions/lib/actions';
+
+export const settingsWSActions = createActionsFromDefinitions({
+  greet: {
+    name: PropTypes.string.isRequired,
+  },
+}, PropTypes.checkPropTypes);
+
+export default settingsWSActions;
diff --git a/src/features/settingsWS/index.js b/src/features/settingsWS/index.js
new file mode 100755
index 000000000..1e268f184
--- /dev/null
+++ b/src/features/settingsWS/index.js
@@ -0,0 +1,35 @@
+import { reaction, runInAction } from 'mobx';
+import { SettingsWSStore } from './store';
+import state, { resetState } from './state';
+
+const debug = require('debug')('Franz:feature:settingsWS');
+
+let store = null;
+
+export default function initAnnouncements(stores, actions) {
+  const { features } = stores;
+
+  // Toggle workspace feature
+  reaction(
+    () => (
+      features.features.isSettingsWSEnabled
+    ),
+    (isEnabled) => {
+      if (isEnabled) {
+        debug('Initializing `settingsWS` feature');
+        store = new SettingsWSStore(stores, null, actions, state);
+        store.initialize();
+        runInAction(() => { state.isFeatureActive = true; });
+      } else if (store) {
+        debug('Disabling `settingsWS` feature');
+        runInAction(() => { state.isFeatureActive = false; });
+        store.teardown();
+        store = null;
+        resetState(); // Reset state to default
+      }
+    },
+    {
+      fireImmediately: true,
+    },
+  );
+}
diff --git a/src/features/settingsWS/state.js b/src/features/settingsWS/state.js
new file mode 100755
index 000000000..7b16b2b6e
--- /dev/null
+++ b/src/features/settingsWS/state.js
@@ -0,0 +1,13 @@
+import { observable } from 'mobx';
+
+const defaultState = {
+  isFeatureActive: false,
+};
+
+export const settingsWSState = observable(defaultState);
+
+export function resetState() {
+  Object.assign(settingsWSState, defaultState);
+}
+
+export default settingsWSState;
diff --git a/src/features/settingsWS/store.js b/src/features/settingsWS/store.js
new file mode 100755
index 000000000..21fd3440e
--- /dev/null
+++ b/src/features/settingsWS/store.js
@@ -0,0 +1,107 @@
+import { observable, reaction } from 'mobx';
+import WebSocket from 'ws';
+import ms from 'ms';
+
+import Store from '../../stores/lib/Store';
+import { WS_API } from '../../environment';
+
+const debug = require('debug')('Franz:feature:settingsWS:store');
+
+export class SettingsWSStore extends Store {
+  ws = null;
+
+  @observable connected = false;
+
+  pingTimeout = null;
+
+  reconnectTimeout = null;
+
+  constructor(stores, api, actions, state) {
+    super(stores, api, actions);
+    this.state = state;
+  }
+
+  setup() {
+    this.connect();
+
+    reaction(() => !this.connected, this.reconnect.bind(this));
+  }
+
+  connect() {
+    try {
+      const wsURL = `${WS_API}/ws/${this.stores.user.data.id}`;
+      debug('Setting up WebSocket to', wsURL);
+
+      this.ws = new WebSocket(wsURL);
+
+      this.ws.on('open', () => {
+        debug('Opened WebSocket');
+        this.send({
+          action: 'authorize',
+          token: this.stores.user.authToken,
+        });
+
+        this.connected = true;
+
+        this.heartbeat();
+      });
+
+      this.ws.on('message', (data) => {
+        const resp = JSON.parse(data);
+        debug('Received message', resp);
+
+        if (resp.id) {
+          this.stores.user.getUserInfoRequest.patch((result) => {
+            if (!result) return;
+
+            debug('Patching user object with new values');
+            Object.assign(result, resp);
+          });
+        }
+      });
+
+      this.ws.on('ping', this.heartbeat.bind(this));
+    } catch (err) {
+      console.err(err);
+    }
+  }
+
+  heartbeat() {
+    debug('Heartbeat');
+    clearTimeout(this.pingTimeout);
+
+    // Use `WebSocket#terminate()` and not `WebSocket#close()`. Delay should be
+    // equal to the interval at which your server sends out pings plus a
+    // conservative assumption of the latency.
+    this.pingTimeout = setTimeout(() => {
+      debug('Terminating connection reconnecting in 35');
+      this.ws.terminate();
+
+      this.connected = false;
+    }, ms('35s'));
+  }
+
+  send(data) {
+    if (this.ws) {
+      this.ws.send(JSON.stringify(data));
+      debug('Sending data', data);
+    } else {
+      debug('WebSocket is not initialized');
+    }
+  }
+
+  reconnect() {
+    if (!this.connected) {
+      debug('Trying to reconnect in 30s');
+      this.reconnectTimeout = setInterval(() => {
+        debug('Trying to reconnect');
+        this.connect();
+      }, ms('30s'));
+    } else {
+      debug('Clearing reconnect interval');
+      clearInterval(this.reconnectTimeout);
+    }
+  }
+}
+
+export default SettingsWSStore;
diff --git a/src/stores/FeaturesStore.js b/src/stores/FeaturesStore.js
index d2842083c..e60d79075 100644
--- a/src/stores/FeaturesStore.js
+++ b/src/stores/FeaturesStore.js
@@ -8,6 +8,7 @@ import spellchecker from '../features/spellchecker';
 import serviceProxy from '../features/serviceProxy';
 import basicAuth from '../features/basicAuth';
 import shareFranz from '../features/shareFranz';
+import settingsWS from '../features/settingsWS';
 
 import { DEFAULT_FEATURES_CONFIG } from '../config';
 
@@ -58,5 +59,6 @@ export default class FeaturesStore extends Store {
     serviceProxy(this.stores, this.actions);
     basicAuth(this.stores, this.actions);
     shareFranz(this.stores, this.actions);
+    settingsWS(this.stores, this.actions);
   }
 }