glowing-bear/js/websockets.js

var weechat = angular.module('weechat', ['localStorage', 'weechatModels', 'plugins', 'ngSanitize']);

weechat.filter('toArray', function () {
    'use strict';

    return function (obj) {
        if (!(obj instanceof Object)) {
            return obj;
        }

        return Object.keys(obj).map(function (key) {
            return Object.defineProperty(obj[key], '$key', {__proto__: null, value: key});
        });
    }
});

weechat.factory('handlers', ['$rootScope', 'models', 'plugins', function($rootScope, models, plugins) {

    var handleBufferClosing = function(message) {
        var bufferMessage = message['objects'][0]['content'][0];
        var buffer = new models.Buffer(bufferMessage);
        models.closeBuffer(buffer);
    }

    var handleLine = function(line, initial) {
        var message = new models.BufferLine(line);
        // Only react to line if its displayed
        if(message.displayed) {
            var buffer = models.getBuffer(message.buffer);
            message = plugins.PluginManager.contentForMessage(message);
            buffer.addLine(message);

            if (buffer.active) {
                $rootScope.scrollToBottom();
            }

            if (!initial && !buffer.active) {
                if (buffer.notify>1 && _.contains(message.tags, 'notify_message') && !_.contains(message.tags, 'notify_none')) {
                    buffer.unread++;
                    $rootScope.$emit('notificationChanged');
                }

                if(buffer.notify!=0 && message.highlight || _.contains(message.tags, 'notify_private') ) {
                    buffer.notification++;
                    $rootScope.createHighlight(buffer, message);
                    $rootScope.$emit('notificationChanged');
                }
            }
        }
    }

    var handleBufferLineAdded = function(message) {
        message['objects'][0]['content'].forEach(function(l) {
            handleLine(l, false);
        });
    }

    var handleBufferOpened = function(message) {
        var bufferMessage = message['objects'][0]['content'][0];
        var buffer = new models.Buffer(bufferMessage);
        models.addBuffer(buffer);
    }

    var handleBufferTitleChanged = function(message) {
        var obj = message['objects'][0]['content'][0];
        var buffer = obj['pointers'][0];
        var old = models.getBuffer(buffer);
        old.fullName = obj['full_name'];
        old.title = obj['title'];
        old.number = obj['number'];
    }

    var handleBufferRenamed = function(message) {
        var obj = message['objects'][0]['content'][0];
        var buffer = obj['pointers'][0];
        var old = models.getBuffer(buffer);
        old.fullName = obj['full_name'];
        old.shortName = obj['short_name'];
    }

    /*
     * Handle answers to (lineinfo) messages
     *
     * (lineinfo) messages are specified by this client. It is request after bufinfo completes
     */
    var handleLineInfo = function(message) {
        var lines = message['objects'][0]['content'].reverse();
        lines.forEach(function(l) {
            handleLine(l, true);
        });
    }

    /*
     * Handle answers to hotlist request
     */
    var handleHotlistInfo = function(message) {
        if (message.objects.length == 0) {
            return;
        }
        var hotlist = message['objects'][0]['content'];
        hotlist.forEach(function(l) {
            var buffer = models.getBuffer(l.buffer);
            // 1 is message
            buffer.unread += l.count[1];
            // 2 is ?
            buffer.unread += l.count[2];
            // 3 is highlight
            buffer.notification += l.count[3];
            /* Since there is unread messages, we can guess
            * what the last read line is and update it accordingly
            */
            var unreadSum = _.reduce(l.count, function(memo, num){ return memo + num; }, 0);
            buffer.lastSeen = buffer.lines.length - 1 - unreadSum;
        });
    }

    /*
     * Handle nicklist event
     */
    var handleNicklist = function(message) {
        var nicklist = message['objects'][0]['content'];
        var group = 'root';
        nicklist.forEach(function(n) {
            var buffer = models.getBuffer(n.pointers[0]);
            if(n.group == 1) {
                var g = new models.NickGroup(n);
                group = g.name;
                buffer.nicklist[group] = g;
            }else{
                var nick = new models.Nick(n);
                buffer.addNick(group, nick);
            }
        });
    }
    /*
     * Handle nicklist diff event
     */
    var handleNicklistDiff = function(message) {
        var nicklist = message['objects'][0]['content'];
        var group;
        nicklist.forEach(function(n) {
            var buffer = models.getBuffer(n.pointers[0]);
            var d = n['_diff'];
            if(n.group == 1) {
                group = n.name;
                if(group==undefined) {
                    var g = new models.NickGroup(n);
                    buffer.nicklist[group] = g;
                    group = g.name;
                }
            }
            else {
                var nick = new models.Nick(n);
                if(d == 43) { // +
                    buffer.addNick(group, nick);
                }else if (d == 45) { // -
                    buffer.delNick(group, nick);
                }else if (d == 42) { // *
                    buffer.updateNick(group, nick);
                }
            }
        });
    }

    var handleEvent = function(event) {

        if (_.has(eventHandlers, event['id'])) {
            eventHandlers[event['id']](event);
        }

    }

    var eventHandlers = {
        _buffer_closing: handleBufferClosing,
        _buffer_line_added: handleBufferLineAdded,
        _buffer_opened: handleBufferOpened,
        _buffer_title_changed: handleBufferTitleChanged,
        _buffer_renamed: handleBufferRenamed,
        _nicklist: handleNicklist,
        _nicklist_diff: handleNicklistDiff
    }

    return {
        handleEvent: handleEvent,
        handleLineInfo: handleLineInfo,
        handleHotlistInfo: handleHotlistInfo,
        handleNicklist: handleNicklist
    }

}]);

