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

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sangria-core_2.13 Show documentation

Scala GraphQL implementation

The newest version!

package sangria.validation.rules

import sangria.ast.AstLocation

import scala.collection.mutable.{Map => MutableMap}

import sangria.ast
import sangria.ast.AstVisitorCommand
import sangria.validation._

/** Unique input field names
  *
  * A GraphQL input object value is only valid if all supplied fields are uniquely named.
  */
class UniqueInputFieldNames extends ValidationRule {
  override def visitor(ctx: ValidationContext) = new AstValidatingVisitor {
    val knownNameStack = ValidatorStack.empty[MutableMap[String, Option[AstLocation]]]
    var knownNames = MutableMap[String, Option[AstLocation]]()

    override val onEnter: ValidationVisit = {
      case ast.ObjectValue(_, _, _) =>
        knownNameStack.push(knownNames)
        knownNames = MutableMap[String, Option[AstLocation]]()

        AstVisitorCommand.RightContinue

      case ast.ObjectField(name, _, _, pos) =>
        if (knownNames contains name)
          Left(
            Vector(
              DuplicateInputFieldViolation(
                name,
                ctx.sourceMapper,
                knownNames(name).toList ++ pos.toList)))
        else {
          knownNames += name -> pos
          AstVisitorCommand.RightContinue
        }
    }

    override def onLeave: ValidationVisit = { case ast.ObjectValue(_, _, _) =>
      knownNames = knownNameStack.pop()

      AstVisitorCommand.RightContinue
    }
  }
}