2020-02-06 19:11:24 -06:00
|
|
|
import fetch from "node-fetch";
|
|
|
|
import { throttle } from "throttle-debounce";
|
|
|
|
import * as fs from "fs";
|
2020-02-08 13:03:27 -06:00
|
|
|
import { strict as assert } from "assert";
|
2020-02-06 19:11:24 -06:00
|
|
|
|
2020-02-08 13:03:27 -06:00
|
|
|
/**
|
|
|
|
* Downloads file from `url` and stores it at `destFilePath`.
|
|
|
|
* `onProgress` callback is periodically called to track the progress of downloading,
|
|
|
|
* it gets the already read and total amount of bytes to read as its parameters.
|
|
|
|
*/
|
2020-02-06 19:11:24 -06:00
|
|
|
export async function downloadFile(
|
|
|
|
url: string,
|
|
|
|
destFilePath: fs.PathLike,
|
|
|
|
onProgress: (readBytes: number, totalBytes: number) => void
|
|
|
|
): Promise<void> {
|
2020-02-08 13:25:03 -06:00
|
|
|
onProgress = throttle(200, /* noTrailing: */ true, onProgress);
|
2020-02-06 19:11:24 -06:00
|
|
|
|
|
|
|
const response = await fetch(url);
|
|
|
|
|
|
|
|
const totalBytes = Number(response.headers.get('content-length'));
|
2020-02-08 13:03:27 -06:00
|
|
|
assert(!Number.isNaN(totalBytes), "Sanity check of content-length protocol");
|
|
|
|
|
2020-02-06 19:11:24 -06:00
|
|
|
let readBytes = 0;
|
|
|
|
|
|
|
|
return new Promise<void>((resolve, reject) => response.body
|
|
|
|
.on("data", (chunk: Buffer) => {
|
|
|
|
readBytes += chunk.length;
|
|
|
|
onProgress(readBytes, totalBytes);
|
|
|
|
})
|
|
|
|
.on("end", resolve)
|
|
|
|
.on("error", reject)
|
|
|
|
.pipe(fs.createWriteStream(destFilePath))
|
|
|
|
);
|
|
|
|
}
|