Adding contramap to VendorExtensible

guardrail-dev · Oct 3, 2019 · df5d50c · df5d50c
1 parent eaca759
commit df5d50c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/modules/codegen/src/main/scala/com/twilio/guardrail/extract/VendorExtension.scala b/modules/codegen/src/main/scala/com/twilio/guardrail/extract/VendorExtension.scala
@@ -9,6 +9,12 @@ import io.swagger.v3.oas.models.security.SecurityScheme
 object VendorExtension {
   trait VendorExtensible[-F] {
     def extract[T](from: F, key: String)(implicit T: Extractable[T]): Option[T]
+    def contramap[B](f: B => F): VendorExtensible[B] = {
+      val chain = this
+      new VendorExtensible[B] {
+        def extract[T](from: B, key: String)(implicit T: Extractable[T]): Option[T] = chain.extract(f(from), key)
+      }
+    }
   }
 
   object VendorExtensible {

diff --git a/modules/codegen/src/main/scala/com/twilio/guardrail/generators/syntax/package.scala b/modules/codegen/src/main/scala/com/twilio/guardrail/generators/syntax/package.scala
@@ -16,11 +16,11 @@ package syntax {
       ("  " * indent) + value.toString().linesIterator.filterNot(_.contains(": null")).mkString("\n" + ("  " * indent))
   }
 
-  class RichCollection[A, Repr](xs: IterableLike[A, Repr]) {
+  class RichCollection[A, Repr](xs: IterableLike[A, Repr]){
     def distinctBy[B, That](f: A => B)(implicit cbf: CanBuildFrom[Repr, A, That]) = {
       val builder = cbf(xs.repr)
-      val i       = xs.iterator
-      var set     = Set[B]()
+      val i = xs.iterator
+      var set = Set[B]()
       while (i.hasNext) {
         val o = i.next
         val b = f(o)
@@ -88,9 +88,9 @@ package object syntax {
     def toDashedCase: String = splitParts(s).mkString("-")
   }
 
-  implicit def RichSchema    = new RichNotNullShower[Schema[_]](_)
+  implicit def RichSchema = new RichNotNullShower[Schema[_]](_)
   implicit def RichOperation = new RichNotNullShower[Operation](_)
-  implicit def RichPathItem  = new RichNotNullShower[PathItem](_)
+  implicit def RichPathItem = new RichNotNullShower[PathItem](_)
   implicit def RichParameter = new RichNotNullShower[Parameter](_)
 
   implicit def toRichCollection[A, Repr](xs: IterableLike[A, Repr]) = new RichCollection(xs)