ck.wirespec.plugin.gradle.gradle.0.9.28.source-code.CustomWirespecTask.kt Maven / Gradle / Ivy

Go to download
package community.flock.wirespec.plugin.gradle

import community.flock.wirespec.compiler.core.emit.common.Emitter
import community.flock.wirespec.plugin.FilesContent
import community.flock.wirespec.plugin.PackageName
import community.flock.wirespec.plugin.compile
import community.flock.wirespec.plugin.toDirectory
import org.gradle.api.file.DirectoryProperty
import org.gradle.api.provider.Property
import org.gradle.api.tasks.Input
import org.gradle.api.tasks.InputDirectory
import org.gradle.api.tasks.Internal
import org.gradle.api.tasks.Optional
import org.gradle.api.tasks.TaskAction
import org.gradle.api.tasks.options.Option

abstract class CustomWirespecTask : BaseWirespecTask() {

    @get:InputDirectory
    @get:Option(option = "input", description = "input directory")
    abstract val input: DirectoryProperty

    @get:Input
    @get:Option(option = "emitter", description = "emitter")
    abstract val emitter: Property>

    @get:Optional
    @get:Input
    @get:Option(option = "shared", description = "shared code")
    abstract val shared: Property

    @get:Optional
    @get:Input
    @get:Option(option = "extension", description = "file extension")
    abstract val extension: Property

    @Internal
    protected fun getFilesContent(): FilesContent = input.asFileTree
        .map { it.name.split(".").first() to it.readText(Charsets.UTF_8) }
        .map { (name, reader) -> name to reader }

    @TaskAction
    fun custom() {
        val ext = extension.get()
        val defaultPkg = PackageName("community.flock.wirespec")
        val emitterPkg = packageName.map { PackageName(it) }.getOrElse(defaultPkg)
        val emitter: Emitter = try {
            emitter
                .map { it.getDeclaredConstructor().newInstance() }
                .get() as Emitter
        } catch (e: Exception) {
            logger.error("Cannot create instance of emitter: ${emitter.get().simpleName}", e)
            throw e
        }

        getFilesContent()
            .compile(wirespecLogger, emitter)
            .also {
                output.dir(defaultPkg.toDirectory()).get().asFile.apply {
                    mkdirs()
                    resolve("Wirespec.$ext").writeText(shared.get())
                }
            }
            .onEach { (name, result) ->
                output.dir(emitterPkg.toDirectory()).get().asFile.apply {
                    mkdirs()
                    resolve("${name}.$ext").writeText(result)
                }
            }
    }
}