From 8371d5fae11fc5a3cb6da240c7d30fec594233cf Mon Sep 17 00:00:00 2001
From: dhawalseth <dhawal.seth@gmail.com>
Date: Wed, 7 Feb 2024 14:21:51 -0800
Subject: [PATCH] Update download-artifact.ts

---
 src/download-artifact.ts | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/download-artifact.ts b/src/download-artifact.ts
index aedfe12..17ddc77 100644
--- a/src/download-artifact.ts
+++ b/src/download-artifact.ts
@@ -21,6 +21,7 @@ async function run(): Promise<void> {
     path: core.getInput(Inputs.Path, {required: false}),
     token: core.getInput(Inputs.GitHubToken, {required: false}),
     repository: core.getInput(Inputs.Repository, {required: false}),
+    maxAttempts: core.getInput(constants_1.Inputs.MaxAttempts, { required: false }),
     runID: parseInt(core.getInput(Inputs.RunID, {required: false})),
     pattern: core.getInput(Inputs.Pattern, {required: false}),
     mergeMultiple: core.getBooleanInput(Inputs.MergeMultiple, {required: false})
@@ -55,6 +56,14 @@ async function run(): Promise<void> {
     }
   }
 
+  if (inputs.maxAttempts) {
+      core.info(`Max attempts for retrying download: ${inputs.maxAttempts}`);
+      options.findBy["maxAttempts"] = inputs.maxAttempts;
+  }
+  else {
+      throw new Error(`Invalid retryCount: '${inputs.maxAttempts}'. Must be greater than 0`);
+  }
+
   let artifacts: Artifact[] = []
 
   if (isSingleArtifactDownload) {