Merge pull request #184 from microsoft/users/sergey.koryshev/4113-ref…

…actor-nodejs-downloading Add logic to handle HTTP-errors during file downloading
microsoft · Mar 9, 2023 · c729a59 · c729a59
2 parents 2de34c5 + b5652ea
commit c729a59
Show file tree

Hide file tree

Showing 3 changed files with 47 additions and 40 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -24,3 +24,7 @@
 ### 2.0.0
 
 - Bump `azure-pipelines-task-lib` to `4.1.0`
+
+### 2.0.1
+
+- <https://github.com/microsoft/azure-pipelines-tool-lib/pull/184>
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "azure-pipelines-tool-lib",
-  "version": "2.0.0",
+  "version": "2.0.1",
   "description": "Azure Pipelines Tool Installer Lib for CI/CD Tasks",
   "main": "tool.js",
   "scripts": {

diff --git a/tool.ts b/tool.ts
@@ -249,46 +249,49 @@ export async function downloadTool(
             }
 
             tl.debug('creating stream');
-            let file: NodeJS.WritableStream = fs.createWriteStream(destPath);
-            file.on('open', async (fd) => {
-                try {
-                    let stream = response.message.pipe(file);
-                    stream.on('close', () => {
-                        tl.debug('download complete');
-                        let fileSizeInBytes: number;
-                        try {
-                            fileSizeInBytes = _getFileSizeOnDisk(destPath);
-                        }
-                        catch (err) {
-                            fileSizeInBytes = NaN;
-                            tl.warning(`Unable to check file size of ${destPath} due to error: ${err.Message}`);
-                        }
-
-                        if (!isNaN(fileSizeInBytes)) {
-                            tl.debug(`Downloaded file size: ${fileSizeInBytes} bytes`);
-                        } else {
-                            tl.debug(`File size on disk was not found`);
-                        }
-
-                        if (!isNaN(downloadedContentLength) &&
-                            !isNaN(fileSizeInBytes) &&
-                            fileSizeInBytes !== downloadedContentLength) {
-                            tl.warning(`Content-Length (${downloadedContentLength} bytes) did not match downloaded file size (${fileSizeInBytes} bytes).`);
-                        }
-
-                        resolve(destPath);
-                    });
-                }
-                catch (err) {
+            const file: NodeJS.WritableStream = fs.createWriteStream(destPath);
+            file
+                .on('open', async (fd) => {
+                    try {
+                        response.message
+                            .on('error', (err) => {
+                                file.end();
+                                reject(err);
+                            })
+                            .pipe(file);
+                    } catch (err) {
+                        reject(err);
+                    }
+                })
+                .on('close', () => {
+                    tl.debug('download complete');
+                    let fileSizeInBytes: number;
+                    try {
+                        fileSizeInBytes = _getFileSizeOnDisk(destPath);
+                    } catch (err) {
+                        fileSizeInBytes = NaN;
+                        tl.warning(`Unable to check file size of ${destPath} due to error: ${err.Message}`);
+                    }
+
+                    if (!isNaN(fileSizeInBytes)) {
+                        tl.debug(`Downloaded file size: ${fileSizeInBytes} bytes`);
+                    } else {
+                        tl.debug(`File size on disk was not found`);
+                    }
+
+                    if (!isNaN(downloadedContentLength) &&
+                        !isNaN(fileSizeInBytes) &&
+                        fileSizeInBytes !== downloadedContentLength) {
+                        tl.warning(`Content-Length (${downloadedContentLength} bytes) did not match downloaded file size (${fileSizeInBytes} bytes).`);
+                    }
+
+                    resolve(destPath);
+                })
+                .on('error', (err) => {
+                    file.end();
                     reject(err);
-                }
-            });
-            file.on('error', (err) => {
-                file.end();
-                reject(err);
-            });
-        }
-        catch (error) {
+                });
+        } catch (error) {
             reject(error);
         }
     });