Show artifact download progress

dorny · Feb 15, 2021 · da9cc2c · da9cc2c
1 parent 8819b4b
commit da9cc2c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/input-providers/artifact-provider.ts b/src/input-providers/artifact-provider.ts
@@ -67,7 +67,7 @@ export class ArtifactProvider implements InputProvider {
     }
 
     for (const art of artifacts) {
-      await downloadArtifact(this.octokit, art.id, art.name, this.token)
+      await downloadArtifact(this.octokit, art.id, art.name, art.size_in_bytes, this.token)
       const reportName = this.getReportName(art.name)
       const files: FileContent[] = []
       const zip = new Zip(art.name)

diff --git a/src/utils/github-utils.ts b/src/utils/github-utils.ts
@@ -33,6 +33,7 @@ export async function downloadArtifact(
   octokit: InstanceType<typeof GitHub>,
   artifactId: number,
   fileName: string,
+  size: number,
   token: string
 ): Promise<void> {
   core.startGroup(`Downloading artifact ${fileName}`)
@@ -72,9 +73,9 @@ export async function downloadArtifact(
     const fileWriterStream = createWriteStream(fileName)
 
     core.info(`Downloading ${url}`)
-    downloadStream.on('downloadProgress', ({transferred, total, percent}) => {
-      const percentage = Math.round(percent * 100)
-      core.info(`Progress: ${transferred}/${total} (${percentage}%)`)
+    downloadStream.on('downloadProgress', ({transferred}) => {
+      const percentage = Math.round(transferred / size * 100)
+      core.info(`Progress: ${transferred}/${size} (${percentage}%)`)
     })
     await asyncStream(downloadStream, fileWriterStream)
   } finally {