From 5ebac4c6edf547c881baf77c987a097dd3481632 Mon Sep 17 00:00:00 2001
From: Dmitriy Novozhilov <dmitriy.novozhilov@jetbrains.com>
Date: Tue, 19 Apr 2022 11:06:47 +0400
Subject: [PATCH] [Build] Add publication of source of test framework

---
 .../build.gradle.kts                          | 34 ++++++++++++++-----
 1 file changed, 26 insertions(+), 8 deletions(-)

diff --git a/prepare/kotlin-compiler-internal-test-framework/build.gradle.kts b/prepare/kotlin-compiler-internal-test-framework/build.gradle.kts
index b14c71802ae..5ccd2aa2d97 100644
--- a/prepare/kotlin-compiler-internal-test-framework/build.gradle.kts
+++ b/prepare/kotlin-compiler-internal-test-framework/build.gradle.kts
@@ -2,20 +2,38 @@ plugins {
     java
 }
 
+val testModules = listOf(
+    ":compiler:tests-common-jvm6",
+    ":compiler:test-infrastructure",
+    ":compiler:test-infrastructure-utils",
+    ":compiler:tests-compiler-utils",
+    ":compiler:tests-common-new",
+    ":generators:test-generator"
+)
+
+val mainModules = listOf(
+    ":generators"
+)
+
 dependencies {
-    embedded(projectTests(":compiler:tests-common-jvm6")) { isTransitive = false }
-    embedded(projectTests(":compiler:test-infrastructure")) { isTransitive = false }
-    embedded(projectTests(":compiler:test-infrastructure-utils")) { isTransitive = false }
-    embedded(projectTests(":compiler:tests-compiler-utils")) { isTransitive = false }
-    embedded(projectTests(":compiler:tests-common-new")) { isTransitive = false }
+    mainModules.forEach {
+        embedded(project(it)) { isTransitive = false }
+    }
+    testModules.forEach {
+        embedded(projectTests(it)) { isTransitive = false }
+    }
+
     embedded(intellijJavaRt()) { isTransitive = false }
-    embedded(project(":generators")) { isTransitive = false }
-    embedded(projectTests(":generators:test-generator")) { isTransitive = false }
 }
 
 publish()
 runtimeJar()
-sourcesJar().configure {
+sourcesJar {
+    from {
+        mainModules.map { project(it).mainSourceSet.allSource } + testModules.map { project(it).testSourceSet.allSource }
+    }
+
     dependsOn(":compiler:fir:checkers:generateCheckersComponents")
 }
+
 javadocJar()