From 241d1f4327ae928bfa3e4c80bcafcc9761ef1b14 Mon Sep 17 00:00:00 2001
From: Danny Thomas <dannyt@netflix.com>
Date: Thu, 9 Jun 2016 13:15:02 -0700
Subject: [PATCH] Avoid using Gradle wrapper APIs for locating distributions

---
 .../ToolingApiGradleHandleFactory.groovy      | 34 ++++++++++++-------
 1 file changed, 22 insertions(+), 12 deletions(-)

diff --git a/src/main/groovy/nebula/test/functional/internal/toolingapi/ToolingApiGradleHandleFactory.groovy b/src/main/groovy/nebula/test/functional/internal/toolingapi/ToolingApiGradleHandleFactory.groovy
index a2dd2ab..5242434 100644
--- a/src/main/groovy/nebula/test/functional/internal/toolingapi/ToolingApiGradleHandleFactory.groovy
+++ b/src/main/groovy/nebula/test/functional/internal/toolingapi/ToolingApiGradleHandleFactory.groovy
@@ -5,12 +5,9 @@ import groovy.transform.TypeCheckingMode
 import nebula.test.functional.internal.GradleHandle
 import nebula.test.functional.internal.GradleHandleBuildListener
 import nebula.test.functional.internal.GradleHandleFactory
-import org.gradle.initialization.layout.BuildLayout
-import org.gradle.initialization.layout.BuildLayoutFactory
 import org.gradle.tooling.BuildLauncher
 import org.gradle.tooling.GradleConnector
 import org.gradle.tooling.ProjectConnection
-import org.gradle.wrapper.WrapperExecutor
 
 @CompileStatic
 public class ToolingApiGradleHandleFactory implements GradleHandleFactory {
@@ -54,20 +51,33 @@ public class ToolingApiGradleHandleFactory implements GradleHandleFactory {
         }
     }
 
-    private File configureWrapperDistributionIfUsed(GradleConnector connector, File projectDir) {
+    private static void configureWrapperDistributionIfUsed(GradleConnector connector, File projectDir) {
+        // Search above us, in the project that owns the test
         File target = projectDir.absoluteFile
-        while(target!=null) {
-            // Search above us, in the project that owns the test
-            BuildLayout layoutParent = new BuildLayoutFactory().getLayoutFor(target, true)
-            WrapperExecutor wrapperParent = WrapperExecutor.forProjectDirectory(layoutParent.rootDirectory, System.out)
-            if (wrapperParent.distribution) {
-                connector.useDistribution(wrapperParent.distribution)
-                return target
+        while (target != null) {
+            URI distribution = prepareDistributionURI(target)
+            if (distribution) {
+                connector.useDistribution(distribution)
+                return
             }
             target = target.parentFile
         }
     }
 
+    // Translated from org.gradle.wrapper.WrapperExecutor to avoid coupling to Gradle API
+    private static URI prepareDistributionURI(File target) {
+        File propertiesFile = new File(target, "gradle/wrapper/gradle-wrapper.properties")
+        if (propertiesFile.exists()) {
+            Properties properties = new Properties()
+            propertiesFile.withInputStream {
+                properties.load(it)
+            }
+            URI source = new URI(properties.getProperty("distributionUrl"))
+            return source.getScheme() == null ? (new File(propertiesFile.getParentFile(), source.getSchemeSpecificPart())).toURI() : source;
+        }
+        return null
+    }
+
     private boolean isForkedProcess() {
         if (fork) {
             return true
@@ -76,7 +86,7 @@ public class ToolingApiGradleHandleFactory implements GradleHandleFactory {
         Boolean.parseBoolean(System.getProperty(FORK_SYS_PROP, Boolean.FALSE.toString()))
     }
 
-    private BuildLauncher createBuildLauncher(ProjectConnection connection, List<String> arguments, List<String> jvmArguments) {
+    private static BuildLauncher createBuildLauncher(ProjectConnection connection, List<String> arguments, List<String> jvmArguments) {
         BuildLauncher launcher = connection.newBuild();
         launcher.withArguments(arguments as String[]);
         launcher.setJvmArguments(jvmArguments as String[])