From a781be725134b79f204b0f41f408ccadc014d6d2 Mon Sep 17 00:00:00 2001
From: Sankalp Kotewar <98868223+kotewar@users.noreply.github.com>
Date: Tue, 22 Nov 2022 05:53:13 +0000
Subject: [PATCH] Take input and update env var

---
 action.yml            | 4 ++++
 dist/restore/index.js | 7 ++++---
 dist/save/index.js    | 1 +
 src/constants.ts      | 3 ++-
 src/restore.ts        | 6 +++---
 5 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/action.yml b/action.yml
index 244abf5..334e201 100644
--- a/action.yml
+++ b/action.yml
@@ -18,6 +18,10 @@ inputs:
     description: 'Fail the workflow if the cache is not found for the given key.'
     required: false
     default: "false"
+  save-cache-on-any-failure:
+    description: 'Save cache despite of any failure in the build steps'
+    required: false
+    default: "no"
 outputs:
   cache-hit:
     description: 'A boolean value to indicate an exact match was found for the primary key'
diff --git a/dist/restore/index.js b/dist/restore/index.js
index 14ff4d8..7896566 100644
--- a/dist/restore/index.js
+++ b/dist/restore/index.js
@@ -4948,6 +4948,7 @@ var Inputs;
     Inputs["RestoreKeys"] = "restore-keys";
     Inputs["UploadChunkSize"] = "upload-chunk-size";
     Inputs["StrictRestore"] = "strict-restore";
+    Inputs["SaveCacheOnAnyFailure"] = "save-cache-on-any-failure";
 })(Inputs = exports.Inputs || (exports.Inputs = {}));
 var Outputs;
 (function (Outputs) {
@@ -48990,10 +48991,10 @@ function run() {
             });
             const cacheKey = yield cache.restoreCache(cachePaths, primaryKey, restoreKeys);
             //Check if user wants to save cache despite of failure in any previous job
-            const saveCache = process.env[constants_1.Variables.SaveCacheOnAnyFailure];
+            const saveCache = process.env[constants_1.Inputs.SaveCacheOnAnyFailure];
             if (saveCache === "yes") {
-                // core.exportVariable(Variables.SaveCacheOnAnyFailure, saveCache);
-                core.info(`Environment Variable ${constants_1.Variables.SaveCacheOnAnyFailure} is set to yes, the cache will be saved despite of any failure in the build.`);
+                core.exportVariable(constants_1.Variables.SaveCacheOnAnyFailure, saveCache);
+                core.info(`Input Variable ${constants_1.Variables.SaveCacheOnAnyFailure} is set to yes, the cache will be saved despite of any failure in the build.`);
             }
             if (!cacheKey) {
                 if (core.getInput(constants_1.Inputs.StrictRestore) == "true") {
diff --git a/dist/save/index.js b/dist/save/index.js
index dde5857..3da4e9a 100644
--- a/dist/save/index.js
+++ b/dist/save/index.js
@@ -4948,6 +4948,7 @@ var Inputs;
     Inputs["RestoreKeys"] = "restore-keys";
     Inputs["UploadChunkSize"] = "upload-chunk-size";
     Inputs["StrictRestore"] = "strict-restore";
+    Inputs["SaveCacheOnAnyFailure"] = "save-cache-on-any-failure";
 })(Inputs = exports.Inputs || (exports.Inputs = {}));
 var Outputs;
 (function (Outputs) {
diff --git a/src/constants.ts b/src/constants.ts
index 81a51eb..66f08ae 100644
--- a/src/constants.ts
+++ b/src/constants.ts
@@ -3,7 +3,8 @@ export enum Inputs {
     Path = "path",
     RestoreKeys = "restore-keys",
     UploadChunkSize = "upload-chunk-size",
-    StrictRestore = "strict-restore"
+    StrictRestore = "strict-restore",
+    SaveCacheOnAnyFailure = "save-cache-on-any-failure"
 }
 
 export enum Outputs {
diff --git a/src/restore.ts b/src/restore.ts
index 7c38b93..065cfd4 100644
--- a/src/restore.ts
+++ b/src/restore.ts
@@ -36,11 +36,11 @@ async function run(): Promise<void> {
         );
 
         //Check if user wants to save cache despite of failure in any previous job
-        const saveCache = process.env[Variables.SaveCacheOnAnyFailure];
+        const saveCache = process.env[Inputs.SaveCacheOnAnyFailure];
         if (saveCache === "yes") {
-            // core.exportVariable(Variables.SaveCacheOnAnyFailure, saveCache);
+            core.exportVariable(Variables.SaveCacheOnAnyFailure, saveCache);
             core.info(
-                `Environment Variable ${Variables.SaveCacheOnAnyFailure} is set to yes, the cache will be saved despite of any failure in the build.`
+                `Input Variable ${Variables.SaveCacheOnAnyFailure} is set to yes, the cache will be saved despite of any failure in the build.`
             );
         }