fix: Allow other implementations of SchemaLoader (#1036)

ghostdogpr · Sep 8, 2021 · ad677b0 · ad677b0
1 parent ea3a017
commit ad677b0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tools/src/main/scala/caliban/tools/SchemaLoader.scala b/tools/src/main/scala/caliban/tools/SchemaLoader.scala
@@ -5,14 +5,17 @@ import caliban.parsing.Parser
 import caliban.parsing.adt.Document
 import zio.{ Task, UIO }
 
-sealed trait SchemaLoader {
+trait SchemaLoader {
   def load: Task[Document]
 }
 
 object SchemaLoader {
   case class FromCaliban private (api: GraphQL[_])                                          extends SchemaLoader {
     override def load: Task[Document] = UIO(api.toDocument)
   }
+  case class FromDocument private (doc: Document)                                           extends SchemaLoader {
+    override def load: Task[Document] = UIO(doc)
+  }
   case class FromFile private (path: String)                                                extends SchemaLoader {
     override def load: Task[Document] =
       Task(scala.io.Source.fromFile(path))
@@ -27,6 +30,7 @@ object SchemaLoader {
   }
 
   def fromCaliban[R](api: GraphQL[R]): SchemaLoader                                       = FromCaliban(api)
+  def fromDocument(doc: Document): SchemaLoader                                           = FromDocument(doc)
   def fromFile(path: String): SchemaLoader                                                = FromFile(path)
   def fromString(schema: String): SchemaLoader                                            = FromString(schema)
   def fromIntrospection(url: String, headers: Option[List[Options.Header]]): SchemaLoader =