mrva/vscode-codeql
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Extract downloadWithProgress function in ensureCli

Browse Source
This commit is contained in:
Koen Vlaswinkel
2023-12-19 16:49:28 +01:00
parent 216f10d327
commit 641f714fa4
1 changed files with 28 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
extensions/ql-vscode/test/vscode-tests/ensureCli.ts
View File

@@ -88,36 +88,7 @@ export async function ensureCli(useCli: boolean) {
`CLI version ${CLI_VERSION} zip file not found. Downloading from '${url}' into '${downloadedFilePath}'.`,
);
const assetStream = await fetch(url);
const contentLength = Number(
assetStream.headers.get("content-length") || 0,
);
console.log("Total content size", Math.round(contentLength / _1MB), "MB");
const archiveFile = createWriteStream(downloadedFilePath);
const body = assetStream.body;
await new Promise<void>((resolve, reject) => {
let numBytesDownloaded = 0;
let lastMessage = 0;
body.on("data", (data) => {
numBytesDownloaded += data.length;
if (numBytesDownloaded - lastMessage > _10MB) {
console.log(
"Downloaded",
Math.round(numBytesDownloaded / _1MB),
"MB",
);
lastMessage = numBytesDownloaded;
}
archiveFile.write(data);
});
body.on("finish", () => {
archiveFile.end(() => {
console.log("Finished download into", downloadedFilePath);
resolve();
});
});
body.on("error", reject);
});
await downloadWithProgress(url, downloadedFilePath);
} else {
console.log(
`CLI version ${CLI_VERSION} zip file found at '${downloadedFilePath}'.`,
@@ -135,6 +106,33 @@ export async function ensureCli(useCli: boolean) {
}
}
async function downloadWithProgress(url: string, filePath: string) {
const assetStream = await fetch(url);
const contentLength = Number(assetStream.headers.get("content-length") || 0);
console.log("Total content size", Math.round(contentLength / _1MB), "MB");
const archiveFile = createWriteStream(filePath);
const body = assetStream.body;
await new Promise<void>((resolve, reject) => {
let numBytesDownloaded = 0;
let lastMessage = 0;
body.on("data", (data) => {
numBytesDownloaded += data.length;
if (numBytesDownloaded - lastMessage > _10MB) {
console.log("Downloaded", Math.round(numBytesDownloaded / _1MB), "MB");
lastMessage = numBytesDownloaded;
}
archiveFile.write(data);
});
body.on("finish", () => {
archiveFile.end(() => {
console.log("Finished download into", filePath);
resolve();
});
});
body.on("error", reject);
});
}
/**
* Url to download from
*/