fix #2117, fix #2118

Author: mathieuancelin

otoroshi/app/api/api.scala

@@ -79,7 +79,9 @@ case class ResourceVersion(
     schema.getOrElse {
       Try(
         Json.parse(
-          org.json4s.jackson.JsonMethods.pretty(fi.oph.scalaschema.SchemaFactory.default.createSchema(clazz).toJson)
+          org.json4s.jackson.JsonMethods.pretty(
+            fi.oph.scalaschema.SchemaFactory.default.createSchema(clazz).toJson
+          )
         )
       ) match {
         case Failure(e) => {
@@ -2471,7 +2473,7 @@ class GenericApiController(ApiAction: ApiAction, cc: ControllerComponents)(impli
   }
 
   def openapi() = Action { req =>
-    val body = otoroshi.api.OpenApi.generate(env, req.getQueryString("version"))
+    val body = otoroshi.api.OpenApi.generate(env, req.getQueryString("version"), req.getQueryString("extension_group"))
     Ok(body).as("application/json").withHeaders("Access-Control-Allow-Origin" -> "*")
   }
 }

otoroshi/app/api/openapi.scala

@@ -885,10 +885,10 @@ object OpenApi {
     finalSchemas
   }
 
-  def generate(env: Env, version: Option[String]): String = {
+  def generate(env: Env, version: Option[String], extensionGroup: Option[String]): String = {
     // TODO: missing live metrics api
     // TODO: missing analytics api
-    if (env.isDev) {
+    val finalDoc = if (env.isDev) {
       val additionalPathsFile = env.environment.resourceAsStream("/schemas/additionalPaths.json").get
       val additionalPathsRaw  = new String(additionalPathsFile.readAllBytes(), StandardCharsets.UTF_8)
       val additionalPathsJson = Json.parse(additionalPathsRaw).asObject
@@ -978,5 +978,52 @@ object OpenApi {
         }
       )
     }
+
+    extensionGroup match {
+      case None => finalDoc
+      case Some(group) => {
+        cache.getOrElseUpdate(
+          group, {
+
+            env.logger.info(s"Compute sub openapi.json document for group: '${group}'")
+
+            def findNeededComponents(js: JsValue): Set[String] = {
+              js.stringify.split("#/components/schemas/").map(str => str.split(""""""").head).toSet
+            }
+
+            def buildFilteredComponentsSchema(initialObj: JsValue, componentsSchema: JsObject): JsObject = {
+              var previousNeededComponents: Set[String] = Set.empty
+              var previousComponents: JsValue = Json.obj("paths" -> initialObj, "schemas" -> Json.obj())
+              var run = true
+              while (run) {
+                val neededComponents = findNeededComponents(previousComponents)
+                if (neededComponents.size == previousNeededComponents.size) {
+                  run = false
+                } else {
+                  previousComponents = Json.obj("paths" -> initialObj, "schemas" -> JsObject(componentsSchema.value.filter {
+                    case (key, _) if neededComponents.contains(key) => true
+                    case _ => false
+                  }))
+                  previousNeededComponents = neededComponents
+                }
+              }
+              previousComponents.select("schemas").asObject
+            }
+
+            val json = Json.parse(finalDoc).asObject
+            val paths = json.select("paths").asObject
+            val tags = json.select("tags").asArray
+            val components = json.select("components").asObject
+            val componentsSchema = json.select("components").select("schemas").asObject
+            val filteredPaths = JsObject(paths.value.filter(_._1.startsWith(s"/apis/${group}")))
+            val filteredTags = JsArray(tags.value.filter(_.select("name").asString.startsWith(group)))
+            val filteredComponentsSchema = buildFilteredComponentsSchema(filteredPaths, componentsSchema)
+            val customComponents = components ++ Json.obj("schemas" -> filteredComponentsSchema)
+            val customDoc = json ++ Json.obj("tags" -> filteredTags, "paths" -> filteredPaths, "components" -> customComponents)
+            customDoc.prettify
+          }
+        )
+      }
+    }
   }
 }