web/src/js/connection.js


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
40
41

import {ConnectionActions} from "./actions.js";
import {AppDispatcher} from "./dispatcher.js";
import * as webSocketActions from "./ducks/websocket"
import * as eventLogActions from "./ducks/eventLog"
import * as flowActions from "./ducks/flows"

export default function Connection(url, dispatch) {
    if (url[0] === "/") {
        url = location.origin.replace("http", "ws") + url;
    }

    var ws = new WebSocket(url);
    ws.onopen = function () {
        dispatch(webSocketActions.connected())
        dispatch(flowActions.fetchFlows())
        dispatch(eventLogActions.fetchLogEntries())
        ConnectionActions.open()
    };
    ws.onmessage = function (m) {
        var message = JSON.parse(m.data);
        AppDispatcher.dispatchServerAction(message);
        switch (message.type) {
            case eventLogActions.UPDATE_LOG:
                return dispatch(eventLogActions.updateLogEntries(message))
            case flowActions.UPDATE_FLOWS:
                return dispatch(flowActions.updateFlows(message))
            default:
                console.warn("unknown message", message)
        }
    };
    ws.onerror = function () {
        ConnectionActions.error();
        dispatch(eventLogActions.addLogEntry("WebSocket connection error."));
    };
    ws.onclose = function () {
        ConnectionActions.close();
        dispatch(eventLogActions.addLogEntry("WebSocket connection closed."));
        dispatch(webSocketActions.disconnected());
    };
    return ws;
}