commonMain.co.touchlab.skie.plugin.switflink.SwiftBundlingConfigurator.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gradle-plugin-impl-jvm Show documentation
Internal implementation for SKIE Gradle plugin.
There is a newer version: 0.9.0-RC.5
Show newest version
package co.touchlab.skie.plugin.switflink

import co.touchlab.skie.plugin.configuration.skieExtension
import co.touchlab.skie.plugin.kgpShim
import co.touchlab.skie.plugin.shim.KotlinNativeCompilationShim
import co.touchlab.skie.plugin.shim.KotlinSourceSetShim
import co.touchlab.skie.plugin.util.doLastOptimized
import co.touchlab.skie.plugin.util.lowerCamelCaseName
import co.touchlab.skie.plugin.util.registerSkieTask
import co.touchlab.skie.plugin.util.writeToZip
import co.touchlab.skie.util.cache.syncDirectoryContentIfDifferent
import org.gradle.api.Project
import org.gradle.api.file.SourceDirectorySet
import org.gradle.api.provider.Provider
import java.io.File

object SwiftBundlingConfigurator {

    const val KLIB_SKIE_SWIFT_DIRECTORY: String = "default/skie/swift"

    fun configureCustomSwiftBundling(project: Project) {
        project.kgpShim.kotlinNativeTargets.configureEach {
            compilations.configureEach {
                project.configureSwiftBundlingToKlib(this)
            }
        }
    }

    private fun Project.configureSwiftBundlingToKlib(compilation: KotlinNativeCompilationShim) {
        val processSwiftSourcesTaskProvider = createProcessSwiftSourcesTask(compilation)

        compilation.configureCompileTask(processSwiftSourcesTaskProvider)
    }

    private fun Project.createProcessSwiftSourcesTask(compilation: KotlinNativeCompilationShim): Provider {
        val compilationPrefix = compilation.name.takeIf { it.lowercase() != "main" }

        val baseName = lowerCamelCaseName("processSwiftSources", compilationPrefix, compilation.target.name)

        val swiftSourceSet = createSwiftSourceSet(compilation)

        val isSwiftBundlingEnabledProperty = skieExtension.swiftBundling.enabled

        return registerSkieTask(baseName) {
            inputs.files(swiftSourceSet)
            output.set(compilation.skieCompilationDirectory.map { it.swift.bundled.directory })

            onlyIf {
                isSwiftBundlingEnabledProperty.get()
            }
        }
    }

    private fun Project.createSwiftSourceSet(compilation: KotlinNativeCompilationShim): SourceDirectorySet {
        val swiftSourceSetName = "${compilation.target.name}:${compilation.name} Swift sources"

        val swiftSourceSet = objects.sourceDirectorySet(swiftSourceSetName, swiftSourceSetName).apply {
            filter.include("**/*.swift")
        }

        compilation.allKotlinSourceSets.configureEach {
            val swiftDirectory = project.layout.projectDirectory.dir(swiftSourceDirectory)

            swiftSourceSet.srcDirs(swiftDirectory)
        }

        return swiftSourceSet
    }

    private fun KotlinNativeCompilationShim.configureCompileTask(processSwiftSourcesTaskProvider: Provider) {
        val processSwiftSourcesTaskOutput = processSwiftSourcesTaskProvider.flatMap { it.output }

        val compileTaskOutputFileProvider = compileTaskOutputFileProvider

        compileTaskProvider.configure {
            inputs.files(processSwiftSourcesTaskOutput)

            doLastOptimized {
                copySwiftSourcesToKlib(compileTaskOutputFileProvider.get(), processSwiftSourcesTaskOutput)
            }
        }
    }

    private fun copySwiftSourcesToKlib(klib: File, processSwiftSourcesTaskOutput: Provider) {
        val swiftSourcesDirectory = processSwiftSourcesTaskOutput.get()

        if (!swiftSourcesDirectory.exists()) {
            return
        }

        klib.writeToZip { fileSystem ->
            val klibSwiftSourcesDirectory = fileSystem.getPath("/$KLIB_SKIE_SWIFT_DIRECTORY")

            swiftSourcesDirectory.toPath().syncDirectoryContentIfDifferent(klibSwiftSourcesDirectory)
        }
    }

    // TODO Allow users to customize the swift source directory path
    private val KotlinSourceSetShim.swiftSourceDirectory: String
        get() = "src/$name/swift"
}