org.http4k.connect.plugin.Http4kConnectAdapterVisitor.kt Maven / Gradle / Ivy
package org.http4k.connect.plugin
import com.google.devtools.ksp.getAllSuperTypes
import com.google.devtools.ksp.symbol.KSAnnotated
import com.google.devtools.ksp.symbol.KSClassDeclaration
import com.google.devtools.ksp.symbol.KSNode
import com.google.devtools.ksp.symbol.KSType
import com.google.devtools.ksp.symbol.KSTypeReference
import com.google.devtools.ksp.visitor.KSEmptyVisitor
import com.squareup.kotlinpoet.FileSpec
import com.squareup.kotlinpoet.ksp.toClassName
import java.util.Locale
class Http4kConnectAdapterVisitor(private val log: (Any?) -> Unit) :
KSEmptyVisitor, List>() {
override fun visitClassDeclaration(classDeclaration: KSClassDeclaration, data: List) =
classDeclaration.http4kConnectActionTypes.map(KSTypeReference::resolve).map {
log(
"Processing http4k-connect adapter: " + classDeclaration.simpleName.asString() +
" with action type: $it"
)
FileSpec.builder(
it.toClassName().packageName,
classDeclaration.simpleName.asString().lowercase(Locale.getDefault()) + "Extensions"
)
.apply {
data.filterForActionsOf(it)
.flatMap { it.accept(Http4kConnectActionVisitor(log), classDeclaration) }
.forEach(::addFunction)
}
.build()
}.toList()
override fun defaultHandler(node: KSNode, data: List) = error("unsupported")
}
internal val KSClassDeclaration.http4kConnectActionTypes
get() = getAllFunctions()
.filter { it.simpleName.getShortName() == "invoke" }
.map { it.parameters.first().type }
fun List.filterForActionsOf(actionType: KSType) =
filterIsInstance()
.filter {
it.getAllSuperTypes().map(KSType::starProjection)
.contains(actionType.starProjection())
}