1 var once = require('once');
3 var noop = function() {};
5 var isRequest = function(stream) {
6 return stream.setHeader && typeof stream.abort === 'function';
9 var isChildProcess = function(stream) {
10 return stream.stdio && Array.isArray(stream.stdio) && stream.stdio.length === 3
13 var eos = function(stream, opts, callback) {
14 if (typeof opts === 'function') return eos(stream, null, opts);
17 callback = once(callback || noop);
19 var ws = stream._writableState;
20 var rs = stream._readableState;
21 var readable = opts.readable || (opts.readable !== false && stream.readable);
22 var writable = opts.writable || (opts.writable !== false && stream.writable);
23 var cancelled = false;
25 var onlegacyfinish = function() {
26 if (!stream.writable) onfinish();
29 var onfinish = function() {
31 if (!readable) callback.call(stream);
34 var onend = function() {
36 if (!writable) callback.call(stream);
39 var onexit = function(exitCode) {
40 callback.call(stream, exitCode ? new Error('exited with error code: ' + exitCode) : null);
43 var onerror = function(err) {
44 callback.call(stream, err);
47 var onclose = function() {
48 process.nextTick(onclosenexttick);
51 var onclosenexttick = function() {
52 if (cancelled) return;
53 if (readable && !(rs && (rs.ended && !rs.destroyed))) return callback.call(stream, new Error('premature close'));
54 if (writable && !(ws && (ws.ended && !ws.destroyed))) return callback.call(stream, new Error('premature close'));
57 var onrequest = function() {
58 stream.req.on('finish', onfinish);
61 if (isRequest(stream)) {
62 stream.on('complete', onfinish);
63 stream.on('abort', onclose);
64 if (stream.req) onrequest();
65 else stream.on('request', onrequest);
66 } else if (writable && !ws) { // legacy streams
67 stream.on('end', onlegacyfinish);
68 stream.on('close', onlegacyfinish);
71 if (isChildProcess(stream)) stream.on('exit', onexit);
73 stream.on('end', onend);
74 stream.on('finish', onfinish);
75 if (opts.error !== false) stream.on('error', onerror);
76 stream.on('close', onclose);
80 stream.removeListener('complete', onfinish);
81 stream.removeListener('abort', onclose);
82 stream.removeListener('request', onrequest);
83 if (stream.req) stream.req.removeListener('finish', onfinish);
84 stream.removeListener('end', onlegacyfinish);
85 stream.removeListener('close', onlegacyfinish);
86 stream.removeListener('finish', onfinish);
87 stream.removeListener('exit', onexit);
88 stream.removeListener('end', onend);
89 stream.removeListener('error', onerror);
90 stream.removeListener('close', onclose);