weechat.factory('connection', ['$q', '$rootScope', '$log', '$store', 'handlers', 'models', function($q, $rootScope, $log, storage, handlers, models) {
    protocol = new weeChat.Protocol();
    var websocket = null;

    var callbacks = {}
    var currentCallBackId = 0;

    /*
     * Returns the current callback id
     */
    var getCurrentCallBackId = function() {
        
        currentCallBackId += 1;

        if (currentCallBackId > 1000) {
            currentCallBackId = 0;
        }

        return currentCallBackId;
    }

    /*
     * Create a callback, adds it to the callback list
     * and return it.
     */
    var createCallback = function() {
        var defer = $q.defer();
        var cbId = getCurrentCallBackId();

        callbacks[cbId] = {
            time: new Date,
            cb: defer,
        }

        defer.id = cbId

        return defer;
    }

    /*
     * Fails every currently subscribed callback for the
     * given reason
     *
     * @param reason reason for failure
     */
    failCallbacks = function(reason) {
        for(i in callbacks) {
            callbacks[i].cb.reject(reason);
        }

    }

    /* Send a message to the websocket and returns a promise.
     * See: http://docs.angularjs.org/api/ng.$q
     *
     * @param message message to send
     * @returns a promise
     */
    var send = function(message) {
        message.replace(/[\r\n]+$/g, "").split("\n");
        var cb = createCallback(message);
        websocket.send("(" + cb.id + ") " + message);
        return cb.promise;
    }

    /*
     * Send all messages to the websocket and returns a promise that is resolved
     * when all message are resolved.
     *
     * @param messages list of messages
     * @returns a promise
     */
    var sendAll = function(messages) {
        var promises = [];
        for(i in messages) {
            var promise = send(messages[i]);
            promises.push(promise);
        };
        return $q.all(promises);
    };

    // Takes care of the connection and websocket hooks
    var connect = function (host, port, passwd, ssl) {
        var proto = ssl ? 'wss':'ws';
        websocket = new WebSocket(proto+"://" + host + ':' + port + "/weechat");
        websocket.binaryType = "arraybuffer"

        websocket.onopen = function (evt) {

            $log.info("Connected to relay");

            // First message must be an init request
            // with the password
            send(weeChat.Protocol.formatInit({
                    password: passwd,
                    compression: 'off'
            }));

            /*
             * Send all the initialization commands
             * and resolve them as a single promise.
             */
            sendAll([
                weeChat.Protocol.formatInfo({
                    name: 'version',
                }),

                weeChat.Protocol.formatHdata({
                    path: 'buffer:gui_buffers(*)',
                    keys: ['local_variables,notify,number,full_name,short_name,title']                        
                }),

                weeChat.Protocol.formatHdata({
                    path: "buffer:gui_buffers(*)/own_lines/last_line(-"+storage.get('lines')+")/data",
                    keys: []
                }),

                weeChat.Protocol.formatHdata({
                    path: "hotlist:gui_hotlist(*)",
                    keys: []
                }),

                weeChat.Protocol.formatNicklist({
                })

            ]).then(
                function(messages) {
                    var version = messages[0];
                    var bufinfo = messages[1];
                    var lineinfo = messages[2];
                    var hotlist = messages[3];
                    var nicklist = messages[4];

                    $log.info("Parsing bufinfo");
                    var bufferInfos = bufinfo['objects'][0]['content'];
                    // buffers objects
                    for (var i = 0; i < bufferInfos.length ; i++) {
                        var buffer = new models.Buffer(bufferInfos[i]);
                        models.addBuffer(buffer);
                        // Switch to first buffer on startup
                        if (i == 0) {
                            models.setActiveBuffer(buffer.id);
                        }
                    }

                    handlers.handleLineInfo(lineinfo);
                    handlers.handleHotlistInfo(hotlist)
                    handlers.handleNicklist(nicklist)
                    send(weeChat.Protocol.formatSync({}));
                    $log.info("Synced");

                    // here we are really connected !
                    $rootScope.connected = true;
                },
                function(error) {
                    // If the first command fails, it means that we do not have the
                    // proper password
                    $rootScope.passwordError = true;
                }
            );

        }

        websocket.onclose = function (evt) {
            $log.info("Disconnected from relay");
            $rootScope.connected = false;
            failCallbacks('disconnection');
            $rootScope.$apply();
        }

        websocket.onmessage = function (evt) {
	    message = protocol.parse(evt.data)
            if (_.has(callbacks, message['id'])) {
                var promise = callbacks[message['id']];
                promise.cb.resolve(message);
                delete(callbacks[message['id']]);
            } else {
                handlers.handleEvent(message);
            }
            $rootScope.commands.push("RECV: " + evt.data + " TYPE:" + evt.type) ;
            $rootScope.$apply();
        }

        websocket.onerror = function (evt) {
            // on error it means the connection problem
            // come from the relay not from the password.

            if (evt.type == "error" && websocket.readyState != 1) {
                failCallbacks('error');
                $rootScope.errorMessage = true;
            }
            $log.error("Relay error " + evt.data);
        }

        this.websocket = websocket;
    }

    var disconnect = function() {
        /* TODO: Send protocol disconnect */
        this.websocket.close();
    }

    var sendMessage = function(message) {
        send(weeChat.Protocol.formatInput({
            buffer: models.getActiveBuffer()['fullName'],
            data: message
        }));
    }

    var sendCoreCommand = function(command) {
        send(weeChat.Protocol.formatInput({
            buffer: 'core.weechat',
            data: command
        }));
    }


    return {
        send: send,
        connect: connect,
        disconnect: disconnect,
        sendMessage: sendMessage,
        sendCoreCommand: sendCoreCommand
    }
}]);

