123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245 |
- /**
- * Module dependencies.
- */
- var Transport = require('../transport');
- var parseqs = require('parseqs');
- var parser = require('engine.io-parser');
- var inherit = require('component-inherit');
- var yeast = require('yeast');
- var debug = require('debug')('engine.io-client:polling');
- /**
- * Module exports.
- */
- module.exports = Polling;
- /**
- * Is XHR2 supported?
- */
- var hasXHR2 = (function () {
- var XMLHttpRequest = require('xmlhttprequest-ssl');
- var xhr = new XMLHttpRequest({ xdomain: false });
- return null != xhr.responseType;
- })();
- /**
- * Polling interface.
- *
- * @param {Object} opts
- * @api private
- */
- function Polling (opts) {
- var forceBase64 = (opts && opts.forceBase64);
- if (!hasXHR2 || forceBase64) {
- this.supportsBinary = false;
- }
- Transport.call(this, opts);
- }
- /**
- * Inherits from Transport.
- */
- inherit(Polling, Transport);
- /**
- * Transport name.
- */
- Polling.prototype.name = 'polling';
- /**
- * Opens the socket (triggers polling). We write a PING message to determine
- * when the transport is open.
- *
- * @api private
- */
- Polling.prototype.doOpen = function () {
- this.poll();
- };
- /**
- * Pauses polling.
- *
- * @param {Function} callback upon buffers are flushed and transport is paused
- * @api private
- */
- Polling.prototype.pause = function (onPause) {
- var self = this;
- this.readyState = 'pausing';
- function pause () {
- debug('paused');
- self.readyState = 'paused';
- onPause();
- }
- if (this.polling || !this.writable) {
- var total = 0;
- if (this.polling) {
- debug('we are currently polling - waiting to pause');
- total++;
- this.once('pollComplete', function () {
- debug('pre-pause polling complete');
- --total || pause();
- });
- }
- if (!this.writable) {
- debug('we are currently writing - waiting to pause');
- total++;
- this.once('drain', function () {
- debug('pre-pause writing complete');
- --total || pause();
- });
- }
- } else {
- pause();
- }
- };
- /**
- * Starts polling cycle.
- *
- * @api public
- */
- Polling.prototype.poll = function () {
- debug('polling');
- this.polling = true;
- this.doPoll();
- this.emit('poll');
- };
- /**
- * Overloads onData to detect payloads.
- *
- * @api private
- */
- Polling.prototype.onData = function (data) {
- var self = this;
- debug('polling got data %s', data);
- var callback = function (packet, index, total) {
- // if its the first message we consider the transport open
- if ('opening' === self.readyState) {
- self.onOpen();
- }
- // if its a close packet, we close the ongoing requests
- if ('close' === packet.type) {
- self.onClose();
- return false;
- }
- // otherwise bypass onData and handle the message
- self.onPacket(packet);
- };
- // decode payload
- parser.decodePayload(data, this.socket.binaryType, callback);
- // if an event did not trigger closing
- if ('closed' !== this.readyState) {
- // if we got data we're not polling
- this.polling = false;
- this.emit('pollComplete');
- if ('open' === this.readyState) {
- this.poll();
- } else {
- debug('ignoring poll - transport state "%s"', this.readyState);
- }
- }
- };
- /**
- * For polling, send a close packet.
- *
- * @api private
- */
- Polling.prototype.doClose = function () {
- var self = this;
- function close () {
- debug('writing close packet');
- self.write([{ type: 'close' }]);
- }
- if ('open' === this.readyState) {
- debug('transport open - closing');
- close();
- } else {
- // in case we're trying to close while
- // handshaking is in progress (GH-164)
- debug('transport not open - deferring close');
- this.once('open', close);
- }
- };
- /**
- * Writes a packets payload.
- *
- * @param {Array} data packets
- * @param {Function} drain callback
- * @api private
- */
- Polling.prototype.write = function (packets) {
- var self = this;
- this.writable = false;
- var callbackfn = function () {
- self.writable = true;
- self.emit('drain');
- };
- parser.encodePayload(packets, this.supportsBinary, function (data) {
- self.doWrite(data, callbackfn);
- });
- };
- /**
- * Generates uri for connection.
- *
- * @api private
- */
- Polling.prototype.uri = function () {
- var query = this.query || {};
- var schema = this.secure ? 'https' : 'http';
- var port = '';
- // cache busting is forced
- if (false !== this.timestampRequests) {
- query[this.timestampParam] = yeast();
- }
- if (!this.supportsBinary && !query.sid) {
- query.b64 = 1;
- }
- query = parseqs.encode(query);
- // avoid port if default for schema
- if (this.port && (('https' === schema && Number(this.port) !== 443) ||
- ('http' === schema && Number(this.port) !== 80))) {
- port = ':' + this.port;
- }
- // prepend ? to query
- if (query.length) {
- query = '?' + query;
- }
- var ipv6 = this.hostname.indexOf(':') !== -1;
- return schema + '://' + (ipv6 ? '[' + this.hostname + ']' : this.hostname) + port + this.path + query;
- };
|