'use strict'; var utils = require('../utils/event') , urlUtils = require('../utils/url') , inherits = require('inherits') , EventEmitter = require('events').EventEmitter , WebsocketDriver = require('./driver/websocket') ; var debug = function() {}; if (process.env.NODE_ENV !== 'production') { debug = require('debug')('sockjs-client:websocket'); } function WebSocketTransport(transUrl, ignore, options) { if (!WebSocketTransport.enabled()) { throw new Error('Transport created when disabled'); } EventEmitter.call(this); debug('constructor', transUrl); var self = this; var url = urlUtils.addPath(transUrl, '/websocket'); if (url.slice(0, 5) === 'https') { url = 'wss' + url.slice(5); } else { url = 'ws' + url.slice(4); } this.url = url; this.ws = new WebsocketDriver(this.url, [], options); this.ws.onmessage = function(e) { debug('message event', e.data); self.emit('message', e.data); }; // Firefox has an interesting bug. If a websocket connection is // created after onunload, it stays alive even when user // navigates away from the page. In such situation let's lie - // let's not open the ws connection at all. See: // https://github.com/sockjs/sockjs-client/issues/28 // https://bugzilla.mozilla.org/show_bug.cgi?id=696085 this.unloadRef = utils.unloadAdd(function() { debug('unload'); self.ws.close(); }); this.ws.onclose = function(e) { debug('close event', e.code, e.reason); self.emit('close', e.code, e.reason); self._cleanup(); }; this.ws.onerror = function(e) { debug('error event', e); self.emit('close', 1006, 'WebSocket connection broken'); self._cleanup(); }; } inherits(WebSocketTransport, EventEmitter); WebSocketTransport.prototype.send = function(data) { var msg = '[' + data + ']'; debug('send', msg); this.ws.send(msg); }; WebSocketTransport.prototype.close = function() { debug('close'); var ws = this.ws; this._cleanup(); if (ws) { ws.close(); } }; WebSocketTransport.prototype._cleanup = function() { debug('_cleanup'); var ws = this.ws; if (ws) { ws.onmessage = ws.onclose = ws.onerror = null; } utils.unloadDel(this.unloadRef); this.unloadRef = this.ws = null; this.removeAllListeners(); }; WebSocketTransport.enabled = function() { debug('enabled'); return !!WebsocketDriver; }; WebSocketTransport.transportName = 'websocket'; // In theory, ws should require 1 round trip. But in chrome, this is // not very stable over SSL. Most likely a ws connection requires a // separate SSL connection, in which case 2 round trips are an // absolute minumum. WebSocketTransport.roundTrips = 2; module.exports = WebSocketTransport;