union types in sangria

unions.scala

object ProfileRef {
   val inactiveDefinition = ObjectType(
    "InactiveProfileRef",
    "An inactive profile",
    fields = fields[UserContext, JsObject](
      Field("id", IntType, None, resolve = ctx => (ctx.value \ "id").as[Int])
    )
  )

  val activeDefinition = ObjectType(
    "ActiveProfileRef",
    "An active profile",
    fields = fields[UserContext, JsObject](
      Field("id", IntType, None, resolve = ctx => (ctx.value \ "id").as[Int])
    )
  )

  val definition = UnionType(
    "ProfileRef",
    Some("The profile of a user"),
    List(inactiveDefinition, activeDefinition)
  )
}

object SchemaDefinition {
  val ID = Argument("id", IntType, description = "id of the character")

  val Query = ObjectType(
    "Query", fields[UserContext, Unit](
      Field("profileRef", ProfileRef.definition,
        arguments = List(ID),
        resolve = (ctx) ⇒ {
          JsObject(Seq(
            "id" -> JsNumber(ctx.arg(ID))
          ))
       })
    ))

  val schema = Schema(Query)
}