com.driver733.mapstructfluent.Processor.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of processor Show documentation

Generator of fluent extension functions for mapstruct mappers

There is a newer version: 1.1.1

package com.driver733.mapstructfluent

import com.google.auto.service.AutoService
import com.squareup.kotlinpoet.FileSpec
import com.squareup.kotlinpoet.FunSpec
import java.io.File
import javax.annotation.processing.*
import javax.lang.model.SourceVersion
import javax.lang.model.element.Element
import javax.lang.model.element.ExecutableElement
import javax.lang.model.element.TypeElement

@AutoService(Processor::class)
@SupportedSourceVersion(SourceVersion.RELEASE_11)
@SupportedAnnotationTypes("org.mapstruct.Mapper")
@SupportedOptions(KAPT_KOTLIN_GENERATED_OPTION_NAME)
class MapstructFluentExtensionsAnnotationProcessor : AbstractProcessor() {

    override fun process(annotations: MutableSet?, roundEnv: RoundEnvironment?) =
            process(roundEnv, processingEnv, this::processMapper)

    private fun processMapper(method: ExecutableElement, mapper: Element, src: String?) {
        FileSpec.builder(
                processingEnv.elementUtils.getPackageOf(method).toString(),
                "${className(mapper).simpleName.capitalize()}$${method.simpleName.capitalize()}FluentExtensions"
        ).addImport(
                processingEnv.elementUtils.getPackageOf(method).toString(),
                className(mapper).simpleName
        ).addImport(
                "org.mapstruct.factory.Mappers", ""
        ).addFunction(
                FunSpec.builder("${method.simpleName}")
                        .receiver(method.parameters.first().kotlinType())
                        .addStatement(
                                "return Mappers.getMapper(${mapper.simpleName}::class.java).${method.simpleName}(this)"
                        )
                        .build()
        ).build().writeTo(
                File(src!!).apply { mkdir() }
        )
    }

}