Improve JsonSchema.Object.addAll

zio · Aug 23, 2024 · 5910022 · 5910022
1 parent 0a00532
commit 5910022
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 8 deletions.
diff --git a/zio-http-gen/src/main/scala/zio/http/gen/openapi/EndpointGen.scala b/zio-http-gen/src/main/scala/zio/http/gen/openapi/EndpointGen.scala
@@ -1007,11 +1007,9 @@ final case class EndpointGen(config: Config) {
       case JsonSchema.ArrayType(None, _, _)         => None
       case JsonSchema.ArrayType(Some(schema), _, _) =>
         schemaToCode(schema, openAPI, name, annotations)
-      case JsonSchema.Object(properties, additionalProperties, _)
-          if properties.nonEmpty && additionalProperties.isRight =>
-        // Can't be an object and a map at the same time
+      case obj: JsonSchema.Object if obj.isInvalid  =>
         throw new Exception("Object with properties and additionalProperties is not supported")
-      case obj @ JsonSchema.Object(properties, additionalProperties, _) if additionalProperties.isLeft =>
+      case obj @ JsonSchema.Object(properties, _, _) if obj.isClosedDictionary =>
         val unvalidatedFields = fieldsOfObject(openAPI, annotations)(obj)
         val fields            = validateFields(unvalidatedFields)
         val nested            =
@@ -1044,10 +1042,10 @@ final case class EndpointGen(config: Config) {
             enums = Nil,
           ),
         )
-      case JsonSchema.Object(_, _, _)                                                                  =>
-        // properties.isEmpty && additionalProperties.isRight
+      case JsonSchema.Object(_, _, _)                                          =>
+        // if obt.isOpenDictionary
         throw new IllegalArgumentException("Top-level maps are not supported")
-      case JsonSchema.Enum(enums)                                                                      =>
+      case JsonSchema.Enum(enums)                                              =>
         Some(
           Code.File(
             List("component", name.capitalize + ".scala"),

diff --git a/zio-http/shared/src/main/scala/zio/http/endpoint/openapi/JsonSchema.scala b/zio-http/shared/src/main/scala/zio/http/endpoint/openapi/JsonSchema.scala
@@ -1212,10 +1212,30 @@ object JsonSchema {
     additionalProperties: Either[Boolean, JsonSchema],
     required: Chunk[java.lang.String],
   ) extends JsonSchema {
+
+    /**
+     * This Object represents an "open dictionary", aka a Map
+     *
+     * See: https://github.com/zio/zio-http/issues/3048#issuecomment-2306291192
+     */
+    def isOpenDictionary: Boolean = properties.isEmpty && additionalProperties.isRight
+
+    /**
+     * This Object represents a "closed dictionary", aka a case class
+     *
+     * See: https://github.com/zio/zio-http/issues/3048#issuecomment-2306291192
+     */
+    def isClosedDictionary: Boolean = additionalProperties.isLeft
+
+    /**
+     * Can't represent a case class and a Map at the same time
+     */
+    def isInvalid: Boolean = properties.nonEmpty && additionalProperties.isRight
+
     def addAll(value: Chunk[(java.lang.String, JsonSchema)]): Object =
       value.foldLeft(this) { case (obj, (name, schema)) =>
         schema match {
-          case Object(properties, additionalProperties, required) =>
+          case thatObj @ Object(properties, additionalProperties, required) if thatObj.isClosedDictionary =>
             obj.copy(
               properties = obj.properties ++ properties,
               additionalProperties = combineAdditionalProperties(obj.additionalProperties, additionalProperties),