From aa9724272b8a9ea9fca3d51295a2cc4707e35ec2 Mon Sep 17 00:00:00 2001
From: Jacob Parish <jacob.parish.1@gmail.com>
Date: Tue, 25 Mar 2025 11:13:23 -0500
Subject: [PATCH] feat: allow specifying a version

---
 __tests__/main.test.ts   | 22 ++++++++++++++++++++--
 action.yml               |  2 +-
 dist/cache-save/index.js |  3 ++-
 dist/setup/index.js      |  3 ++-
 docs/advanced-usage.md   | 13 +++++++++++++
 src/util.ts              |  3 ++-
 6 files changed, 40 insertions(+), 6 deletions(-)

diff --git a/__tests__/main.test.ts b/__tests__/main.test.ts
index dd814ce5..1b80b79b 100644
--- a/__tests__/main.test.ts
+++ b/__tests__/main.test.ts
@@ -301,10 +301,28 @@ describe('main tests', () => {
       );
     });
 
-    it('should enable corepack when input is "true"', async () => {
+    it('should install latest corepack when input is "true"', async () => {
       inputs['corepack'] = 'true';
       await main.run();
-      expect(getCommandOutputSpy).toHaveBeenCalledWith('npm i -g corepack');
+      expect(getCommandOutputSpy).toHaveBeenCalledWith(
+        'npm i -g corepack@latest'
+      );
+    });
+
+    it('should install latest corepack when input is "latest"', async () => {
+      inputs['corepack'] = 'latest';
+      await main.run();
+      expect(getCommandOutputSpy).toHaveBeenCalledWith(
+        'npm i -g corepack@latest'
+      );
+    });
+
+    it('should install a specific version of corepack when specified', async () => {
+      inputs['corepack'] = '0.32.0';
+      await main.run();
+      expect(getCommandOutputSpy).toHaveBeenCalledWith(
+        'npm i -g corepack@0.32.0'
+      );
     });
   });
 });
diff --git a/action.yml b/action.yml
index 3876806c..b97744f8 100644
--- a/action.yml
+++ b/action.yml
@@ -26,7 +26,7 @@ inputs:
   cache-dependency-path:
     description: 'Used to specify the path to a dependency file: package-lock.json, yarn.lock, etc. Supports wildcards or a list of file names for caching multiple dependencies.'
   corepack:
-    description: 'Enables Corepack which allows the use of other package managers.'
+    description: 'Enables Corepack which allows the use of other package managers. Can provide a version string to install a specific version.'
     default: 'false'
 # TODO: add input to control forcing to pull from cloud or dist.
 #       escape valve for someone having issues or needing the absolute latest which isn't cached yet
diff --git a/dist/cache-save/index.js b/dist/cache-save/index.js
index dd2f9fb4..488b3e0d 100644
--- a/dist/cache-save/index.js
+++ b/dist/cache-save/index.js
@@ -88341,7 +88341,8 @@ exports.unique = unique;
 function enableCorepack(input) {
     return __awaiter(this, void 0, void 0, function* () {
         if (input.length && input !== 'false') {
-            yield (0, cache_utils_1.getCommandOutput)('npm i -g corepack');
+            const version = input === 'true' ? 'latest' : input;
+            yield (0, cache_utils_1.getCommandOutput)(`npm i -g corepack@${version}`);
         }
     });
 }
diff --git a/dist/setup/index.js b/dist/setup/index.js
index c5ea08da..9f7d342c 100644
--- a/dist/setup/index.js
+++ b/dist/setup/index.js
@@ -98019,7 +98019,8 @@ exports.unique = unique;
 function enableCorepack(input) {
     return __awaiter(this, void 0, void 0, function* () {
         if (input.length && input !== 'false') {
-            yield (0, cache_utils_1.getCommandOutput)('npm i -g corepack');
+            const version = input === 'true' ? 'latest' : input;
+            yield (0, cache_utils_1.getCommandOutput)(`npm i -g corepack@${version}`);
         }
     });
 }
diff --git a/docs/advanced-usage.md b/docs/advanced-usage.md
index cf84d0fd..42f1c945 100644
--- a/docs/advanced-usage.md
+++ b/docs/advanced-usage.md
@@ -432,3 +432,16 @@ steps:
 - name: Install dependencies
   run: yarn install --immutable
 ```
+
+You can also pass a version string to install a specific version of corepack.
+
+```yaml
+steps:
+- uses: actions/checkout@v4
+- uses: actions/setup-node@v4
+  with:
+    node-version: '18.x'
+    corepack: '0.32.0'
+- name: Install dependencies
+  run: yarn install --immutable
+```
diff --git a/src/util.ts b/src/util.ts
index e5621aca..dc546abc 100644
--- a/src/util.ts
+++ b/src/util.ts
@@ -110,6 +110,7 @@ export const unique = () => {
 
 export async function enableCorepack(input: string): Promise<void> {
   if (input.length && input !== 'false') {
-    await getCommandOutput('npm i -g corepack');
+    const version = input === 'true' ? 'latest' : input;
+    await getCommandOutput(`npm i -g corepack@${version}`);
   }
 }