6 var Transport = require('../transport');
7 var parser = require('engine.io-parser');
8 var util = require('util');
9 var debug = require('debug')('engine:ws');
12 * Export the constructor.
15 module.exports = WebSocket;
20 * @param {http.IncomingMessage}
24 function WebSocket (req) {
25 Transport.call(this, req);
27 this.socket = req.websocket;
28 this.socket.on('message', this.onData.bind(this));
29 this.socket.once('close', this.onClose.bind(this));
30 this.socket.on('error', this.onError.bind(this));
31 this.socket.on('headers', onHeaders);
33 this.perMessageDeflate = null;
35 function onHeaders (headers) {
36 self.emit('headers', headers);
41 * Inherits from Transport.
44 util.inherits(WebSocket, Transport);
52 WebSocket.prototype.name = 'websocket';
55 * Advertise upgrade support.
60 WebSocket.prototype.handlesUpgrades = true;
63 * Advertise framing support.
68 WebSocket.prototype.supportsFraming = true;
71 * Processes the incoming data.
73 * @param {String} encoded packet
77 WebSocket.prototype.onData = function (data) {
78 debug('received "%s"', data);
79 Transport.prototype.onData.call(this, data);
83 * Writes a packet payload.
85 * @param {Array} packets
89 WebSocket.prototype.send = function (packets) {
92 for (var i = 0; i < packets.length; i++) {
93 var packet = packets[i];
94 parser.encodePacket(packet, self.supportsBinary, send);
97 function send (data) {
98 debug('writing "%s"', data);
100 // always creates a new object since ws modifies it
102 if (packet.options) {
103 opts.compress = packet.options.compress;
106 if (self.perMessageDeflate) {
107 var len = 'string' === typeof data ? Buffer.byteLength(data) : data.length;
108 if (len < self.perMessageDeflate.threshold) {
109 opts.compress = false;
113 self.writable = false;
114 self.socket.send(data, opts, onEnd);
117 function onEnd (err) {
118 if (err) return self.onError('write error', err.stack);
119 self.writable = true;
125 * Closes the transport.
130 WebSocket.prototype.doClose = function (fn) {