io.javalin.plugin.openapi.annotations.scanForAnnotations.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of javalin-openapi Show documentation

The newest version!

package io.javalin.plugin.openapi.annotations

import io.github.classgraph.ClassGraph
import io.javalin.plugin.openapi.dsl.OpenApiDocumentation

internal fun scanForAnnotations(packagePrefixes: Set): Map {
    // Load cached results if they exist
    cachedResults[packagePrefixes]?.also { return it }

    val documentationByPathInfo = mutableMapOf()

    val graph = ClassGraph()
            .enableAnnotationInfo()
            .enableMethodInfo()
            .ignoreClassVisibility()
            .whitelistPackages(*packagePrefixes.toTypedArray())
            .scan()
    val annotationName = OpenApi::class.java.canonicalName
    val annotations = graph.getClassesWithMethodAnnotation(annotationName)
            .flatMap { it.methodInfo }
            .filter { it.annotationInfo.any { it.name == annotationName } }
            .map { it.loadClassAndGetMethod() }
            .map { it.getAnnotation(OpenApi::class.java) }
            .filter { it.path != NULL_STRING }

    assert(annotations.none { a -> annotations.any { b -> a != b && a.pathInfo == b.pathInfo } }) {
        val duplicates = annotations.filter { a -> annotations.any { b -> a != b && a.pathInfo == b.pathInfo } }
                .map { it.pathInfo.path }
                .distinct()
                .joinToString(", ")
        throw IllegalStateException("Found multiple of the name path infos ("
                + duplicates + ") in packages: " + packagePrefixes.joinToString(", "))
    }

    annotations.forEach {
        documentationByPathInfo[it.pathInfo] = it.asOpenApiDocumentation()
    }

    // Cache results
    cachedResults[packagePrefixes] = documentationByPathInfo

    return documentationByPathInfo
}

private val cachedResults = mutableMapOf, Map>()