Socketio wrapper directive with event handler cleaner / unlistener

socket.js

App.factory('socket', ['$rootScope', 
    function($rootScope) {
        var socket = io.connect('127.0.0.1:8080'); 
        return {
            on: function(eventName, callback) { 
                //set up callback wrapper temp var so we can return it later
                var temp = function() { 
                	var args = arguments;   
                	$rootScope.$apply(function() {
                		callback.apply(socket, args);
                	});
                };
                socket.on(eventName, temp);
                //return wrapper
                return temp; 
            },
            off: function(eventName, callback) { 
                socket.removeListener(eventName, callback);
            },
            emit: function (eventName, data, callback) {
                socket.emit(eventName, data, function () {
                	var args = arguments;
                	$rootScope.$apply(function () {
                        if (callback) {
                            callback.apply(socket, args);
                        }
                	});
                })
            },
            socket: socket.socket
        };
    }
]);