model.Model.kt Maven / Gradle / Ivy

Go to download

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

"Annotation Processing and Code Generation Utils"

The newest version!

package de.peekandpoke.ultra.meta.model

import com.squareup.kotlinpoet.ClassName
import de.peekandpoke.ultra.meta.GenericUsages
import de.peekandpoke.ultra.meta.ProcessorUtils
import javax.lang.model.element.TypeElement

fun ProcessorUtils.model(types: List) = Model(ctx, types)

class Model(
    override val ctx: ProcessorUtils.Context,
    types: List
) : ProcessorUtils, Iterable {

    val types = types.map { it.toMType() }

    private val genericUsages = GenericUsages(this, types)

    fun getGenericUsages(className: ClassName) = genericUsages.get(className)

    private fun TypeElement.toMType(): MType = MType(
        this@Model,
        this,
        this.asTypeName()
    )

    fun getDirectChildTypes(parent: MType): List {
        return types.filter {
            it.directSuperTypes.any { superType -> superType == parent }
        }
    }

    override fun iterator(): Iterator = types.iterator()
}