index.js 1.02 KB
Newer Older
Rosanny Sihombing's avatar
Rosanny Sihombing committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
/*!
 * destroy
 * Copyright(c) 2014 Jonathan Ong
 * MIT Licensed
 */

'use strict'

/**
 * Module dependencies.
 * @private
 */

var ReadStream = require('fs').ReadStream
var Stream = require('stream')

/**
 * Module exports.
 * @public
 */

module.exports = destroy

/**
Rosanny Sihombing's avatar
Rosanny Sihombing committed
25
 * Destroy a stream.
Rosanny Sihombing's avatar
Rosanny Sihombing committed
26
27
28
29
30
 *
 * @param {object} stream
 * @public
 */

Rosanny Sihombing's avatar
Rosanny Sihombing committed
31
32
33
34
35
36
37
function destroy(stream) {
  if (stream instanceof ReadStream) {
    return destroyReadStream(stream)
  }

  if (!(stream instanceof Stream)) {
    return stream
Rosanny Sihombing's avatar
Rosanny Sihombing committed
38
39
  }

Rosanny Sihombing's avatar
Rosanny Sihombing committed
40
41
  if (typeof stream.destroy === 'function') {
    stream.destroy()
Rosanny Sihombing's avatar
Rosanny Sihombing committed
42
43
44
45
46
47
48
49
50
51
52
53
  }

  return stream
}

/**
 * Destroy a ReadStream.
 *
 * @param {object} stream
 * @private
 */

Rosanny Sihombing's avatar
Rosanny Sihombing committed
54
function destroyReadStream(stream) {
Rosanny Sihombing's avatar
Rosanny Sihombing committed
55
56
57
58
59
60
  stream.destroy()

  if (typeof stream.close === 'function') {
    // node.js core bug work-around
    stream.on('open', onOpenClose)
  }
Rosanny Sihombing's avatar
Rosanny Sihombing committed
61

Rosanny Sihombing's avatar
Rosanny Sihombing committed
62
  return stream
Rosanny Sihombing's avatar
Rosanny Sihombing committed
63
64
65
66
67
68
69
}

/**
 * On open handler to close stream.
 * @private
 */

Rosanny Sihombing's avatar
Rosanny Sihombing committed
70
function onOpenClose() {
Rosanny Sihombing's avatar
Rosanny Sihombing committed
71
72
73
74
75
  if (typeof this.fd === 'number') {
    // actually close down the fd
    this.close()
  }
}