Add K/N compiler args builders.

2021-08-18 15:02:43 +03:00
parent b00a43802a
commit 54b3de775d
1 changed files with 183 additions and 0 deletions
@@ -0,0 +1,183 @@
+/*
+ * Copyright 2010-2021 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 org.jetbrains.kotlin.gradle.targets.native.tasks
+
+import org.gradle.api.file.FileCollection
+import org.gradle.api.file.FileTree
+import org.jetbrains.kotlin.gradle.dsl.KotlinCommonToolOptions
+import org.jetbrains.kotlin.gradle.plugin.mpp.BitcodeEmbeddingMode
+import org.jetbrains.kotlin.gradle.tasks.CompilerPluginOptions
+import org.jetbrains.kotlin.konan.target.CompilerOutputKind
+import org.jetbrains.kotlin.konan.target.KonanTarget
+import org.jetbrains.kotlin.project.model.LanguageSettings
+import java.io.File
+
+internal class CompilerPluginData(
+    val files: FileCollection,
+    val options: CompilerPluginOptions
+)
+
+internal class SharedCompilationData(
+    val manifestFile: File,
+    val isAllowCommonizer: Boolean
+)
+
+internal fun buildKotlinNativeKlibCompilerArgs(
+    outFile: File,
+    optimized: Boolean,
+    debuggable: Boolean,
+    target: KonanTarget,
+    libraries: List<File>,
+
+    languageSettings: LanguageSettings,
+    enableEndorsedLibs: Boolean,
+    kotlinOptions: KotlinCommonToolOptions,
+    compilerPlugins: List<CompilerPluginData>,
+
+    moduleName: String,
+    shortModuleName: String,
+    friendModule: FileCollection,
+    sharedCompilationData: SharedCompilationData?,
+    source: FileTree,
+    commonSourcesTree: FileTree
+): List<String> = mutableListOf<String>().apply {
+    addAll(buildKotlinNativeMainArgs(outFile, optimized, debuggable, target, CompilerOutputKind.LIBRARY, libraries))
+
+    sharedCompilationData?.let {
+        add("-Xexpect-actual-linker")
+        add("-Xmetadata-klib")
+        addArg("-manifest", sharedCompilationData.manifestFile.absolutePath)
+        addKey("-no-default-libs", sharedCompilationData.isAllowCommonizer)
+    }
+
+    // Configure FQ module name to avoid cyclic dependencies in klib manifests (see KT-36721).
+    addArg("-module-name", moduleName)
+    add("-Xshort-module-name=$shortModuleName")
+
+    val friends = friendModule.files
+    if (friends.isNotEmpty()) {
+        addArg("-friend-modules", friends.joinToString(File.pathSeparator) { it.absolutePath })
+    }
+
+    addAll(buildKotlinNativeCommonArgs(languageSettings, enableEndorsedLibs, kotlinOptions, compilerPlugins))
+
+    addAll(source.map { it.absolutePath })
+    if (!commonSourcesTree.isEmpty) {
+        add("-Xcommon-sources=${commonSourcesTree.joinToString(separator = ",") { it.absolutePath }}")
+    }
+}
+
+internal fun buildKotlinNativeBinaryLinkerArgs(
+    outFile: File,
+    optimized: Boolean,
+    debuggable: Boolean,
+    target: KonanTarget,
+    outputKind: CompilerOutputKind,
+    libraries: List<File>,
+
+    languageSettings: LanguageSettings,
+    enableEndorsedLibs: Boolean,
+    kotlinOptions: KotlinCommonToolOptions,
+    compilerPlugins: List<CompilerPluginData>,
+
+    processTests: Boolean,
+    entryPoint: String?,
+    embedBitcode: BitcodeEmbeddingMode,
+    linkerOpts: List<String>,
+    binaryOptions: Map<String, String>,
+    isStaticFramework: Boolean,
+    exportLibraries: List<File>,
+    includeLibraries: List<File>,
+    additionalOptions: Collection<String>
+): List<String> = mutableListOf<String>().apply {
+    addAll(buildKotlinNativeMainArgs(outFile, optimized, debuggable, target, outputKind, libraries))
+    addAll(additionalOptions)
+
+    addKey("-tr", processTests)
+    addArgIfNotNull("-entry", entryPoint)
+    when (embedBitcode) {
+        BitcodeEmbeddingMode.MARKER -> add("-Xembed-bitcode-marker")
+        BitcodeEmbeddingMode.BITCODE -> add("-Xembed-bitcode")
+        else -> Unit
+    }
+    linkerOpts.forEach { addArg("-linker-option", it) }
+    binaryOptions.forEach { (name, value) -> add("-Xbinary=$name=$value") }
+    addKey("-Xstatic-framework", isStaticFramework)
+
+    addAll(buildKotlinNativeCommonArgs(languageSettings, enableEndorsedLibs, kotlinOptions, compilerPlugins))
+
+    exportLibraries.forEach { add("-Xexport-library=${it.absolutePath}") }
+    includeLibraries.forEach { add("-Xinclude=${it.absolutePath}") }
+}
+
+private fun buildKotlinNativeMainArgs(
+    outFile: File,
+    optimized: Boolean,
+    debuggable: Boolean,
+    target: KonanTarget,
+    outputKind: CompilerOutputKind,
+    libraries: List<File>
+): List<String> = mutableListOf<String>().apply {
+    addKey("-opt", optimized)
+    addKey("-g", debuggable)
+    addKey("-ea", debuggable)
+    addArg("-target", target.name)
+    addArg("-p", outputKind.name.toLowerCase())
+    addArg("-o", outFile.absolutePath)
+    libraries.forEach { addArg("-l", it.absolutePath) }
+}
+
+internal fun buildKotlinNativeCommonArgs(
+    languageSettings: LanguageSettings,
+    enableEndorsedLibs: Boolean,
+    kotlinOptions: KotlinCommonToolOptions,
+    compilerPlugins: List<CompilerPluginData>
+): List<String> = mutableListOf<String>().apply {
+    add("-Xmulti-platform")
+    addKey("-no-endorsed-libs", !enableEndorsedLibs)
+
+    compilerPlugins.forEach { plugin ->
+        plugin.files.map { it.canonicalPath }.sorted().forEach { add("-Xplugin=$it") }
+        addArgs("-P", plugin.options.arguments)
+    }
+
+    with(languageSettings) {
+        addArgIfNotNull("-language-version", languageVersion)
+        addArgIfNotNull("-api-version", apiVersion)
+        addKey("-progressive", progressiveMode)
+        enabledLanguageFeatures.forEach { add("-XXLanguage:+$it") }
+        optInAnnotationsInUse.forEach { add("-opt-in=$it") }
+    }
+
+    addKey("-Werror", kotlinOptions.allWarningsAsErrors)
+    addKey("-nowarn", kotlinOptions.suppressWarnings)
+    addKey("-verbose", kotlinOptions.verbose)
+
+    addAll(kotlinOptions.freeCompilerArgs)
+}
+
+private fun MutableList<String>.addArg(parameter: String, value: String) {
+    add(parameter)
+    add(value)
+}
+
+private fun MutableList<String>.addArgs(parameter: String, values: Iterable<String>) {
+    values.forEach {
+        addArg(parameter, it)
+    }
+}
+
+private fun MutableList<String>.addArgIfNotNull(parameter: String, value: String?) {
+    if (value != null) {
+        addArg(parameter, value)
+    }
+}
+
+private fun MutableList<String>.addKey(key: String, enabled: Boolean) {
+    if (enabled) {
+        add(key)
+    }
+}