From b3f16cfc967d25ee085c94e93a1572a4c30fd847 Mon Sep 17 00:00:00 2001
From: Vyacheslav Gerasimov <vyacheslav.gerasimov@jetbrains.com>
Date: Sun, 1 Mar 2020 16:51:47 +0300
Subject: [PATCH] Build: Add test for dist artifacts manifests and
 compiler.version

---
 build.gradle.kts                 |  1 +
 gradle/checkArtifacts.gradle.kts | 88 ++++++++++++++++++++++++++++++++
 2 files changed, 89 insertions(+)
 create mode 100644 gradle/checkArtifacts.gradle.kts

diff --git a/build.gradle.kts b/build.gradle.kts
index 31c99dd92e0..b77e4e90226 100644
--- a/build.gradle.kts
+++ b/build.gradle.kts
@@ -148,6 +148,7 @@ rootProject.apply {
     from(rootProject.file("gradle/report.gradle.kts"))
     from(rootProject.file("gradle/javaInstrumentation.gradle.kts"))
     from(rootProject.file("gradle/jps.gradle.kts"))
+    from(rootProject.file("gradle/checkArtifacts.gradle.kts"))
 }
 
 IdeVersionConfigurator.setCurrentIde(project)
diff --git a/gradle/checkArtifacts.gradle.kts b/gradle/checkArtifacts.gradle.kts
new file mode 100644
index 00000000000..e14247ef0f9
--- /dev/null
+++ b/gradle/checkArtifacts.gradle.kts
@@ -0,0 +1,88 @@
+import java.io.InputStream
+import java.util.jar.Manifest
+import java.util.zip.ZipFile
+
+val isTeamcityBuild = project.hasProperty("teamcity") || System.getenv("TEAMCITY_VERSION") != null
+
+val distDir: String by rootProject.extra
+val repoDir: String = "${rootProject.buildDir}/repo"
+val kotlinVersion: String by rootProject.extra
+
+val checkMavenArtifacts = tasks.register("checkMavenArtifacts") {
+    doLast {
+        fileTree(repoDir).checkArtifacts { zip, artifactName ->
+            if (!artifactName.endsWith("-sources.jar"))
+                zip.checkCompilerVersion(kotlinVersion)?.let { reportProblem(artifactName, it) }
+
+            zip.checkManifest(kotlinVersion)?.let { reportProblem(artifactName, it) }
+        }
+    }
+}
+
+val checkDist = tasks.register("checkDistArtifacts") {
+    doLast {
+        fileTree(distDir).checkArtifacts { zip, artifactName ->
+            zip.checkCompilerVersion(kotlinVersion)?.let { reportProblem(artifactName, it) }
+        }
+    }
+}
+
+tasks.register("checkArtifacts") {
+    dependsOn(checkDist)
+    dependsOn(checkMavenArtifacts)
+}
+
+fun FileTree.checkArtifacts(action: (zip: ZipFile, artifactName: String) -> Unit) {
+    filter { it.extension == "jar" }.forEach { jar ->
+        val zip = ZipFile(jar)
+        val artifactName = jar.relativeTo(file(rootDir)).invariantSeparatorsPath
+
+        if (isTeamcityBuild)
+            testStarted(artifactName)
+
+        action(zip, artifactName)
+
+        if (isTeamcityBuild)
+            testFinished(artifactName)
+    }
+}
+
+fun ZipFile.checkManifest(version: String) = checkZipEntry("META-INF/MANIFEST.MF") { entryStream ->
+    val implementationVersion = Manifest(entryStream).mainAttributes.getValue("Implementation-Version")
+    "Manifest contains invalid 'Implementation-Version' value, expected: $version found: $implementationVersion".takeIf {
+        implementationVersion?.let { it != version } ?: false
+    }
+}
+
+fun ZipFile.checkCompilerVersion(version: String) = checkZipEntry("META-INF/compiler.version") {
+    val artifactVersion = it.bufferedReader().readLine()
+    "Invalid compiler.version content, expected: $version found: $artifactVersion"
+        .takeIf { artifactVersion != version }
+}
+
+fun ZipFile.checkZipEntry(entryName: String, action: (entryStream: InputStream) -> String?): String? =
+    getEntry(entryName)?.let { entry -> getInputStream(entry).use(action) }
+
+fun reportProblem(artifact: String, message: String) {
+    if (isTeamcityBuild)
+        testFailed(artifact, "Artifact contains problems", message)
+    else
+        println("Artifact $artifact contains problems:\n$message")
+
+}
+
+fun escape(s: String): String {
+    return s.replace("[\\|'\\[\\]]".toRegex(), "\\|$0").replace("\n".toRegex(), "|n").replace("\r".toRegex(), "|r")
+}
+
+fun testStarted(testName: String) {
+    println("##teamcity[testStarted name='%s']".format(escape(testName)))
+}
+
+fun testFinished(testName: String) {
+    println("##teamcity[testFinished name='%s']".format(escape(testName)))
+}
+
+fun testFailed(name: String, message: String, details: String) {
+    println("##teamcity[testFailed name='%s' message='%s' details='%s']".format(escape(name), escape(message), escape(details)))
+}
\ No newline at end of file