summaryrefslogtreecommitdiff
path: root/src/web/app/common/scripts/stream.js
blob: b31e570444ac79b3c5f17314bb6fda73d1036225 (plain)
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
const ReconnectingWebSocket = require('reconnecting-websocket');
const riot = require('riot');

module.exports = me => {
	let state = 'initializing';
	const stateEv = riot.observable();
	const event = riot.observable();
	const host = CONFIG.api.url.replace('http', 'ws');
	const socket = new ReconnectingWebSocket(`${host}?i=${me.token}`);

	socket.onopen = () => {
		state = 'connected';
		stateEv.trigger('connected');
	};

	socket.onclose = () => {
		state = 'reconnecting';
		stateEv.trigger('closed');
	};

	socket.onmessage = message => {
		try {
			const message = JSON.parse(message.data);
			if (message.type) {
				event.trigger(message.type, message.body);
			}
		} catch (e) {
			// noop
		}
	};

	event.on('i_updated', me.update);

	return {
		stateEv: stateEv,
		getState: () => state,
		event: event
	};
};