sangria.validation.rules.ScalarLeafs.scala Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of sangria_2.11 Show documentation

Scala GraphQL implementation

There is a newer version: 2.0.1

package sangria.validation.rules


import sangria.ast
import sangria.ast.AstVisitorCommand
import sangria.renderer.SchemaRenderer
import sangria.schema.LeafType
import sangria.validation._

/**
 * Scalar leafs
 *
 * A GraphQL document is valid only if all leaf fields (fields without
 * sub selections) are of scalar or enum types.
 */
class ScalarLeafs extends ValidationRule {
  override def visitor(ctx: ValidationContext) = new AstValidatingVisitor {
    override val onEnter: ValidationVisit = {
      case ast.Field(_, name, _, _, sels, _, _, pos) =>
        ctx.typeInfo.tpe match {
          case Some(fieldType) =>
            fieldType.namedType match {
              case tpe if tpe.isInstanceOf[LeafType] && sels.nonEmpty =>
                Left(Vector(NoSubselectionAllowedViolation(name, SchemaRenderer.renderTypeName(tpe, true), ctx.sourceMapper, pos.toList)))
              case tpe if !tpe.isInstanceOf[LeafType] && sels.isEmpty =>
                Left(Vector(RequiredSubselectionViolation(name, SchemaRenderer.renderTypeName(fieldType, false), ctx.sourceMapper, pos.toList)))
              case _ => AstVisitorCommand.RightContinue
            }
          case None => AstVisitorCommand.RightContinue
        }
    }
  }
}