kotlin.script.experimental.jvm.util.jvmClassLoaderUtil.kt Maven / Gradle / Ivy

/*
 * Copyright 2010-2019 JetBrains s.r.o. and Kotlin Programming Language contributors.
 * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
 */

package kotlin.script.experimental.jvm.util

import java.io.File
import java.io.FileInputStream
import java.io.IOException
import java.io.InputStream
import java.net.JarURLConnection
import java.net.URL
import java.util.jar.JarFile
import java.util.jar.JarInputStream
import kotlin.script.experimental.jvm.impl.toFileOrNull

fun ClassLoader.forAllMatchingFiles(namePattern: String, body: (String, InputStream) -> Unit) {
    val processedDirs = HashSet()
    val processedJars = HashSet()
    val nameRegex = namePatternToRegex(namePattern)

    fun iterateResources(vararg keyResourcePaths: String) {
        for (keyResourcePath in keyResourcePaths) {
            val resourceRootCalc = ClassLoaderResourceRootFIlePathCalculator(keyResourcePath)
            for (url in getResources(keyResourcePath)) {
                if (url.protocol == "jar") {
                    val jarConnection = url.openConnection() as? JarURLConnection
                    val jarUrl = jarConnection?.jarFileURL
                    if (jarUrl != null && !processedJars.contains(jarUrl)) {
                        processedJars.add(jarUrl)
                        try {
                            jarConnection.jarFile
                        } catch (_: IOException) {
                            // TODO: consider error reporting
                            null
                        }?.let {
                            forAllMatchingFilesInJarFile(it, nameRegex, body)
                        }
                    }
                } else {
                    val rootDir = url.toFileOrNull()?.let { resourceRootCalc(it) }
                    if (rootDir != null && rootDir.isDirectory && !processedDirs.contains(rootDir)) {
                        processedDirs.add(rootDir)
                        forAllMatchingFilesInDirectory(rootDir, namePattern, body)
                    }
                }
            }
        }
    }

    iterateResources("", JAR_MANIFEST_RESOURCE_NAME)
}

internal val wildcardChars = "*?".toCharArray()
internal val patternCharsToEscape = ".*?+()[]^\${}|".toCharArray().also { assert(wildcardChars.all { wc -> it.contains(wc) }) }

private fun Char.escape(): String = (if (this == '\\' || patternCharsToEscape.contains(this)) "\\" else "") + this

internal val pathSeparatorChars = "/".let { if (File.separatorChar == '/') it else it + File.separator }.toCharArray()
internal val pathElementPattern = if (File.separatorChar == '/') "[^/]*" else "[^/${File.separatorChar.escape()}]*"
internal val pathSeparatorPattern = if (File.separatorChar == '/') "/" else "[/${File.separatorChar.escape()}]."
internal val specialPatternChars = patternCharsToEscape + pathSeparatorChars

private fun String.toUniversalSeparator(): String = if (File.separatorChar == '/') this else replace(File.separatorChar, '/')

internal fun forAllMatchingFilesInDirectory(baseDir: File, namePattern: String, body: (String, InputStream) -> Unit) {
    val patternStart = namePattern.indexOfAny(wildcardChars)
    if (patternStart < 0) {
        // assuming a single file
        baseDir.resolve(namePattern).takeIf { it.exists() && it.isFile }?.let { file ->
            body(file.relativeToOrSelf(baseDir).path.toUniversalSeparator(), file.inputStream())
        }
    } else {
        val patternDirStart = namePattern.lastIndexOfAny(pathSeparatorChars, patternStart)
        val root = if (patternDirStart <= 0) baseDir else baseDir.resolve(namePattern.substring(0, patternDirStart))
        if (root.exists() && root.isDirectory) {
            val re = namePatternToRegex(namePattern.substring(patternDirStart + 1))
            root.walkTopDown().filter {
                re.matches(it.relativeToOrSelf(root).path)
            }.forEach { file ->
                body(file.relativeToOrSelf(baseDir).path.toUniversalSeparator(), file.inputStream())
            }
        }
    }
}

internal fun forAllMatchingFilesInJarStream(jarInputStream: JarInputStream, nameRegex: Regex, body: (String, InputStream) -> Unit) {
    do {
        val entry = jarInputStream.nextJarEntry
        if (entry != null) {
            try {
                if (!entry.isDirectory && nameRegex.matches(entry.name)) {
                    body(entry.name, jarInputStream)
                }
            } finally {
                jarInputStream.closeEntry()
            }
        }
    } while (entry != null)
}

internal fun forAllMatchingFilesInJar(jarFile: File, nameRegex: Regex, body: (String, InputStream) -> Unit) {
    JarInputStream(FileInputStream(jarFile)).use {
        forAllMatchingFilesInJarStream(it, nameRegex, body)
    }
}

internal fun forAllMatchingFilesInJarFile(jarFile: JarFile, nameRegex: Regex, body: (String, InputStream) -> Unit) {
    jarFile.entries().asSequence().forEach { entry ->
        if (!entry.isDirectory && nameRegex.matches(entry.name)) {
            jarFile.getInputStream(entry).use { stream ->
                body(entry.name, stream)
            }
        }
    }
}

internal fun namePatternToRegex(pattern: String): Regex = Regex(
    buildString {
        var current = 0
        loop@ while (current < pattern.length) {
            val nextIndex = pattern.indexOfAny(specialPatternChars, current)
            val next = if (nextIndex < 0) pattern.length else nextIndex
            append(pattern.substring(current, next))
            current = next + 1
            when {
                next >= pattern.length -> break@loop

                pathSeparatorChars.contains(pattern[next]) -> append(pathSeparatorPattern)

                pattern[next] == '?' -> append('.')

                pattern[next] == '*' && next + 1 < pattern.length && pattern[next + 1] == '*' -> {
                    append(".*")
                    current++
                }

                pattern[next] == '*' -> append(pathElementPattern)

                else -> {
                    append('\\')
                    append(pattern[next])
                }
            }
        }
    }
)