mattermost/web/react/stores/socket_store.jsx

// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved.
// See License.txt for license information.

import UserStore from './user_store.jsx';
import PostStore from './post_store.jsx';
import ChannelStore from './channel_store.jsx';
import BrowserStore from './browser_store.jsx';
import ErrorStore from './error_store.jsx';
import EventEmitter from 'events';

import * as Utils from '../utils/utils.jsx';
import * as AsyncClient from '../utils/async_client.jsx';
import * as EventHelpers from '../dispatcher/event_helpers.jsx';

import Constants from '../utils/constants.jsx';
const SocketEvents = Constants.SocketEvents;

const CHANGE_EVENT = 'change';

var conn;

class SocketStoreClass extends EventEmitter {
    constructor() {
        super();

        this.initialize = this.initialize.bind(this);
        this.emitChange = this.emitChange.bind(this);
        this.addChangeListener = this.addChangeListener.bind(this);
        this.removeChangeListener = this.removeChangeListener.bind(this);
        this.sendMessage = this.sendMessage.bind(this);
        this.failCount = 0;

        this.initialize();
    }
    initialize() {
        if (!UserStore.getCurrentId()) {
            return;
        }

        if (!global.window.hasOwnProperty('mm_session_token_index')) {
            return;
        }

        this.setMaxListeners(0);

        if (window.WebSocket && !conn) {
            var protocol = 'ws://';
            if (window.location.protocol === 'https:') {
                protocol = 'wss://';
            }

            var connUrl = protocol + location.host + ((/:\d+/).test(location.host) ? '' : Utils.getWebsocketPort(protocol)) + '/api/v1/websocket?' + Utils.getSessionIndex();

            if (this.failCount === 0) {
                console.log('websocket connecting to ' + connUrl); //eslint-disable-line no-console
            }
            conn = new WebSocket(connUrl);

            conn.onopen = () => {
                if (this.failCount > 0) {
                    console.log('websocket re-established connection'); //eslint-disable-line no-console

                    if (ErrorStore.getLastError()) {
                        ErrorStore.storeLastError(null);
                        ErrorStore.emitChange();
                    }
                }

                this.failCount = 0;
            };

            conn.onclose = () => {
                conn = null;
                setTimeout(
                    () => {
                        this.initialize();
                    },
                    3000
                );
            };

            conn.onerror = (evt) => {
                if (this.failCount === 0) {
                    console.log('websocket error ' + evt); //eslint-disable-line no-console
                }

                this.failCount = this.failCount + 1;

                ErrorStore.storeLastError({connErrorCount: this.failCount, message: 'Please check connection, Mattermost unreachable. If issue persists, ask administrator to check WebSocket port.'});
                ErrorStore.emitChange();
            };

            conn.onmessage = (evt) => {
                const msg = JSON.parse(evt.data);
                this.handleMessage(msg);
                this.emitChange(msg);
            };
        }
    }
    emitChange(msg) {
        this.emit(CHANGE_EVENT, msg);
    }
    addChangeListener(callback) {
        this.on(CHANGE_EVENT, callback);
    }
    removeChangeListener(callback) {
        this.removeListener(CHANGE_EVENT, callback);
    }
    handleMessage(msg) {
        switch (msg.action) {
        case SocketEvents.POSTED:
            handleNewPostEvent(msg);
            break;

        case SocketEvents.POST_EDITED:
            handlePostEditEvent(msg);
            break;

        case SocketEvents.POST_DELETED:
            handlePostDeleteEvent(msg);
            break;

        case SocketEvents.NEW_USER:
            handleNewUserEvent();
            break;

        case SocketEvents.USER_ADDED:
            handleUserAddedEvent(msg);
            break;

        case SocketEvents.USER_REMOVED:
            handleUserRemovedEvent(msg);
            break;

        case SocketEvents.CHANNEL_VIEWED:
            handleChannelViewedEvent(msg);
            break;

        case SocketEvents.PREFERENCE_CHANGED:
            handlePreferenceChangedEvent(msg);
            break;

        default:
        }
    }
    sendMessage(msg) {
        if (conn && conn.readyState === WebSocket.OPEN) {
            conn.send(JSON.stringify(msg));
        } else if (!conn || conn.readyState === WebSocket.Closed) {
            conn = null;
            this.initialize();
        }
    }
}

