Angularjs - Factory for Socket.io

angular-socket-factory.js

    function socket($rootScope) {
        var socket = io.connect('https://HOST:PORT');

        return {
            emit: function (eventName, data, callback) {
                socket.emit(eventName, data, function () {
                    var args = arguments;
                    $rootScope.$apply(function () {
                        if (callback) {
                            callback.apply(socket, args);
                        }
                    });
                });
            },
            
            on: function (eventName, callback) {
                socket.on(eventName, function () {
                    var args = arguments;
                    $rootScope.$apply(function () {
                        callback.apply(socket, args);
                    });
                });
            },

            disconnect: function () {
                socket.disconnect();
            }
        }
    }