weechat.controller('WeechatCtrl', ['$rootScope', '$scope', '$store', '$timeout', '$log', 'models', 'connection', function ($rootScope, $scope, $store, $timeout, $log, models, connection, testService) {
    if(window.Notification) {
        // Request notification permission
        Notification.requestPermission(function (status) {
            $log.info('Notification permission status:',status);
            if (Notification.permission !== status) {
                Notification.permission = status;
            }
        });

    }
    if(window.webkitNotifications != undefined) {
        if (window.webkitNotifications.checkPermission() == 0) { // 0 is PERMISSION_ALLOWED
            $log.info('Notification permission status:', window.webkitNotifications.checkPermission() == 0);
            window.webkitNotifications.requestPermission();
        }
    }
    // Check for firefox & app installed
    if(navigator.mozApps != undefined) {
        navigator.mozApps.getSelf().onsuccess = function _onAppReady(evt) {
            var app = evt.target.result;
            if(app) {
                $scope.isinstalled = true;
            }else {
                $scope.isinstalled = false;
            }
        } 
    }else {
        $scope.isinstalled = false;
    }

    $rootScope.$on('activeBufferChanged', function() {
        $rootScope.scrollToBottom();
        document.getElementById('sendMessage').focus();
        var ab = models.getActiveBuffer();
        $rootScope.pageTitle = ab.shortName + ' | ' + ab.title;

        // If user wants to sync hotlist with weechat
        // we will send a /buffer bufferName command every time
        // the user switches a buffer. This will ensure that notifications
        // are cleared in the buffer the user switches to
        if($scope.hotlistsync && ab.fullName) { 
            connection.sendCoreCommand('/buffer ' + ab.fullName);
        }

        // Clear search term on buffer change
        $scope.search = '';
        
        // Check if we should show nicklist or not
        $scope.showNicklist = $scope.updateShowNicklist();
    });
    $rootScope.$on('notificationChanged', function() {
        var notifications = _.reduce(models.model.buffers, function(memo, num) { return (memo||0) + num.notification;});
        if (notifications > 0 ) {
            $scope.favico = new Favico({
                animation:'none'
            });
            $scope.favico.badge(notifications);
        }else {
            var unread = _.reduce(models.model.buffers, function(memo, num) { return (memo||0) + num.unread;});
            $scope.favico = new Favico({
                animation:'none',
                bgColor : '#5CB85C',
                textColor : '#ff0',
            });
            $scope.favico.badge(unread);
        }
    });

    $scope.buffers = models.model.buffers;
    $scope.activeBuffer = models.getActiveBuffer

    $rootScope.commands = []

    $rootScope.models = models;

    $rootScope.buffer = []

    $rootScope.iterCandidate = null;

    $store.bind($scope, "host", "localhost");
    $store.bind($scope, "port", "9001");
    $store.bind($scope, "proto", "weechat");
    $store.bind($scope, "ssl", false);
    $store.bind($scope, "lines", "40");
    $store.bind($scope, "savepassword", false);
    if($scope.savepassword) {
        $store.bind($scope, "password", "");
    }

    // Save setting for displaying only buffers with unread messages
    $store.bind($scope, "onlyUnread", false);
    // Save setting for not showing timestamp
    $store.bind($scope, "notimestamp", false);
    // Save setting for syncing hotlist
    $store.bind($scope, "hotlistsync", true);
    // Save setting for displaying nicklist
    $store.bind($scope, "nonicklist", false); 

    $scope.setActiveBuffer = function(key) {
        models.setActiveBuffer(key);
    };

    $rootScope.scrollToBottom = function() {
        // FIXME doesn't work if the settimeout runs without a short delay
        var scroll = function() {
            var readmarker = document.getElementById('readmarker');
            if(readmarker) {
                readmarker.scrollIntoView();
            }else{
                var bl = document.getElementById('bufferlines');
                var sTop = bl.scrollTop;
                var sVal = bl.scrollHeight - bl.clientHeight;
                if(sTop < sVal) {
                    bl.scrollTop = sVal;
                }
            }
        }
        // Here be scrolling dragons
        $timeout(scroll);
        $timeout(scroll, 100);
        $timeout(scroll, 300);
        $timeout(scroll, 500);
    }

    $scope.sendMessage = function() {
        connection.sendMessage($scope.command);
        $scope.command = "";
    };

    $scope.connect = function() {
        connection.connect($scope.host, $scope.port, $scope.password, $scope.ssl);
    }
    $scope.disconnect = function() {
        connection.disconnect();
    }
    $scope.install = function() {
        if(navigator.mozApps != undefined) {
            var request = navigator.mozApps.install('http://torhve.github.io/glowing-bear/manifest.webapp');
            request.onsuccess = function () {
                $scope.isinstalled = true;
                // Save the App object that is returned
                var appRecord = this.result;
                // Start the app.
                appRecord.launch();
                alert('Installation successful!');
            };
            request.onerror = function () {
                // Display the error information from the DOMError object
                alert('Install failed, error: ' + this.error.name);
            };
        }else{
            alert('Sorry. Only supported in Firefox v26+');
        }
    }


    /* Function gets called from bufferLineAdded code if user should be notified */
    $rootScope.createHighlight = function(buffer, message) {
        var messages = "";
        message.content.forEach(function(part) {
            if (part.text != undefined)
                messages += part.text + " ";
        });

        var title = buffer.fullName;
        var content = messages;

        var timeout = 15*1000;
        $log.info('Displaying notification:buffer:',buffer,',message:',message,',with timeout:',timeout);
        var notification = new Notification(title, {body:content, icon:'img/favicon.png'});
        // Cancel notification automatically
        notification.onshow = function() {
            setTimeout(function() { notification.close() }, timeout);
        }
    };

    $scope.hasUnread = function(buffer) {
        // if search is set, return every buffer 
        if($scope.search && $scope.search != "") {
            return true;
        }
        if($scope.onlyUnread) {
            // Always show current buffer in list
            if (models.getActiveBuffer() == buffer) {
                return true;
            }
            return buffer.unread > 0 || buffer.notification > 0;
        }
        return true;
    };

    // Watch model and update show setting when it changes
    $scope.$watch('nonicklist', function() {
        $scope.showNicklist = $scope.updateShowNicklist();
    });
    $scope.showNicklist = false;
    // Utility function that template can use to check if nicklist should 
    // be displayed for current buffer or not
    // is called on buffer switch
    $scope.updateShowNicklist = function() {
        var ab = models.getActiveBuffer();
        if(!ab) {
            return false;
        }
        // Check if option no nicklist is set
        if($scope.nonicklist) {
            return false;
        }
        // Use flat nicklist to check if empty
        if(ab.flatNicklist().length === 0) { 
            return false;
        }
        return true;
    }

    $rootScope.switchToActivityBuffer = function() {
        // Find next buffer with activity and switch to it
        for(i in $scope.buffers) {
            var buffer = $scope.buffers[i];
            if(buffer.notification > 0) {
                $scope.setActiveBuffer(buffer.id);
                break;
            }else if(buffer.unread > 0) {
                $scope.setActiveBuffer(buffer.id);
                break;
            }
        }
    }

    $rootScope.completeNick = function() {
        // input DOM node
        var inputNode = document.getElementById('sendMessage');

        // get current input
        var inputText = inputNode.value;

        // get current caret position
        var caretPos = inputNode.selectionStart;

        // create flat array of nicks
        var activeBuffer = models.getActiveBuffer();

        // complete nick
        var nickComp = IrcUtils.completeNick(inputText, caretPos,
            $rootScope.iterCandidate, activeBuffer.flatNicklist(), ':');

        // remember iteration candidate
        $rootScope.iterCandidate = nickComp.iterCandidate;

        // update current input
        inputNode.value = nickComp.text;

        // update current caret position
        inputNode.focus();
        inputNode.setSelectionRange(nickComp.caretPos, nickComp.caretPos);
    }

    $scope.handleKeyPress = function($event) {
        // don't do anything if not connected
        if (!$rootScope.connected) {
            return true;
        }
        
        // Support different browser quirks
        var code = $event.keyCode ? $event.keyCode : $event.charCode;

        // any other key than Tab resets nick completion iteration
        var tmpIterCandidate = $rootScope.iterCandidate;
        $rootScope.iterCandidate = null;

        // Left Alt+[0-9] -> jump to buffer
        if ($event.altKey && !$event.ctrlKey && (code > 47 && code < 58)) {
            if (code == 48) {
                code = 58;
            }

            var bufferNumber = code - 48;
            var activeBuffer = models.getBufferByIndex(bufferNumber);
            if (activeBuffer) {
                models.setActiveBuffer(activeBuffer.id);
                $event.preventDefault();
            }
        }

        // Tab -> nick completion
        if (code == 9 && !$event.altKey && !$event.ctrlKey) {
            $event.preventDefault();
            $rootScope.iterCandidate = tmpIterCandidate;
            $rootScope.completeNick();
            return true;
        }

        // Alt+A -> switch to buffer with activity
        if ($event.altKey && (code == 97 || code == 65)) {
            $event.preventDefault();
            $rootScope.switchToActivityBuffer();
            return true;
        }

        // Alt+L -> focus on input bar
        if ($event.altKey && (code == 76 || code == 108)) {
            $event.preventDefault();
            var inputNode = document.getElementById('sendMessage');
            inputNode.focus();
            inputNode.setSelectionRange(inputNode.value.length, inputNode.value.length);
            return true;
        }

        // Escape -> disconnect
        if (code == 27) {
            $event.preventDefault();
            connection.disconnect();
            return true;
        }

        // Ctrl+G -> focus on buffer filter input
        if ($event.ctrlKey && (code == 103 || code == 71)) {
            $event.preventDefault();
            document.getElementById('bufferFilter').focus();
            return true;
        }
    };
    $scope.handleSearchBoxKey = function($event) {
        // Support different browser quirks
        var code = $event.keyCode ? $event.keyCode : $event.charCode;
        // Handle escape
        if(code == 27) {
            $event.preventDefault();
            $scope.search = '';
        } // Handle enter
        else if (code == 13) {
            $event.preventDefault();
            // TODO Switch to first matching buffer and reset query
            $scope.search = '';
        }
    }

}]
);