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

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.AbstractProcessor
import javax.annotation.processing.Processor
import javax.annotation.processing.RoundEnvironment
import javax.annotation.processing.SupportedAnnotationTypes
import javax.annotation.processing.SupportedOptions
import javax.annotation.processing.SupportedSourceVersion
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(), MapperMethodProcessor {

    private val genericProcessor = GenericProcessor()

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

    override fun fileSpecBuilder(mapper: Element) = fileSpecBuilder(processingEnv, mapper, "FluentExtensions")

    override fun process(fileSpecBuilder: FileSpec.Builder, method: ExecutableElement, mapper: Element, src: String?) {
        fileSpecBuilder
            .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() }
            )
    }
}