Use Gradle Wrapper if specified without Gradle version

Signed-off-by: Anders Fischer-Nielsen <andersfischern@me.com>

src/cache-cleaner.ts

@@ -7,10 +7,12 @@ import path from 'path'
 export class CacheCleaner {
     private readonly gradleUserHome: string
     private readonly tmpDir: string
+    private readonly gradleVersion?: string
 
-    constructor(gradleUserHome: string, tmpDir: string) {
+    constructor(gradleUserHome: string, tmpDir: string, gradleVersion?: string) {
         this.gradleUserHome = gradleUserHome
         this.tmpDir = tmpDir
+        this.gradleVersion = gradleVersion
     }
 
     async prepare(): Promise<void> {
@@ -42,9 +44,15 @@ export class CacheCleaner {
         )
         fs.writeFileSync(path.resolve(cleanupProjectDir, 'build.gradle'), 'task("noop") {}')
 
-        await exec.exec(`gradle -g ${this.gradleUserHome} --no-daemon --build-cache --no-scan --quiet noop`, [], {
-            cwd: cleanupProjectDir
-        })
+        const gradleExecutable = this.gradleVersion === 'wrapper' ? './gradlew' : 'gradle'
+
+        await exec.exec(
+            `${gradleExecutable} -g ${this.gradleUserHome} --no-daemon --build-cache --no-scan --quiet noop`,
+            [],
+            {
+                cwd: cleanupProjectDir
+            }
+        )
     }
 
     private async ageAllFiles(fileName = '*'): Promise<void> {

src/caches.ts

@@ -6,6 +6,7 @@ import {GradleStateCache} from './cache-base'
 import {CacheCleaner} from './cache-cleaner'
 
 const CACHE_RESTORED_VAR = 'GRADLE_BUILD_ACTION_CACHE_RESTORED'
+const GRADLE_VERSION = 'GRADLE_VERSION'
 
 export async function restore(gradleUserHome: string, cacheListener: CacheListener): Promise<void> {
     // Bypass restore cache on all but first action step in workflow.
@@ -48,7 +49,8 @@ export async function restore(gradleUserHome: string, cacheListener: CacheListen
 
     if (isCacheCleanupEnabled() && !isCacheReadOnly()) {
         core.info('Preparing cache for cleanup.')
-        const cacheCleaner = new CacheCleaner(gradleUserHome, process.env['RUNNER_TEMP']!)
+        const gradleVersion = core.getState(GRADLE_VERSION)
+        const cacheCleaner = new CacheCleaner(gradleUserHome, process.env['RUNNER_TEMP']!, gradleVersion)
         await cacheCleaner.prepare()
     }
 }
@@ -78,7 +80,8 @@ export async function save(
 
     if (isCacheCleanupEnabled()) {
         core.info('Forcing cache cleanup.')
-        const cacheCleaner = new CacheCleaner(gradleUserHome, process.env['RUNNER_TEMP']!)
+        const gradleVersion = core.getState(GRADLE_VERSION)
+        const cacheCleaner = new CacheCleaner(gradleUserHome, process.env['RUNNER_TEMP']!, gradleVersion)
         await cacheCleaner.forceCleanup()
     }
 

src/main.ts

@@ -6,6 +6,8 @@ import * as setupGradle from './setup-gradle'
 import * as execution from './execution'
 import * as provision from './provision'
 
+const GRADLE_VERSION = 'GRADLE_VERSION'
+
 /**
  * The main entry point for the action, called by Github Actions for the step.
  */
@@ -39,6 +41,10 @@ run()
 
 async function provisionGradle(workspaceDirectory: string): Promise<string | undefined> {
     const gradleVersion = core.getInput('gradle-version')
+
+    // Save the Gradle version for use in the post-action step.
+    core.saveState(GRADLE_VERSION, gradleVersion)
+
     if (gradleVersion !== '' && gradleVersion !== 'wrapper') {
         return path.resolve(await provision.gradleVersion(gradleVersion))
     }