(function (global, factory) { if (typeof define === "function" && define.amd) { define(['module', 'exports', 'angular', 'ws'], factory); } else if (typeof exports !== "undefined") { factory(module, exports, require('angular'), require('ws')); } else { var mod = { exports: {} }; factory(mod, mod.exports, global.angular, global.ws); global.angularWebsocket = mod.exports; } })(this, function (module, exports, _angular, ws) { 'use strict'; Object.defineProperty(exports, "__esModule", { value: true }); var _angular2 = _interopRequireDefault(_angular); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } var _typeof = typeof Symbol === "function" && typeof Symbol.iterator === "symbol" ? function (obj) { return typeof obj; } : function (obj) { return obj && typeof Symbol === "function" && obj.constructor === Symbol ? "symbol" : typeof obj; }; var Socket; if (typeof window === 'undefined') { try { Socket = ws.Client || ws.client || ws; } catch (e) {} } // Browser Socket = Socket || window.WebSocket || window.MozWebSocket; var noop = _angular2.default.noop; var objectFreeze = Object.freeze ? Object.freeze : noop; var objectDefineProperty = Object.defineProperty; var isString = _angular2.default.isString; var isFunction = _angular2.default.isFunction; var isDefined = _angular2.default.isDefined; var isObject = _angular2.default.isObject; var isArray = _angular2.default.isArray; var forEach = _angular2.default.forEach; var arraySlice = Array.prototype.slice; // ie8 wat if (!Array.prototype.indexOf) { Array.prototype.indexOf = function (elt /*, from*/) { var len = this.length >>> 0; var from = Number(arguments[1]) || 0; from = from < 0 ? Math.ceil(from) : Math.floor(from); if (from < 0) { from += len; } for (; from < len; from++) { if (from in this && this[from] === elt) { return from; } } return -1; }; } // $WebSocketProvider.$inject = ['$rootScope', '$q', '$timeout', '$websocketBackend']; function $WebSocketProvider($rootScope, $q, $timeout, $websocketBackend) { function $WebSocket(url, protocols, options) { if (!options && isObject(protocols) && !isArray(protocols)) { options = protocols; protocols = undefined; } this.protocols = protocols; this.url = url || 'Missing URL'; this.ssl = /(wss)/i.test(this.url); // this.binaryType = ''; // this.extensions = ''; // this.bufferedAmount = 0; // this.trasnmitting = false; // this.buffer = []; // TODO: refactor options to use isDefined this.scope = options && options.scope || $rootScope; this.rootScopeFailover = options && options.rootScopeFailover && true; this.useApplyAsync = options && options.useApplyAsync || false; this.initialTimeout = options && options.initialTimeout || 500; // 500ms this.maxTimeout = options && options.maxTimeout || 5 * 60 * 1000; // 5 minutes this.reconnectIfNotNormalClose = options && options.reconnectIfNotNormalClose || false; this.binaryType = options && options.binaryType || 'blob'; this._reconnectAttempts = 0; this.sendQueue = []; this.onOpenCallbacks = []; this.onMessageCallbacks = []; this.onErrorCallbacks = []; this.onCloseCallbacks = []; objectFreeze(this._readyStateConstants); if (url) { this._connect(); } else { this._setInternalState(0); } } $WebSocket.prototype._readyStateConstants = { 'CONNECTING': 0, 'OPEN': 1, 'CLOSING': 2, 'CLOSED': 3, 'RECONNECT_ABORTED': 4 }; $WebSocket.prototype._normalCloseCode = 1000; $WebSocket.prototype._reconnectableStatusCodes = [4000]; $WebSocket.prototype.safeDigest = function safeDigest(autoApply) { if (autoApply && !this.scope.$$phase) { this.scope.$digest(); } }; $WebSocket.prototype.bindToScope = function bindToScope(scope) { var self = this; if (scope) { this.scope = scope; if (this.rootScopeFailover) { this.scope.$on('$destroy', function () { self.scope = $rootScope; }); } } return self; }; $WebSocket.prototype._connect = function _connect(force) { if (force || !this.socket || this.socket.readyState !== this._readyStateConstants.OPEN) { this.socket = $websocketBackend.create(this.url, this.protocols); this.socket.onmessage = _angular2.default.bind(this, this._onMessageHandler); this.socket.onopen = _angular2.default.bind(this, this._onOpenHandler); this.socket.onerror = _angular2.default.bind(this, this._onErrorHandler); this.socket.onclose = _angular2.default.bind(this, this._onCloseHandler); this.socket.binaryType = this.binaryType; } }; $WebSocket.prototype.fireQueue = function fireQueue() { while (this.sendQueue.length && this.socket.readyState === this._readyStateConstants.OPEN) { var data = this.sendQueue.shift(); this.socket.send(isString(data.message) || this.binaryType != 'blob' ? data.message : JSON.stringify(data.message)); data.deferred.resolve(); } }; $WebSocket.prototype.notifyOpenCallbacks = function notifyOpenCallbacks(event) { for (var i = 0; i < this.onOpenCallbacks.length; i++) { this.onOpenCallbacks[i].call(this, event); } }; $WebSocket.prototype.notifyCloseCallbacks = function notifyCloseCallbacks(event) { for (var i = 0; i < this.onCloseCallbacks.length; i++) { this.onCloseCallbacks[i].call(this, event); } }; $WebSocket.prototype.notifyErrorCallbacks = function notifyErrorCallbacks(event) { for (var i = 0; i < this.onErrorCallbacks.length; i++) { this.onErrorCallbacks[i].call(this, event); } }; $WebSocket.prototype.onOpen = function onOpen(cb) { this.onOpenCallbacks.push(cb); return this; }; $WebSocket.prototype.onClose = function onClose(cb) { this.onCloseCallbacks.push(cb); return this; }; $WebSocket.prototype.onError = function onError(cb) { this.onErrorCallbacks.push(cb); return this; }; $WebSocket.prototype.onMessage = function onMessage(callback, options) { if (!isFunction(callback)) { throw new Error('Callback must be a function'); } if (options && isDefined(options.filter) && !isString(options.filter) && !(options.filter instanceof RegExp)) { throw new Error('Pattern must be a string or regular expression'); } this.onMessageCallbacks.push({ fn: callback, pattern: options ? options.filter : undefined, autoApply: options ? options.autoApply : true }); return this; }; $WebSocket.prototype._onOpenHandler = function _onOpenHandler(event) { this._reconnectAttempts = 0; this.notifyOpenCallbacks(event); this.fireQueue(); }; $WebSocket.prototype._onCloseHandler = function _onCloseHandler(event) { var self = this; if (self.useApplyAsync) { self.scope.$applyAsync(function () { self.notifyCloseCallbacks(event); }); } else { self.notifyCloseCallbacks(event); self.safeDigest(true); } if (this.reconnectIfNotNormalClose && event.code !== this._normalCloseCode || this._reconnectableStatusCodes.indexOf(event.code) > -1) { this.reconnect(); } }; $WebSocket.prototype._onErrorHandler = function _onErrorHandler(event) { var self = this; if (self.useApplyAsync) { self.scope.$applyAsync(function () { self.notifyErrorCallbacks(event); }); } else { self.notifyErrorCallbacks(event); self.safeDigest(true); } }; $WebSocket.prototype._onMessageHandler = function _onMessageHandler(message) { var pattern; var self = this; var currentCallback; for (var i = 0; i < self.onMessageCallbacks.length; i++) { currentCallback = self.onMessageCallbacks[i]; pattern = currentCallback.pattern; if (pattern) { if (isString(pattern) && message.data === pattern) { applyAsyncOrDigest(currentCallback.fn, currentCallback.autoApply, message); } else if (pattern instanceof RegExp && pattern.exec(message.data)) { applyAsyncOrDigest(currentCallback.fn, currentCallback.autoApply, message); } } else { applyAsyncOrDigest(currentCallback.fn, currentCallback.autoApply, message); } } function applyAsyncOrDigest(callback, autoApply, args) { args = arraySlice.call(arguments, 2); if (self.useApplyAsync) { self.scope.$applyAsync(function () { callback.apply(self, args); }); } else { callback.apply(self, args); self.safeDigest(autoApply); } } }; $WebSocket.prototype.close = function close(force) { if (force || !this.socket.bufferedAmount) { this.socket.close(); } return this; }; $WebSocket.prototype.send = function send(data) { var deferred = $q.defer(); var self = this; var promise = cancelableify(deferred.promise); if (self.readyState === self._readyStateConstants.RECONNECT_ABORTED) { deferred.reject('Socket connection has been closed'); } else { self.sendQueue.push({ message: data, deferred: deferred }); self.fireQueue(); } // Credit goes to @btford function cancelableify(promise) { promise.cancel = cancel; var then = promise.then; promise.then = function () { var newPromise = then.apply(this, arguments); return cancelableify(newPromise); }; return promise; } function cancel(reason) { self.sendQueue.splice(self.sendQueue.indexOf(data), 1); deferred.reject(reason); return self; } if ($websocketBackend.isMocked && $websocketBackend.isMocked() && $websocketBackend.isConnected(this.url)) { this._onMessageHandler($websocketBackend.mockSend()); } return promise; }; $WebSocket.prototype.reconnect = function reconnect() { this.close(); var backoffDelay = this._getBackoffDelay(++this._reconnectAttempts); var backoffDelaySeconds = backoffDelay / 1000; console.log('Reconnecting in ' + backoffDelaySeconds + ' seconds'); $timeout(_angular2.default.bind(this, this._connect), backoffDelay); return this; }; // Exponential Backoff Formula by Prof. Douglas Thain // http://dthain.blogspot.co.uk/2009/02/exponential-backoff-in-distributed.html $WebSocket.prototype._getBackoffDelay = function _getBackoffDelay(attempt) { var R = Math.random() + 1; var T = this.initialTimeout; var F = 2; var N = attempt; var M = this.maxTimeout; return Math.floor(Math.min(R * T * Math.pow(F, N), M)); }; $WebSocket.prototype._setInternalState = function _setInternalState(state) { if (Math.floor(state) !== state || state < 0 || state > 4) { throw new Error('state must be an integer between 0 and 4, got: ' + state); } // ie8 wat if (!objectDefineProperty) { this.readyState = state || this.socket.readyState; } this._internalConnectionState = state; forEach(this.sendQueue, function (pending) { pending.deferred.reject('Message cancelled due to closed socket connection'); }); }; // Read only .readyState if (objectDefineProperty) { objectDefineProperty($WebSocket.prototype, 'readyState', { get: function get() { return this._internalConnectionState || this.socket.readyState; }, set: function set() { throw new Error('The readyState property is read-only'); } }); } return function (url, protocols, options) { return new $WebSocket(url, protocols, options); }; } // $WebSocketBackendProvider.$inject = ['$log']; function $WebSocketBackendProvider($log) { this.create = function create(url, protocols) { var match = /wss?:\/\//.exec(url); if (!match) { throw new Error('Invalid url provided'); } if (protocols) { return new Socket(url, protocols); } return new Socket(url); }; this.createWebSocketBackend = function createWebSocketBackend(url, protocols) { $log.warn('Deprecated: Please use .create(url, protocols)'); return this.create(url, protocols); }; } _angular2.default.module('ngWebSocket', []).factory('$websocket', ['$rootScope', '$q', '$timeout', '$websocketBackend', $WebSocketProvider]).factory('WebSocket', ['$rootScope', '$q', '$timeout', 'WebsocketBackend', $WebSocketProvider]).service('$websocketBackend', ['$log', $WebSocketBackendProvider]).service('WebSocketBackend', ['$log', $WebSocketBackendProvider]); _angular2.default.module('angular-websocket', ['ngWebSocket']); exports.default = _angular2.default.module('ngWebSocket'); module.exports = exports['default']; });