socket.js 8.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471
  1. /**
  2. * Module dependencies.
  3. */
  4. var Emitter = require('events').EventEmitter;
  5. var parser = require('socket.io-parser');
  6. var url = require('url');
  7. var debug = require('debug')('socket.io:socket');
  8. var hasBin = require('has-binary');
  9. /**
  10. * Module exports.
  11. */
  12. module.exports = exports = Socket;
  13. /**
  14. * Blacklisted events.
  15. *
  16. * @api public
  17. */
  18. exports.events = [
  19. 'error',
  20. 'connect',
  21. 'disconnect',
  22. 'newListener',
  23. 'removeListener'
  24. ];
  25. /**
  26. * Flags.
  27. *
  28. * @api private
  29. */
  30. var flags = [
  31. 'json',
  32. 'volatile',
  33. 'broadcast'
  34. ];
  35. /**
  36. * `EventEmitter#emit` reference.
  37. */
  38. var emit = Emitter.prototype.emit;
  39. /**
  40. * Interface to a `Client` for a given `Namespace`.
  41. *
  42. * @param {Namespace} nsp
  43. * @param {Client} client
  44. * @api public
  45. */
  46. function Socket(nsp, client){
  47. this.nsp = nsp;
  48. this.server = nsp.server;
  49. this.adapter = this.nsp.adapter;
  50. this.id = nsp.name + '#' + client.id;
  51. this.client = client;
  52. this.conn = client.conn;
  53. this.rooms = {};
  54. this.acks = {};
  55. this.connected = true;
  56. this.disconnected = false;
  57. this.handshake = this.buildHandshake();
  58. }
  59. /**
  60. * Inherits from `EventEmitter`.
  61. */
  62. Socket.prototype.__proto__ = Emitter.prototype;
  63. /**
  64. * Apply flags from `Socket`.
  65. */
  66. flags.forEach(function(flag){
  67. Socket.prototype.__defineGetter__(flag, function(){
  68. this.flags = this.flags || {};
  69. this.flags[flag] = true;
  70. return this;
  71. });
  72. });
  73. /**
  74. * `request` engine.io shortcut.
  75. *
  76. * @api public
  77. */
  78. Socket.prototype.__defineGetter__('request', function(){
  79. return this.conn.request;
  80. });
  81. /**
  82. * Builds the `handshake` BC object
  83. *
  84. * @api private
  85. */
  86. Socket.prototype.buildHandshake = function(){
  87. return {
  88. headers: this.request.headers,
  89. time: (new Date) + '',
  90. address: this.conn.remoteAddress,
  91. xdomain: !!this.request.headers.origin,
  92. secure: !!this.request.connection.encrypted,
  93. issued: +(new Date),
  94. url: this.request.url,
  95. query: url.parse(this.request.url, true).query || {}
  96. };
  97. };
  98. /**
  99. * Emits to this client.
  100. *
  101. * @return {Socket} self
  102. * @api public
  103. */
  104. Socket.prototype.emit = function(ev){
  105. if (~exports.events.indexOf(ev)) {
  106. emit.apply(this, arguments);
  107. } else {
  108. var args = Array.prototype.slice.call(arguments);
  109. var packet = {};
  110. packet.type = hasBin(args) ? parser.BINARY_EVENT : parser.EVENT;
  111. packet.data = args;
  112. var flags = this.flags || {};
  113. // access last argument to see if it's an ACK callback
  114. if ('function' == typeof args[args.length - 1]) {
  115. if (this._rooms || flags.broadcast) {
  116. throw new Error('Callbacks are not supported when broadcasting');
  117. }
  118. debug('emitting packet with ack id %d', this.nsp.ids);
  119. this.acks[this.nsp.ids] = args.pop();
  120. packet.id = this.nsp.ids++;
  121. }
  122. if (this._rooms || flags.broadcast) {
  123. this.adapter.broadcast(packet, {
  124. except: [this.id],
  125. rooms: this._rooms,
  126. flags: flags
  127. });
  128. } else {
  129. // dispatch packet
  130. this.packet(packet, {
  131. volatile: flags.volatile,
  132. compress: flags.compress
  133. });
  134. }
  135. // reset flags
  136. delete this._rooms;
  137. delete this.flags;
  138. }
  139. return this;
  140. };
  141. /**
  142. * Targets a room when broadcasting.
  143. *
  144. * @param {String} name
  145. * @return {Socket} self
  146. * @api public
  147. */
  148. Socket.prototype.to =
  149. Socket.prototype.in = function(name){
  150. this._rooms = this._rooms || [];
  151. if (!~this._rooms.indexOf(name)) this._rooms.push(name);
  152. return this;
  153. };
  154. /**
  155. * Sends a `message` event.
  156. *
  157. * @return {Socket} self
  158. * @api public
  159. */
  160. Socket.prototype.send =
  161. Socket.prototype.write = function(){
  162. var args = Array.prototype.slice.call(arguments);
  163. args.unshift('message');
  164. this.emit.apply(this, args);
  165. return this;
  166. };
  167. /**
  168. * Writes a packet.
  169. *
  170. * @param {Object} packet object
  171. * @param {Object} opts options
  172. * @api private
  173. */
  174. Socket.prototype.packet = function(packet, opts){
  175. packet.nsp = this.nsp.name;
  176. opts = opts || {};
  177. opts.compress = false !== opts.compress;
  178. this.client.packet(packet, opts);
  179. };
  180. /**
  181. * Joins a room.
  182. *
  183. * @param {String} room
  184. * @param {Function} fn optional, callback
  185. * @return {Socket} self
  186. * @api private
  187. */
  188. Socket.prototype.join = function(room, fn){
  189. debug('joining room %s', room);
  190. var self = this;
  191. if (this.rooms.hasOwnProperty(room)) {
  192. fn && fn(null);
  193. return this;
  194. }
  195. this.adapter.add(this.id, room, function(err){
  196. if (err) return fn && fn(err);
  197. debug('joined room %s', room);
  198. self.rooms[room] = room;
  199. fn && fn(null);
  200. });
  201. return this;
  202. };
  203. /**
  204. * Leaves a room.
  205. *
  206. * @param {String} room
  207. * @param {Function} fn optional, callback
  208. * @return {Socket} self
  209. * @api private
  210. */
  211. Socket.prototype.leave = function(room, fn){
  212. debug('leave room %s', room);
  213. var self = this;
  214. this.adapter.del(this.id, room, function(err){
  215. if (err) return fn && fn(err);
  216. debug('left room %s', room);
  217. delete self.rooms[room];
  218. fn && fn(null);
  219. });
  220. return this;
  221. };
  222. /**
  223. * Leave all rooms.
  224. *
  225. * @api private
  226. */
  227. Socket.prototype.leaveAll = function(){
  228. this.adapter.delAll(this.id);
  229. this.rooms = {};
  230. };
  231. /**
  232. * Called by `Namespace` upon succesful
  233. * middleware execution (ie: authorization).
  234. *
  235. * @api private
  236. */
  237. Socket.prototype.onconnect = function(){
  238. debug('socket connected - writing packet');
  239. this.nsp.connected[this.id] = this;
  240. this.join(this.id);
  241. this.packet({ type: parser.CONNECT });
  242. };
  243. /**
  244. * Called with each packet. Called by `Client`.
  245. *
  246. * @param {Object} packet
  247. * @api private
  248. */
  249. Socket.prototype.onpacket = function(packet){
  250. debug('got packet %j', packet);
  251. switch (packet.type) {
  252. case parser.EVENT:
  253. this.onevent(packet);
  254. break;
  255. case parser.BINARY_EVENT:
  256. this.onevent(packet);
  257. break;
  258. case parser.ACK:
  259. this.onack(packet);
  260. break;
  261. case parser.BINARY_ACK:
  262. this.onack(packet);
  263. break;
  264. case parser.DISCONNECT:
  265. this.ondisconnect();
  266. break;
  267. case parser.ERROR:
  268. this.emit('error', packet.data);
  269. }
  270. };
  271. /**
  272. * Called upon event packet.
  273. *
  274. * @param {Object} packet object
  275. * @api private
  276. */
  277. Socket.prototype.onevent = function(packet){
  278. var args = packet.data || [];
  279. debug('emitting event %j', args);
  280. if (null != packet.id) {
  281. debug('attaching ack callback to event');
  282. args.push(this.ack(packet.id));
  283. }
  284. emit.apply(this, args);
  285. };
  286. /**
  287. * Produces an ack callback to emit with an event.
  288. *
  289. * @param {Number} id packet id
  290. * @api private
  291. */
  292. Socket.prototype.ack = function(id){
  293. var self = this;
  294. var sent = false;
  295. return function(){
  296. // prevent double callbacks
  297. if (sent) return;
  298. var args = Array.prototype.slice.call(arguments);
  299. debug('sending ack %j', args);
  300. var type = hasBin(args) ? parser.BINARY_ACK : parser.ACK;
  301. self.packet({
  302. id: id,
  303. type: type,
  304. data: args
  305. });
  306. sent = true;
  307. };
  308. };
  309. /**
  310. * Called upon ack packet.
  311. *
  312. * @api private
  313. */
  314. Socket.prototype.onack = function(packet){
  315. var ack = this.acks[packet.id];
  316. if ('function' == typeof ack) {
  317. debug('calling ack %s with %j', packet.id, packet.data);
  318. ack.apply(this, packet.data);
  319. delete this.acks[packet.id];
  320. } else {
  321. debug('bad ack %s', packet.id);
  322. }
  323. };
  324. /**
  325. * Called upon client disconnect packet.
  326. *
  327. * @api private
  328. */
  329. Socket.prototype.ondisconnect = function(){
  330. debug('got disconnect packet');
  331. this.onclose('client namespace disconnect');
  332. };
  333. /**
  334. * Handles a client error.
  335. *
  336. * @api private
  337. */
  338. Socket.prototype.onerror = function(err){
  339. if (this.listeners('error').length) {
  340. this.emit('error', err);
  341. } else {
  342. console.error('Missing error handler on `socket`.');
  343. console.error(err.stack);
  344. }
  345. };
  346. /**
  347. * Called upon closing. Called by `Client`.
  348. *
  349. * @param {String} reason
  350. * @throw {Error} optional error object
  351. * @api private
  352. */
  353. Socket.prototype.onclose = function(reason){
  354. if (!this.connected) return this;
  355. debug('closing socket - reason %s', reason);
  356. this.leaveAll();
  357. this.nsp.remove(this);
  358. this.client.remove(this);
  359. this.connected = false;
  360. this.disconnected = true;
  361. delete this.nsp.connected[this.id];
  362. this.emit('disconnect', reason);
  363. };
  364. /**
  365. * Produces an `error` packet.
  366. *
  367. * @param {Object} err error object
  368. * @api private
  369. */
  370. Socket.prototype.error = function(err){
  371. this.packet({ type: parser.ERROR, data: err });
  372. };
  373. /**
  374. * Disconnects this client.
  375. *
  376. * @param {Boolean} close if `true`, closes the underlying connection
  377. * @return {Socket} self
  378. * @api public
  379. */
  380. Socket.prototype.disconnect = function(close){
  381. if (!this.connected) return this;
  382. if (close) {
  383. this.client.disconnect();
  384. } else {
  385. this.packet({ type: parser.DISCONNECT });
  386. this.onclose('server namespace disconnect');
  387. }
  388. return this;
  389. };
  390. /**
  391. * Sets the compress flag.
  392. *
  393. * @param {Boolean} compress if `true`, compresses the sending data
  394. * @return {Socket} self
  395. * @api public
  396. */
  397. Socket.prototype.compress = function(compress){
  398. this.flags = this.flags || {};
  399. this.flags.compress = compress;
  400. return this;
  401. };