[FIR] Add SourceSet.generatedDir() and use it in build.gradle.kts

buildSrc/src/main/kotlin/sourceSets.kt

@@ -37,19 +37,27 @@ val SourceSet.projectDefault: Project.() -> Unit
         }
     }
 
-val SourceSet.generatedTestDir: Project.() -> Unit
+val SourceSet.generatedDir: Project.() -> Unit
     get() = {
-        val generationRoot = projectDir.resolve("tests-gen")
-        java.srcDir(generationRoot.name)
-
-        if (kotlinBuildProperties.isInJpsBuildIdeaSync) {
-            apply(plugin = "idea")
-            idea {
-                this.module.generatedSourceDirs.add(generationRoot)
-            }
-        }
+        generatedDir(this, "gen")
     }
 
+val SourceSet.generatedTestDir: Project.() -> Unit
+    get() = {
+        generatedDir(this, "tests-gen")
+    }
+
+private fun SourceSet.generatedDir(project: Project, dirName: String) {
+    val generationRoot = project.projectDir.resolve(dirName)
+    java.srcDir(generationRoot.name)
+
+    if (project.kotlinBuildProperties.isInJpsBuildIdeaSync) {
+        project.apply(plugin = "idea")
+        project.idea {
+            this.module.generatedSourceDirs.add(generationRoot)
+        }
+    }
+}
 
 val Project.sourceSets: SourceSetContainer
     get() = javaPluginExtension().sourceSets

compiler/fir/checkers/build.gradle.kts

@@ -20,7 +20,7 @@ dependencies {
 sourceSets {
     "main" {
         projectDefault()
-        this.java.srcDir("gen")
+        generatedDir()
     }
     "test" { none() }
 }

compiler/fir/checkers/checkers.js/build.gradle.kts

@@ -22,7 +22,7 @@ dependencies {
 sourceSets {
     "main" {
         projectDefault()
-        this.java.srcDir("gen")
+        generatedDir()
     }
     "test" { none() }
 }

compiler/fir/checkers/checkers.jvm/build.gradle.kts

@@ -23,7 +23,7 @@ dependencies {
 sourceSets {
     "main" {
         projectDefault()
-        this.java.srcDir("gen")
+        generatedDir()
     }
     "test" { none() }
 }

compiler/fir/checkers/checkers.native/build.gradle.kts

@@ -22,7 +22,7 @@ dependencies {
 sourceSets {
     "main" {
         projectDefault()
-        this.java.srcDir("gen")
+        generatedDir()
     }
     "test" { none() }
 }

compiler/fir/tree/build.gradle.kts

@@ -16,7 +16,7 @@ dependencies {
 sourceSets {
     "main" {
         projectDefault()
-        this.java.srcDir("gen")
+        generatedDir()
     }
 }
 

compiler/ir/ir.tree/build.gradle.kts

@@ -19,7 +19,7 @@ dependencies {
 sourceSets {
     "main" {
         projectDefault()
-        this.java.srcDir("gen")
+        generatedDir()
     }
     "test" {}
 }