Propagate kotlin_snapshot_version to buildSrc and kts files (#2332)

Author: qwwdfsad

buildSrc/build.gradle.kts

@@ -9,6 +9,7 @@ plugins {
 }
 
 val cacheRedirectorEnabled = System.getenv("CACHE_REDIRECTOR")?.toBoolean() == true
+val buildSnapshotTrain = properties["build_snapshot_train"]?.toString()?.toBoolean() == true
 
 repositories {
     if (cacheRedirectorEnabled) {
@@ -20,6 +21,10 @@ repositories {
         maven("https://dl.bintray.com/kotlin/kotlin-eap")
         maven("https://dl.bintray.com/kotlin/kotlin-dev")
     }
+
+    if (buildSnapshotTrain) {
+        mavenLocal()
+    }
 }
 
 kotlinDslPluginOptions {
@@ -30,8 +35,14 @@ val props = Properties().apply {
     file("../gradle.properties").inputStream().use { load(it) }
 }
 
-fun version(target: String): String =
-    props.getProperty("${target}_version")
+fun version(target: String): String {
+    // Intercept reading from properties file
+    if (target == "kotlin") {
+        val snapshotVersion = properties["kotlin_snapshot_version"]
+        if (snapshotVersion != null) return snapshotVersion.toString()
+    }
+    return props.getProperty("${target}_version")
+}
 
 dependencies {
     implementation(kotlin("gradle-plugin", version("kotlin")))

buildSrc/settings.gradle.kts

@@ -1,17 +1,18 @@
 /*
  * Copyright 2016-2020 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
  */
-
 pluginManagement {
+    val build_snapshot_train: String? by settings
     repositories {
         val cacheRedirectorEnabled = System.getenv("CACHE_REDIRECTOR")?.toBoolean() == true
-
         if (cacheRedirectorEnabled) {
             println("Redirecting repositories for buildSrc buildscript")
-
             maven("https://cache-redirector.jetbrains.com/plugins.gradle.org/m2")
         } else {
             maven("https://plugins.gradle.org/m2")
         }
+        if (build_snapshot_train?.toBoolean() == true) {
+            mavenLocal()
+        }
     }
 }