From 25fc276d5e3f754f915500e91229b8607febc478 Mon Sep 17 00:00:00 2001
From: Stefan Malzner <stefan@adlk.io>
Date: Tue, 9 Apr 2019 01:20:35 +0200
Subject: Add settings websocket

---
 src/features/settingsWS/actions.js |  10 ++++
 src/features/settingsWS/index.js   |  35 ++++++++++++
 src/features/settingsWS/state.js   |  13 +++++
 src/features/settingsWS/store.js   | 107 +++++++++++++++++++++++++++++++++++++
 4 files changed, 165 insertions(+)
 create mode 100755 src/features/settingsWS/actions.js
 create mode 100755 src/features/settingsWS/index.js
 create mode 100755 src/features/settingsWS/state.js
 create mode 100755 src/features/settingsWS/store.js

(limited to 'src/features/settingsWS')

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;
-- 
cgit v1.2.3-54-g00ecf