From f381b1f0275f4373657fc921edf8806215918290 Mon Sep 17 00:00:00 2001
From: daz <daz@gradle.com>
Date: Wed, 8 Nov 2023 22:45:07 -0800
Subject: [PATCH] Allow artifact retention to be configured

- Default to retaining uploaded artifacts for 7 days.
- Add a parameter to allow this to be configured.
---
 action.yml                     |  5 +++++
 src/dependency-graph.ts        |  6 ++++--
 src/input-params.ts            | 16 ++++++++++++++++
 test/jest/input-params.test.ts | 22 ++++++++++++++++++++++
 4 files changed, 47 insertions(+), 2 deletions(-)
 create mode 100644 test/jest/input-params.test.ts

diff --git a/action.yml b/action.yml
index 4862897..b478e27 100644
--- a/action.yml
+++ b/action.yml
@@ -68,6 +68,11 @@ inputs:
     required: false
     default: 'disabled'
 
+  artifact-retention-days:
+    description: Specifies the number of days to retain any artifacts generated by the action. Defaults to 7 days. Set to zero for maximum retention.
+    required: false
+    default: 7
+
   # EXPERIMENTAL & INTERNAL ACTION INPUTS
   # The following action properties allow fine-grained tweaking of the action caching behaviour.
   # These properties are experimental and not (yet) designed for production use, and may change without notice in a subsequent release of `gradle-build-action`.
diff --git a/src/dependency-graph.ts b/src/dependency-graph.ts
index ef5815d..fda1aa4 100644
--- a/src/dependency-graph.ts
+++ b/src/dependency-graph.ts
@@ -11,7 +11,7 @@ import * as path from 'path'
 import fs from 'fs'
 
 import * as layout from './repository-layout'
-import {DependencyGraphOption, getJobMatrix} from './input-params'
+import {DependencyGraphOption, getJobMatrix, getArtifactRetentionDays} from './input-params'
 
 const DEPENDENCY_GRAPH_ARTIFACT = 'dependency-graph'
 
@@ -60,7 +60,9 @@ async function uploadDependencyGraphs(): Promise<string[]> {
     core.info(`Uploading dependency graph files: ${relativeGraphFiles}`)
 
     const artifactClient = artifact.create()
-    artifactClient.uploadArtifact(DEPENDENCY_GRAPH_ARTIFACT, graphFiles, workspaceDirectory)
+    artifactClient.uploadArtifact(DEPENDENCY_GRAPH_ARTIFACT, graphFiles, workspaceDirectory, {
+        retentionDays: getArtifactRetentionDays()
+    })
 
     return graphFiles
 }
diff --git a/src/input-params.ts b/src/input-params.ts
index 0bd7ab4..42f932c 100644
--- a/src/input-params.ts
+++ b/src/input-params.ts
@@ -88,6 +88,22 @@ export function getDependencyGraphOption(): DependencyGraphOption {
     )
 }
 
+export function getArtifactRetentionDays(): number {
+    const val = core.getInput('artifact-retention-days')
+    return parseNumericInput('artifact-retention-days', val, 7)
+}
+
+export function parseNumericInput(paramName: string, paramValue: string, paramDefault: number): number {
+    if (paramValue.length === 0) {
+        return paramDefault
+    }
+    const numericValue = parseInt(paramValue)
+    if (isNaN(numericValue)) {
+        throw TypeError(`The value '${paramValue}' is not a valid numeric value for '${paramName}'.`)
+    }
+    return numericValue
+}
+
 function getBooleanInput(paramName: string, paramDefault = false): boolean {
     const paramValue = core.getInput(paramName)
     switch (paramValue.toLowerCase().trim()) {
diff --git a/test/jest/input-params.test.ts b/test/jest/input-params.test.ts
new file mode 100644
index 0000000..8db626f
--- /dev/null
+++ b/test/jest/input-params.test.ts
@@ -0,0 +1,22 @@
+import * as inputParams from '../../src/input-params'
+
+describe('input params', () => {
+    describe('parses numeric input', () => {
+        it('uses default value', () => {
+            const val = inputParams.parseNumericInput('param-name', '', 88)
+            expect(val).toBe(88)
+        })
+        it('parses numeric input', () => {
+            const val = inputParams.parseNumericInput('param-name', '34', 88)
+            expect(val).toBe(34)
+        })
+        it('fails on non-numeric input', () => {
+            const t = () => {
+                inputParams.parseNumericInput('param-name', 'xyz', 88)
+            };
+
+            expect(t).toThrow(TypeError)
+            expect(t).toThrow("The value 'xyz' is not a valid numeric value for 'param-name'.")
+        })
+    })
+})