se.ansman.dagger.auto.compiler.common.kapt.processing.KaptNode.kt Maven / Gradle / Ivy

Go to download

package se.ansman.dagger.auto.compiler.common.kapt.processing

import com.google.auto.common.MoreElements
import com.squareup.javapoet.AnnotationSpec
import com.squareup.javapoet.ClassName
import com.squareup.javapoet.TypeName
import se.ansman.dagger.auto.compiler.common.processing.Node
import javax.lang.model.element.Element

sealed class KaptNode : Node {
    abstract val resolver: KaptResolver
    override val annotations: List by lazy(LazyThreadSafetyMode.NONE) {
        node.annotationMirrors.map { (KaptAnnotationModel(it, resolver)) }
    }

    @Suppress("UnstableApiUsage")
    override val enclosingType: KaptClassDeclaration? by lazy(LazyThreadSafetyMode.NONE) {
        node.enclosingElement
            ?.takeIf { it.kind.isClass || it.kind.isInterface }
            ?.let(MoreElements::asType)
            ?.let { KaptClassDeclaration(it, resolver) }
    }
}