1 var Stream = require('stream').Stream,
2 util = require('util'),
3 driver = require('websocket-driver'),
4 Headers = require('websocket-driver/lib/websocket/driver/headers'),
5 API = require('./websocket/api'),
6 EventTarget = require('./websocket/api/event_target'),
7 Event = require('./websocket/api/event');
9 var EventSource = function(request, response, options) {
11 options = options || {};
13 this._stream = response.socket;
14 this._ping = options.ping || this.DEFAULT_PING;
15 this._retry = options.retry || this.DEFAULT_RETRY;
17 var scheme = driver.isSecureRequest(request) ? 'https:' : 'http:';
18 this.url = scheme + '//' + request.headers.host + request.url;
19 this.lastEventId = request.headers['last-event-id'] || '';
20 this.readyState = API.CONNECTING;
22 var headers = new Headers(),
25 if (options.headers) {
26 for (var key in options.headers) headers.set(key, options.headers[key]);
29 if (!this._stream || !this._stream.writable) return;
30 process.nextTick(function() { self._open() });
32 this._stream.setTimeout(0);
33 this._stream.setNoDelay(true);
35 var handshake = 'HTTP/1.1 200 OK\r\n' +
36 'Content-Type: text/event-stream\r\n' +
37 'Cache-Control: no-cache, no-store\r\n' +
38 'Connection: close\r\n' +
41 'retry: ' + Math.floor(this._retry * 1000) + '\r\n\r\n';
43 this._write(handshake);
45 this._stream.on('drain', function() { self.emit('drain') });
48 this._pingTimer = setInterval(function() { self.ping() }, this._ping * 1000);
50 ['error', 'end'].forEach(function(event) {
51 self._stream.on(event, function() { self.close() });
54 util.inherits(EventSource, Stream);
56 EventSource.isEventSource = function(request) {
57 if (request.method !== 'GET') return false;
58 var accept = (request.headers.accept || '').split(/\s*,\s*/);
59 return accept.indexOf('text/event-stream') >= 0;
66 _write: function(chunk) {
67 if (!this.writable) return false;
69 return this._stream.write(chunk, 'utf8');
76 if (this.readyState !== API.CONNECTING) return;
78 this.readyState = API.OPEN;
80 var event = new Event('open');
81 event.initEvent('open', false, false);
82 this.dispatchEvent(event);
85 write: function(message) {
86 return this.send(message);
89 end: function(message) {
90 if (message !== undefined) this.write(message);
94 send: function(message, options) {
95 if (this.readyState > API.OPEN) return false;
97 message = String(message).replace(/(\r\n|\r|\n)/g, '$1data: ');
98 options = options || {};
101 if (options.event) frame += 'event: ' + options.event + '\r\n';
102 if (options.id) frame += 'id: ' + options.id + '\r\n';
103 frame += 'data: ' + message + '\r\n\r\n';
105 return this._write(frame);
109 return this._write(':\r\n\r\n');
113 if (this.readyState > API.OPEN) return false;
115 this.readyState = API.CLOSED;
116 this.writable = false;
117 if (this._pingTimer) clearInterval(this._pingTimer);
118 if (this._stream) this._stream.end();
120 var event = new Event('close');
121 event.initEvent('close', false, false);
122 this.dispatchEvent(event);
128 for (var method in instance) EventSource.prototype[method] = instance[method];
129 for (var key in EventTarget) EventSource.prototype[key] = EventTarget[key];
131 module.exports = EventSource;