Resolve settings dependencies from CodeArtifact (#3204)

Author: rli

build.gradle.kts

@@ -12,21 +12,10 @@ plugins {
     id("org.jetbrains.gradle.plugin.idea-ext")
 }
 
-val codeArtifactUrl: Provider<String> = providers.environmentVariable("CODEARTIFACT_URL")
-val codeArtifactToken: Provider<String> = providers.environmentVariable("CODEARTIFACT_AUTH_TOKEN")
-
 allprojects {
     repositories {
-        if (codeArtifactUrl.isPresent && codeArtifactToken.isPresent) {
-            println("Using CodeArtifact proxy: ${codeArtifactUrl.get()}")
-            maven {
-                url = uri(codeArtifactUrl.get())
-                credentials {
-                    username = "aws"
-                    password = codeArtifactToken.get()
-                }
-            }
-        }
+        val codeArtifactMavenRepo: ((RepositoryHandler) -> Unit)? by extra
+        codeArtifactMavenRepo?.invoke(this)
         mavenCentral()
         gradlePluginPortal()
     }

buildSrc/settings.gradle.kts

@@ -1,7 +1,12 @@
 // Copyright 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 // SPDX-License-Identifier: Apache-2.0
-val codeArtifactUrl: Provider<String> = providers.environmentVariable("CODEARTIFACT_URL")
-val codeArtifactToken: Provider<String> = providers.environmentVariable("CODEARTIFACT_AUTH_TOKEN")
+pluginManagement {
+    repositories {
+        val codeArtifactMavenRepo: ((RepositoryHandler) -> Unit)? by extra
+        codeArtifactMavenRepo?.invoke(this)
+        gradlePluginPortal()
+    }
+}
 
 dependencyResolutionManagement {
     versionCatalogs {
@@ -11,16 +16,8 @@ dependencyResolutionManagement {
     }
 
     repositories {
-        if (codeArtifactUrl.isPresent && codeArtifactToken.isPresent) {
-            println("Using CodeArtifact proxy: ${codeArtifactUrl.get()}")
-            maven {
-                url = uri(codeArtifactUrl.get())
-                credentials {
-                    username = "aws"
-                    password = codeArtifactToken.get()
-                }
-            }
-        }
+        val codeArtifactMavenRepo: ((RepositoryHandler) -> Unit)? by extra
+        codeArtifactMavenRepo?.invoke(this)
         mavenCentral()
         gradlePluginPortal()
         maven {

settings.gradle.kts

@@ -1,5 +1,32 @@
 // Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 // SPDX-License-Identifier: Apache-2.0
+val codeArtifactMavenRepo = fun RepositoryHandler.(): MavenArtifactRepository? {
+    val codeArtifactUrl: Provider<String> = providers.environmentVariable("CODEARTIFACT_URL")
+    val codeArtifactToken: Provider<String> = providers.environmentVariable("CODEARTIFACT_AUTH_TOKEN")
+    return if (codeArtifactUrl.isPresent && codeArtifactToken.isPresent) {
+        println("Using CodeArtifact proxy: ${codeArtifactUrl.get()}")
+        maven {
+            url = uri(codeArtifactUrl.get())
+            credentials {
+                username = "aws"
+                password = codeArtifactToken.get()
+            }
+        }
+    } else {
+        null
+    }
+}.also {
+    pluginManagement {
+        repositories {
+            // janky because we need to apply to plugins in this file, but val falls out of scope
+            it()
+            gradlePluginPortal()
+        }
+    }
+}
+
+extra["codeArtifactMavenRepo"] = codeArtifactMavenRepo
+
 rootProject.name = "aws-toolkit-jetbrains"
 
 include("resources")