@@ -30,7 +30,6 @@ import io.github.smiley4.ktorswaggerui.builder.route.RouteMeta
3030import io.github.smiley4.ktorswaggerui.builder.schema.SchemaContext
3131import io.github.smiley4.ktorswaggerui.builder.schema.SchemaContextImpl
3232import io.github.smiley4.ktorswaggerui.data.PluginConfigData
33- import io.github.smiley4.ktorswaggerui.data.TypeDescriptor
3433import io.github.smiley4.ktorswaggerui.dsl.config.PluginConfigDsl
3534import io.github.smiley4.ktorswaggerui.routing.ApiSpec
3635import io.ktor.server.application.Application
@@ -83,12 +82,12 @@ private fun buildOpenApiSpecs(config: PluginConfigData, routes: List<RouteMeta>)
8382 return buildMap {
8483 routesBySpec.forEach { (specName, routes) ->
8584 val specConfig = config.specConfigs[specName] ? : config
86- this [specName] = buildOpenApiSpec(specConfig, routes)
85+ this [specName] = buildOpenApiSpec(specName, specConfig, routes)
8786 }
8887 }
8988}
9089
91- private fun buildOpenApiSpec (pluginConfig : PluginConfigData , routes : List <RouteMeta >): String {
90+ private fun buildOpenApiSpec (specName : String , pluginConfig : PluginConfigData , routes : List <RouteMeta >): String {
9291 return try {
9392 val schemaContext = SchemaContextImpl (pluginConfig.schemaConfig).also {
9493 it.addGlobal(pluginConfig.schemaConfig)
@@ -99,7 +98,7 @@ private fun buildOpenApiSpec(pluginConfig: PluginConfigData, routes: List<RouteM
9998 it.add(routes)
10099 }
101100 val openApi = builder(pluginConfig, schemaContext, exampleContext).build(routes)
102- pluginConfig.postBuild?.invoke (openApi)
101+ pluginConfig.postBuild?.let { it (openApi, specName) }
103102 Json .pretty(openApi)
104103 } catch (e: Exception ) {
105104 logger.error(" Error during openapi-generation" , e)
0 commit comments