forked from openstf/stf
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdownload.js
67 lines (56 loc) · 1.39 KB
/
download.js
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
var fs = require('fs')
var Promise = require('bluebird')
var request = require('request')
var progress = require('request-progress')
var temp = require('temp')
module.exports = function download(url, options) {
var resolver = Promise.defer()
var path = temp.path(options)
function errorListener(err) {
resolver.reject(err)
}
function progressListener(state) {
if (state.total !== null) {
resolver.progress({
lengthComputable: true
, loaded: state.received
, total: state.total
})
}
else {
resolver.progress({
lengthComputable: false
, loaded: state.received
, total: state.received
})
}
}
function closeListener() {
resolver.resolve({
path: path
})
}
resolver.progress({
percent: 0
})
try {
var req = progress(request(url), {
throttle: 100 // Throttle events, not upload speed
})
.on('progress', progressListener)
resolver.promise.finally(function() {
req.removeListener('progress', progressListener)
})
var save = req.pipe(fs.createWriteStream(path))
.on('error', errorListener)
.on('close', closeListener)
resolver.promise.finally(function() {
save.removeListener('error', errorListener)
save.removeListener('close', closeListener)
})
}
catch (err) {
resolver.reject(err)
}
return resolver.promise
}