Exposing the set of members of emitted enumerations

ghostdogpr · Jun 23, 2021 · aed060f · aed060f
1 parent 218a036
commit aed060f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/tools/src/main/scala/caliban/tools/ClientWriter.scala b/tools/src/main/scala/caliban/tools/ClientWriter.scala
@@ -380,6 +380,10 @@ object ClientWriter {
       .map(v => s"""case ${typedef.name}.${safeEnumValue(v.enumValue)} => __EnumValue("${v.enumValue}")""")
       .mkString("\n")}
           }
+
+          val values: Vector[${enumName}] = Vector(${typedef.enumValuesDefinition
+      .map(v => safeEnumValue(v.enumValue))
+      .mkString(", ")})
         }
        """
   }

diff --git a/tools/src/test/scala/caliban/tools/ClientWriterSpec.scala b/tools/src/test/scala/caliban/tools/ClientWriterSpec.scala
@@ -236,6 +236,8 @@ object Client {
       case Origin.MARS  => __EnumValue("MARS")
       case Origin.BELT  => __EnumValue("BELT")
     }
+
+    val values: Vector[Origin] = Vector(EARTH, MARS, BELT)
   }
 
 }
@@ -289,6 +291,8 @@ object Client {
       case Origin.MARS  => __EnumValue("MARS")
       case Origin.BELT  => __EnumValue("BELT")
     }
+
+    val values: Vector[Origin] = Vector(EARTH, MARS, BELT)
   }
 
   case class Routes(origin: Origin, destinations: List[com.example.Destination] = Nil)
@@ -599,6 +603,8 @@ object Client {
       case Episode.JEDI    => __EnumValue("JEDI")
       case Episode.jedi_1  => __EnumValue("jedi")
     }
+
+    val values: Vector[Episode] = Vector(NEWHOPE, EMPIRE, JEDI, jedi_1)
   }
 
 }