Adding support for dynamic additionalProperties

guardrail-dev · Oct 24, 2022 · cc6d79f · cc6d79f
1 parent 9458641
commit cc6d79f
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/modules/core/src/main/scala/dev/guardrail/generators/ProtocolGenerator.scala b/modules/core/src/main/scala/dev/guardrail/generators/ProtocolGenerator.scala
@@ -429,6 +429,16 @@ object ProtocolGenerator {
           .orRefine { case s: StringSchema if Option(s.getEnum).map(_.asScala).exists(_.nonEmpty) => s }(s =>
             fromEnum(nestedClassName.last, s, dtoPackage, components).map(Option(_))
           )
+          .orRefine { case m: MapSchema => m } { m =>
+            m.downField("additionalProperties", _.getAdditionalProperties)
+              .indexedDistribute
+              .flatTraverse(
+                _.refine { case x: ObjectSchema => x }(obj =>
+                  fromModel(nestedClassName, obj, List.empty, concreteTypes, definitions, dtoPackage, supportPackage, defaultPropertyRequirement, components)
+                ).toOption.sequence
+              )
+              .widen
+          }
           .getOrElse(Option.empty[Either[String, NestedProtocolElems[L]]].pure[F])
       } yield defn