Fix CopySamples: make task work for both gradle and gradle.kts files

build-tools/src/main/kotlin/org/jetbrains/kotlin/CopySamples.kt

@@ -3,35 +3,35 @@ package org.jetbrains.kotlin
 import groovy.lang.Closure
 import org.gradle.api.Task
 import org.gradle.api.tasks.Copy
+import org.gradle.api.tasks.InputDirectory
+import java.io.File
 
 /**
  * A task that copies samples and replaces direct repository URLs with ones provided by the cache-redirector service.
  */
 open class CopySamples: Copy() {
-
-    var samplesDir = project.file("samples")
-
-    init {
-        configureReplacements()
-    }
-
-    fun samplesDir(path: Any) {
-        samplesDir = project.file(path)
-    }
+    @InputDirectory
+    var samplesDir: File = project.file("samples")
 
     private fun configureReplacements() {
         from(samplesDir) {
+            it.exclude("**/*.gradle.kts")
             it.exclude("**/*.gradle")
         }
         from(samplesDir) {
             it.include("**/*.gradle")
+            val replacements = replacementsWithWrapper { s -> "maven { url '$s' }" }
             it.filter { line ->
-                replacements.forEach { (repo, replacement) ->
-                    if (line.contains(repo)) {
-                        return@filter line.replace(repo, replacement)
-                    }
-                }
-                return@filter line
+                val repo = line.trim()
+                replacements[repo]?.let { r -> line.replace(repo, r) } ?: line
+            }
+        }
+        from(samplesDir) {
+            it.include("**/*.gradle.kts")
+            val replacements = replacementsWithWrapper { s -> "maven(\"$s\")"}
+            it.filter { line ->
+                val repo = line.trim()
+                replacements[repo]?.let { r -> line.replace(repo, r) } ?: line
             }
         }
     }
@@ -42,14 +42,22 @@ open class CopySamples: Copy() {
         return this
     }
 
-    companion object {
-        val replacements = listOf(
-            "mavenCentral()" to "maven { url 'https://cache-redirector.jetbrains.com/maven-central' }",
-            "jcenter()" to "maven { url 'https://cache-redirector.jetbrains.com/jcenter' }",
-            "https://dl.bintray.com/kotlin/kotlin-dev" to "https://cache-redirector.jetbrains.com/dl.bintray.com/kotlin/kotlin-dev",
-            "https://dl.bintray.com/kotlin/kotlin-eap" to "https://cache-redirector.jetbrains.com/dl.bintray.com/kotlin/kotlin-eap",
-            "https://dl.bintray.com/kotlin/ktor" to "https://cache-redirector.jetbrains.com/dl.bintray.com/kotlin/ktor",
-            "https://plugins.gradle.org/m2" to "https://cache-redirector.jetbrains.com/plugins.gradle.org/m2"
-        )
-    }
+    private fun replacementsWithWrapper(wrap: (String) -> String) =
+        urlReplacements.map { entry ->
+            Pair(wrap(entry.key), wrap(entry.value))
+        }.toMap() + centralReplacements.map { entry ->
+            Pair(entry.key, wrap(entry.value))
+        }.toMap()
+
+    private val urlReplacements = mapOf(
+        "https://dl.bintray.com/kotlin/kotlin-dev" to "https://cache-redirector.jetbrains.com/dl.bintray.com/kotlin/kotlin-dev",
+        "https://dl.bintray.com/kotlin/kotlin-eap" to "https://cache-redirector.jetbrains.com/dl.bintray.com/kotlin/kotlin-eap",
+        "https://dl.bintray.com/kotlin/ktor" to "https://cache-redirector.jetbrains.com/dl.bintray.com/kotlin/ktor",
+        "https://plugins.gradle.org/m2" to "https://cache-redirector.jetbrains.com/plugins.gradle.org/m2"
+    )
+
+    private val centralReplacements = mapOf(
+        "mavenCentral()" to "https://cache-redirector.jetbrains.com/maven-central",
+        "jcenter()" to "https://cache-redirector.jetbrains.com/jcenter",
+    )
 }