aboutsummaryrefslogtreecommitdiff
path: root/buildSrc/src/main/kotlin/org/jetbrains/CrossPlatformExec.kt
blob: a68329d86170142c5d026b448bc829be120defa9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
package org.jetbrains

import org.gradle.api.tasks.AbstractExecTask
import org.gradle.internal.os.OperatingSystem
import java.io.File
import java.nio.file.Files
import java.nio.file.Path
import java.nio.file.Paths

open class CrossPlatformExec : AbstractExecTask<CrossPlatformExec>(CrossPlatformExec::class.java) {
    private val windowsExtensions = listOf(".bat", ".cmd", ".exe")
    private val unixExtensions = listOf("", ".sh")

    private val isWindows = OperatingSystem.current().isWindows

    override fun exec() {
        val commandLine: MutableList<String> = this.commandLine

        if (commandLine.isNotEmpty()) {
            commandLine[0] = findCommand(commandLine[0])
        }

        if (isWindows && commandLine.isNotEmpty() && commandLine[0].isNotBlank()) {
            this.commandLine = listOf("cmd", "/c") + commandLine
        } else {
            this.commandLine = commandLine
        }

        super.exec()
    }

    private fun findCommand(command: String): String {
        val command = normalizeCommandPaths(command)
        val extensions = if (isWindows) windowsExtensions else unixExtensions

        return extensions.map { extension ->
            resolveCommandFromFile(Paths.get("$command$extension"))
        }.firstOrNull() ?: command
    }

    private fun resolveCommandFromFile(commandFile: Path) =
        if (!Files.isExecutable(commandFile)) {
            ""
        } else {
            commandFile.toAbsolutePath().normalize().toString()
        }


    private fun normalizeCommandPaths(command: String): String {
        // need to escape backslash so it works with regex
        val backslashSeparator = "\\"
        val forwardSlashSeparator = "/"

        // get the actual separator
        val separator = if (File.separatorChar == '\\') backslashSeparator else File.separator

        return command
            // first replace all of the backslashes with forward slashes
            .replace(backslashSeparator, forwardSlashSeparator)
            // then replace all forward slashes with whatever the separator actually is
            .replace(forwardSlashSeparator, separator)
    }
}