|
1 | 1 | const args = require('minimist')(process.argv.slice(2)); |
2 | | -const nugget = require('nugget'); |
3 | | -const request = require('request'); |
| 2 | +const fs = require('fs'); |
| 3 | +const got = require('got'); |
| 4 | +const stream = require('stream'); |
| 5 | +const { promisify } = require('util'); |
4 | 6 |
|
5 | | -async function makeRequest (requestOptions, parseResponse) { |
6 | | - return new Promise((resolve, reject) => { |
7 | | - request(requestOptions, (err, res, body) => { |
8 | | - if (!err && res.statusCode >= 200 && res.statusCode < 300) { |
9 | | - if (parseResponse) { |
10 | | - const build = JSON.parse(body); |
11 | | - resolve(build); |
12 | | - } else { |
13 | | - resolve(body); |
14 | | - } |
15 | | - } else { |
16 | | - if (args.verbose) { |
17 | | - console.error('Error occurred while requesting:', requestOptions.url); |
18 | | - if (parseResponse) { |
19 | | - try { |
20 | | - console.log('Error: ', `(status ${res.statusCode})`, err || JSON.parse(res.body), requestOptions); |
21 | | - } catch (err) { |
22 | | - console.log('Error: ', `(status ${res.statusCode})`, err || res.body, requestOptions); |
23 | | - } |
24 | | - } else { |
25 | | - console.log('Error: ', `(status ${res.statusCode})`, err || res.body, requestOptions); |
26 | | - } |
27 | | - } |
28 | | - reject(err); |
29 | | - } |
30 | | - }); |
31 | | - }); |
32 | | -} |
| 7 | +const pipeline = promisify(stream.pipeline); |
33 | 8 |
|
34 | 9 | async function downloadArtifact (name, buildNum, dest) { |
35 | 10 | const circleArtifactUrl = `https://circleci.com/api/v1.1/project/github/electron/electron/${args.buildNum}/artifacts?circle-token=${process.env.CIRCLE_TOKEN}`; |
36 | | - const artifacts = await makeRequest({ |
37 | | - method: 'GET', |
38 | | - url: circleArtifactUrl, |
| 11 | + const responsePromise = got(circleArtifactUrl, { |
39 | 12 | headers: { |
40 | 13 | 'Content-Type': 'application/json', |
41 | 14 | Accept: 'application/json' |
42 | 15 | } |
43 | | - }, true).catch(err => { |
44 | | - if (args.verbose) { |
45 | | - console.log('Error calling CircleCI:', err); |
46 | | - } else { |
47 | | - console.error('Error calling CircleCI to get artifact details'); |
48 | | - } |
49 | 16 | }); |
| 17 | + const [response, artifacts] = await Promise.all([responsePromise, responsePromise.json()]); |
| 18 | + if (response.statusCode !== 200) { |
| 19 | + console.error('Could not fetch circleci artifact list, got status code:', response.statusCode); |
| 20 | + } |
50 | 21 | const artifactToDownload = artifacts.find(artifact => { |
51 | 22 | return (artifact.path === name); |
52 | 23 | }); |
@@ -86,19 +57,10 @@ async function downloadWithRetry (url, directory) { |
86 | 57 | } |
87 | 58 |
|
88 | 59 | function downloadFile (url, directory) { |
89 | | - return new Promise((resolve, reject) => { |
90 | | - const nuggetOpts = { |
91 | | - dir: directory, |
92 | | - quiet: args.verbose |
93 | | - }; |
94 | | - nugget(url, nuggetOpts, (err) => { |
95 | | - if (err) { |
96 | | - reject(err); |
97 | | - } else { |
98 | | - resolve(); |
99 | | - } |
100 | | - }); |
101 | | - }); |
| 60 | + return pipeline( |
| 61 | + got.stream(url), |
| 62 | + fs.createWriteStream(directory) |
| 63 | + ); |
102 | 64 | } |
103 | 65 |
|
104 | 66 | if (!args.name || !args.buildNum || !args.dest) { |
|
0 commit comments