function handleNewPostEvent(msg) {
    // Store post
    const post = JSON.parse(msg.props.post);
    EventHelpers.emitPostRecievedEvent(post);

    // Update channel state
    if (ChannelStore.getCurrentId() === msg.channel_id) {
        if (window.isActive) {
            AsyncClient.updateLastViewedAt();
        }
    } else if (UserStore.getCurrentId() !== msg.user_id || post.type !== Constants.POST_TYPE_JOIN_LEAVE) {
        AsyncClient.getChannel(msg.channel_id);
    }

    // Send desktop notification
    if ((UserStore.getCurrentId() !== msg.user_id || post.props.from_webhook === 'true') && !Utils.isSystemMessage(post)) {
        const msgProps = msg.props;

        let mentions = [];
        if (msgProps.mentions) {
            mentions = JSON.parse(msg.props.mentions);
        }

        const channel = ChannelStore.get(msg.channel_id);
        const user = UserStore.getCurrentUser();
        const member = ChannelStore.getMember(msg.channel_id);

        let notifyLevel = member && member.notify_props ? member.notify_props.desktop : 'default';
        if (notifyLevel === 'default') {
            notifyLevel = user.notify_props.desktop;
        }

        if (notifyLevel === 'none') {
            return;
        } else if (notifyLevel === 'mention' && mentions.indexOf(user.id) === -1 && channel.type !== 'D') {
            return;
        }

        let username = 'Someone';
        if (post.props.override_username && global.window.mm_config.EnablePostUsernameOverride === 'true') {
            username = post.props.override_username;
        } else if (UserStore.hasProfile(msg.user_id)) {
            username = UserStore.getProfile(msg.user_id).username;
        }

        let title = 'Posted';
        if (channel) {
            title = channel.display_name;
        }

        let notifyText = post.message.replace(/\n+/g, ' ');
        if (notifyText.length > 50) {
            notifyText = notifyText.substring(0, 49) + '...';
        }

        if (notifyText.length === 0) {
            if (msgProps.image) {
                Utils.notifyMe(title, username + ' uploaded an image', channel);
            } else if (msgProps.otherFile) {
                Utils.notifyMe(title, username + ' uploaded a file', channel);
            } else {
                Utils.notifyMe(title, username + ' did something new', channel);
            }
        } else {
            Utils.notifyMe(title, username + ' wrote: ' + notifyText, channel);
        }
        if (!user.notify_props || user.notify_props.desktop_sound === 'true') {
            Utils.ding();
        }
    }
}

function handlePostEditEvent(msg) {
    // Store post
    const post = JSON.parse(msg.props.post);
    PostStore.storePost(post);
    PostStore.emitChange();

    // Update channel state
    if (ChannelStore.getCurrentId() === msg.channel_id) {
        if (window.isActive) {
            AsyncClient.updateLastViewedAt();
        }
    }
}

function handlePostDeleteEvent(msg) {
    const post = JSON.parse(msg.props.post);
    EventHelpers.emitPostDeletedEvent(post);
}

function handleNewUserEvent() {
    AsyncClient.getProfiles();
    AsyncClient.getChannelExtraInfo();
}

function handleUserAddedEvent(msg) {
    if (ChannelStore.getCurrentId() === msg.channel_id) {
        AsyncClient.getChannelExtraInfo();
    }

    if (UserStore.getCurrentId() === msg.user_id) {
        AsyncClient.getChannel(msg.channel_id);
    }
}

function handleUserRemovedEvent(msg) {
    if (UserStore.getCurrentId() === msg.user_id) {
        AsyncClient.getChannels();

        if (msg.props.remover_id !== msg.user_id &&
                msg.channel_id === ChannelStore.getCurrentId() &&
                $('#removed_from_channel').length > 0) {
            var sentState = {};
            sentState.channelName = ChannelStore.getCurrent().display_name;
            sentState.remover = UserStore.getProfile(msg.props.remover_id).username;

            BrowserStore.setItem('channel-removed-state', sentState);
            $('#removed_from_channel').modal('show');
        }
    } else if (ChannelStore.getCurrentId() === msg.channel_id) {
        AsyncClient.getChannelExtraInfo();
    }
}

function handleChannelViewedEvent(msg) {
    // Useful for when multiple devices have the app open to different channels
    if (ChannelStore.getCurrentId() !== msg.channel_id && UserStore.getCurrentId() === msg.user_id) {
        AsyncClient.getChannel(msg.channel_id);
    }
}

function handlePreferenceChangedEvent(msg) {
    const preference = JSON.parse(msg.props.preference);
    EventHelpers.emitPreferenceChangedEvent(preference);
}

var SocketStore = new SocketStoreClass();

/*SocketStore.dispatchToken = AppDispatcher.register((payload) => {
    var action = payload.action;

    switch (action.type) {
    default:
    }
    });*/

export default SocketStore;