blob: 1bf055dbb3300ec7d8f8266cee8efd61a1bbb548 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
|
/**
* Module dependencies.
*/
var Transport = require('../transport')
, parser = require('../parser')
, util = require('../util')
, debug = require('debug')('engine.io-client:websocket');
/**
* Module exports.
*/
module.exports = WS;
/**
* Global reference.
*/
var global = 'undefined' != typeof window ? window : global;
/**
* WebSocket transport constructor.
*
* @api {Object} connection options
* @api public
*/
function WS(opts){
Transport.call(this, opts);
};
/**
* Inherits from Transport.
*/
util.inherits(WS, Transport);
/**
* Transport name.
*
* @api public
*/
WS.prototype.name = 'websocket';
/**
* Opens socket.
*
* @api private
*/
WS.prototype.doOpen = function(){
if (!this.check()) {
// let probe timeout
return;
}
var self = this;
this.socket = new (ws())(this.uri());
this.socket.onopen = function(){
self.onOpen();
};
this.socket.onclose = function(){
self.onClose();
};
this.socket.onmessage = function(ev){
self.onData(ev.data);
};
this.socket.onerror = function(e){
self.onError('websocket error', e);
};
};
/**
* Writes data to socket.
*
* @param {Array} array of packets.
* @api private
*/
WS.prototype.write = function(packets){
for (var i = 0, l = packets.length; i < l; i++) {
this.socket.send(parser.encodePacket(packets[i]));
}
};
/**
* Closes socket.
*
* @api private
*/
WS.prototype.doClose = function(){
if (typeof this.socket !== 'undefined') {
this.socket.close();
}
};
/**
* Generates uri for connection.
*
* @api private
*/
WS.prototype.uri = function(){
var query = this.query || {};
var schema = this.secure ? 'wss' : 'ws';
var port = '';
// avoid port if default for schema
if (this.port && (('wss' == schema && this.port != 443)
|| ('ws' == schema && this.port != 80))) {
port = ':' + this.port;
}
// append timestamp to URI
if (this.timestampRequests) {
query[this.timestampParam] = +new Date;
}
query = util.qs(query);
// prepend ? to query
if (query.length) {
query = '?' + query;
}
return schema + '://' + this.host + port + this.path + query;
};
/**
* Feature detection for WebSocket.
*
* @return {Boolean} whether this transport is available.
* @api public
*/
WS.prototype.check = function(){
var websocket = ws();
return !!websocket && !('__initialize' in websocket && this.name === WS.prototype.name);
};
/**
* Getter for WS constructor.
*
* @api private
*/
function ws(){
if ('undefined' != typeof process) {
return require('ws');
}
return global.WebSocket || global.MozWebSocket;
}
|
