From 90d7c1a069e5c84f31b38855343765d58947a7d1 Mon Sep 17 00:00:00 2001
From: daz <daz@gradle.com>
Date: Wed, 24 Jan 2024 16:26:29 -0700
Subject: [PATCH] Apply TOS agreement even if plugin is already applied

Fixes #1044
---
 ...build-action.inject-develocity.init.gradle | 27 ++++++++++++-------
 1 file changed, 17 insertions(+), 10 deletions(-)

diff --git a/src/resources/init-scripts/gradle-build-action.inject-develocity.init.gradle b/src/resources/init-scripts/gradle-build-action.inject-develocity.init.gradle
index ab125d8..56dfa0e 100644
--- a/src/resources/init-scripts/gradle-build-action.inject-develocity.init.gradle
+++ b/src/resources/init-scripts/gradle-build-action.inject-develocity.init.gradle
@@ -110,9 +110,6 @@ if (GradleVersion.current() < GradleVersion.version('6.0')) {
                         logger.quiet("Connection to Develocity: $geUrl, allowUntrustedServer: $geAllowUntrustedServer")
                         buildScan.server = geUrl
                         buildScan.allowUntrustedServer = geAllowUntrustedServer
-                    } else if (buildScanTermsOfServiceUrl && buildScanTermsOfServiceAgree) {
-                        buildScan.termsOfServiceUrl = buildScanTermsOfServiceUrl
-                        buildScan.termsOfServiceAgree = buildScanTermsOfServiceAgree
                     }
                     buildScan.publishAlways()
                     if (buildScan.metaClass.respondsTo(buildScan, 'setUploadInBackground', Boolean)) buildScan.uploadInBackground = buildScanUploadInBackground  // uploadInBackground not available for build-scan-plugin 1.16
@@ -128,6 +125,11 @@ if (GradleVersion.current() < GradleVersion.version('6.0')) {
                         }
                     }
                 }
+
+                if (buildScanTermsOfServiceUrl && buildScanTermsOfServiceAgree) {
+                    buildScan.termsOfServiceUrl = buildScanTermsOfServiceUrl
+                    buildScan.termsOfServiceAgree = buildScanTermsOfServiceAgree
+                }
             }
 
             if (ccudPluginVersion && atLeastGradle4) {
@@ -147,14 +149,11 @@ if (GradleVersion.current() < GradleVersion.version('6.0')) {
             if (!settings.pluginManager.hasPlugin(DEVELOCITY_PLUGIN_ID)) {
                 logger.quiet("Applying $DEVELOCITY_PLUGIN_CLASS via init script")
                 applyPluginExternally(settings.pluginManager, DEVELOCITY_PLUGIN_CLASS)
-                extensionsWithPublicType(settings, DEVELOCITY_EXTENSION_CLASS).collect { settings[it.name] }.each { ext ->
+                eachDevelocityExtension(settings, DEVELOCITY_EXTENSION_CLASS) { ext ->
                     if (geUrl) {
                         logger.quiet("Connection to Develocity: $geUrl, allowUntrustedServer: $geAllowUntrustedServer")
                         ext.server = geUrl
                         ext.allowUntrustedServer = geAllowUntrustedServer
-                    } else if (buildScanTermsOfServiceUrl && buildScanTermsOfServiceAgree) {
-                        ext.buildScan.termsOfServiceUrl = buildScanTermsOfServiceUrl
-                        ext.buildScan.termsOfServiceAgree = buildScanTermsOfServiceAgree
                     }
                     ext.buildScan.publishAlways()
                     ext.buildScan.uploadInBackground = buildScanUploadInBackground
@@ -163,12 +162,19 @@ if (GradleVersion.current() < GradleVersion.version('6.0')) {
             }
 
             if (geUrl && geEnforceUrl) {
-                extensionsWithPublicType(settings, DEVELOCITY_EXTENSION_CLASS).collect { settings[it.name] }.each { ext ->
+                eachDevelocityExtension(settings, DEVELOCITY_EXTENSION_CLASS) { ext ->
                     logger.quiet("Enforcing Develocity: $geUrl, allowUntrustedServer: $geAllowUntrustedServer")
                     ext.server = geUrl
                     ext.allowUntrustedServer = geAllowUntrustedServer
                 }
             }
+
+            if (buildScanTermsOfServiceUrl && buildScanTermsOfServiceAgree) {
+                eachDevelocityExtension(settings, DEVELOCITY_EXTENSION_CLASS) { ext ->
+                    ext.buildScan.termsOfServiceUrl = buildScanTermsOfServiceUrl
+                    ext.buildScan.termsOfServiceAgree = buildScanTermsOfServiceAgree
+                }
+            }
         }
 
         if (ccudPluginVersion) {
@@ -195,8 +201,9 @@ void applyPluginExternally(def pluginManager, String pluginClassName) {
     }
 }
 
-static def extensionsWithPublicType(def container, String publicType) {
-    container.extensions.extensionsSchema.elements.findAll { it.publicType.concreteClass.name == publicType }
+static def eachDevelocityExtension(def settings, def publicType, def action) {
+    settings.extensions.extensionsSchema.elements.findAll { it.publicType.concreteClass.name == publicType }
+        .collect { settings[it.name] }.each(action)
 }
 
 static boolean isNotAtLeast(String versionUnderTest, String referenceVersion) {