Fix decoding of classloader resources

* Fix decoding of classloader resources

* Document internal method

* Fix paths on Windows

(cherry picked from commit 468101f)

Author: jbarr21

core/src/main/kotlin/com/tschuchort/compiletesting/AbstractKotlinCompilation.kt

@@ -21,6 +21,7 @@ import java.io.File
 import java.io.OutputStream
 import java.io.PrintStream
 import java.net.URI
+import java.net.URL
 import java.nio.file.Files
 import java.nio.file.Path
 import java.nio.file.Paths
@@ -62,8 +63,6 @@ abstract class AbstractKotlinCompilation<A : CommonCompilerArguments> internal c
     )
     var compilerPlugins: List<ComponentRegistrar> = emptyList()
 
-
-
     /**
      * Legacy compiler plugins that should be added to the compilation.
      * This option will be removed in the future; you should migrate to [CompilerPluginRegistrar].
@@ -273,23 +272,26 @@ abstract class AbstractKotlinCompilation<A : CommonCompilerArguments> internal c
     }
 
     protected fun getResourcesPath(): String {
-        val resourceName = "META-INF/services/org.jetbrains.kotlin.compiler.plugin.CompilerPluginRegistrar"
         return this::class.java.classLoader.getResources(resourceName)
             .asSequence()
-            .mapNotNull { url ->
-                val uri = URI.create(url.toString().removeSuffix("/$resourceName"))
-                when (uri.scheme) {
-                    "jar" -> Paths.get(URI.create(uri.rawSchemeSpecificPart.removeSuffix("!")))
-                    "file" -> Paths.get(uri)
-                    else -> return@mapNotNull null
-                }.toAbsolutePath()
-            }
+            .mapNotNull { url -> urlToResourcePath(url) }
             .find { resourcesPath ->
                 ServiceLoaderLite.findImplementations(CompilerPluginRegistrar::class.java, listOf(resourcesPath.toFile()))
                     .any { implementation -> implementation == MainComponentAndPluginRegistrar::class.java.name }
             }?.toString() ?: throw AssertionError("Could not get path to CompilerPluginRegistrar service from META-INF")
     }
 
+    /** Maps a URL resource for a class from a JAR or file to an absolute Path on disk  */
+    internal fun urlToResourcePath(url: URL): Path? {
+        val uri = url.toURI()
+        val uriPath = when (uri.scheme) {
+            "jar" -> uri.rawSchemeSpecificPart.removeSuffix("!/$resourceName")
+            "file" -> uri.toString().removeSuffix("/$resourceName")
+            else -> return null
+        }
+        return Paths.get(URI.create(uriPath)).toAbsolutePath()
+    }
+
     /** Searches compiler log for known errors that are hard to debug for the user */
     protected fun searchSystemOutForKnownErrors(compilerSystemOut: String) {
         if (compilerSystemOut.contains("No enum constant com.sun.tools.javac.main.Option.BOOT_CLASS_PATH")) {
@@ -340,8 +342,11 @@ abstract class AbstractKotlinCompilation<A : CommonCompilerArguments> internal c
     protected fun error(s: String) = internalMessageStream.println("error: $s")
 
     internal val internalMessageStreamAccess: PrintStream get() = internalMessageStream
+
+    internal val resourceName = "META-INF/services/org.jetbrains.kotlin.compiler.plugin.CompilerPluginRegistrar"
 }
 
+@ExperimentalCompilerApi
 internal fun convertKotlinExitCode(code: ExitCode) = when(code) {
     ExitCode.OK -> KotlinCompilation.ExitCode.OK
     ExitCode.INTERNAL_ERROR -> KotlinCompilation.ExitCode.INTERNAL_ERROR

core/src/test/kotlin/com/tschuchort/compiletesting/CompilerPluginsTest.kt

@@ -6,6 +6,7 @@ import org.junit.Assert
 import org.junit.Ignore
 import org.junit.Test
 import org.mockito.Mockito
+import java.net.URL
 import javax.annotation.processing.AbstractProcessor
 import javax.annotation.processing.RoundEnvironment
 import javax.lang.model.element.TypeElement
@@ -105,4 +106,25 @@ class CompilerPluginsTest {
         fakePlugin.assertRegistered()
         Assertions.assertThat(result.exitCode).isEqualTo(KotlinCompilation.ExitCode.OK)
     }
+
+    @Test
+    fun `convert jar url resource to path without decoding encoded path`() {
+        // path on disk has "url%3Aport" path segment, but it's encoded from classLoader.getResources()
+        val absolutePath = "jar:file:/path/to/jar/url%253Aport/core-0.4.0.jar!" +
+                "/META-INF/services/org.jetbrains.kotlin.compiler.plugin.ComponentRegistrar"
+        val resultPath = KotlinCompilation().urlToResourcePath(URL(absolutePath)).toString()
+        Assertions.assertThat(resultPath.contains("url:")).isFalse()
+        Assertions.assertThat(resultPath.contains("url%25")).isFalse()
+        Assertions.assertThat(resultPath.contains("url%3A")).isTrue()
+    }
+
+    @Test
+    fun `convert file url resource to path without decoding`() {
+        // path on disk has "repos%3Aoss" path segment, but it's encoded from classLoader.getResources()
+        val absolutePath = "file:/Users/user/repos%253Aoss/kotlin-compile-testing/core/build/resources/main"
+        val resultPath = KotlinCompilation().urlToResourcePath(URL(absolutePath)).toString()
+        Assertions.assertThat(resultPath.contains("repos:")).isFalse()
+        Assertions.assertThat(resultPath.contains("repos%25")).isFalse()
+        Assertions.assertThat(resultPath.contains("repos%3A")).isTrue()
+    }
 }