Add KotlinArtifactsExtensionImpl code generator.

2022-04-14 18:04:34 +02:00
parent 3085f6bacf
commit e974d16e7a
4 changed files with 151 additions and 49 deletions
@@ -24,6 +24,17 @@ val generateAbstractBinaryContainer by generator(
    group = generateGroupName
 }

+val generateAbstractKotlinArtifactsExtensionImplementation by generator(
+    "org.jetbrains.kotlin.generators.gradle.dsl.KotlinArtifactsDSLCodegenKt",
+    sourceSets["main"]
+) {
+    group = generateGroupName
+    systemProperty(
+        "org.jetbrains.kotlin.generators.gradle.dsl.outputSourceRoot",
+        project(":kotlin-gradle-plugin").projectDir.resolve("src/common/kotlin").absolutePath
+    )
+}
+
 val generateKpmNativeVariants by generator(
    "org.jetbrains.kotlin.generators.gradle.dsl.KpmNativeVariantCodegenKt",
    sourceSets["main"]
@@ -0,0 +1,81 @@
+/*
+ * Copyright 2010-2022 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.generators.gradle.dsl
+
+import org.jetbrains.kotlin.gradle.plugin.mpp.BitcodeEmbeddingMode
+import org.jetbrains.kotlin.gradle.plugin.mpp.NativeBuildType
+import org.jetbrains.kotlin.konan.target.KonanTarget
+import java.io.File
+
+fun main() {
+    generateAbstractKotlinArtifactsExtensionImplementation()
+}
+
+private fun generateAbstractKotlinArtifactsExtensionImplementation() {
+    val className = typeName("org.jetbrains.kotlin.gradle.targets.native.tasks.artifact.KotlinArtifactsExtensionImpl")
+
+    val imports = """
+        import org.gradle.api.Project
+        import org.jetbrains.kotlin.gradle.dsl.KotlinArtifact
+        import org.jetbrains.kotlin.gradle.dsl.KotlinArtifactConfig
+        import org.jetbrains.kotlin.gradle.dsl.KotlinArtifactsExtension
+        import org.jetbrains.kotlin.gradle.plugin.mpp.BitcodeEmbeddingMode
+        import org.jetbrains.kotlin.gradle.plugin.mpp.NativeBuildType
+        import org.jetbrains.kotlin.konan.target.KonanTarget
+        import javax.inject.Inject
+    """.trimIndent()
+
+    val generatedCodeWarning = "// DO NOT EDIT MANUALLY! Generated by ${object {}.javaClass.enclosingClass.name}"
+
+    val overrides = listOf(
+        "override val artifactConfigs = project.objects.domainObjectSet(KotlinArtifactConfig::class.java)",
+        "override val artifacts = project.objects.namedDomainObjectSet(KotlinArtifact::class.java)",
+        "override val Native = project.objects.newInstance(KotlinNativeArtifactDSLImpl::class.java, project)"
+    ).joinToString("\n").indented(4)
+
+    val buildTypeConstants = NativeBuildType.values().joinToString("\n") {
+        "val ${it.name} = NativeBuildType.${it.name}"
+    }.indented(4)
+
+    val bitcodeModeConstants = BitcodeEmbeddingMode.values().joinToString(
+        separator = "\n",
+        prefix = "class BitcodeEmbeddingModeDsl {\n",
+        postfix = "\n}"
+    ) {
+        "val ${it.name} = BitcodeEmbeddingMode.${it.name}".indented(4)
+    }.indented(4)
+
+    val bitcodeMode = listOf(
+        "@JvmField",
+        "val EmbedBitcodeMode = BitcodeEmbeddingModeDsl()"
+    ).joinToString("\n").indented(4)
+
+    val konanTargetConstants = KonanTarget.predefinedTargets.values.joinToString("\n") {
+        val nameParts = it.name.split("_")
+        val name = nameParts.drop(1).joinToString(
+            separator = "",
+            prefix = nameParts.first(),
+            transform = String::capitalize
+        )
+        "val $name = KonanTarget.${it.name.toUpperCase()}"
+    }.indented(4)
+
+    val code = listOf(
+        "package ${className.packageName()}",
+        imports,
+        generatedCodeWarning,
+        "abstract class ${className.shortName()} @Inject constructor(project: Project) : KotlinArtifactsExtension {",
+        overrides,
+        buildTypeConstants,
+        bitcodeModeConstants,
+        bitcodeMode,
+        konanTargetConstants,
+        "}"
+    ).joinToString(separator = "\n\n")
+
+    val targetFile = File("$outputSourceRoot/${className.fqName.replace(".", "/")}.kt")
+    targetFile.writeText(code)
+}