sangria.ast.AstVisitor.scala Maven / Gradle / Ivy
package sangria.ast
import sangria.ast
import scala.util.control.Breaks._
object AstVisitor {
import ast.AstVisitorCommand._
def visitAst(
doc: AstNode,
onEnter: AstNode ⇒ AstVisitorCommand.Value = _ ⇒ Continue,
onLeave: AstNode ⇒ AstVisitorCommand.Value = _ ⇒ Continue): Unit = {
def breakOrSkip(cmd: AstVisitorCommand.Value) = cmd match {
case Break ⇒ break()
case Skip ⇒ false
case Continue ⇒ true
}
def loop(node: AstNode): Unit =
node match {
case n @ Document(defs, _, _) ⇒
if (breakOrSkip(onEnter(n))) {
defs.foreach(d ⇒ loop(d))
breakOrSkip(onLeave(n))
}
case n @ FragmentDefinition(_, cond, dirs, sels, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
loop(cond)
dirs.foreach(d ⇒ loop(d))
sels.foreach(s ⇒ loop(s))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ OperationDefinition(_, _, vars, dirs, sels, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
vars.foreach(d ⇒ loop(d))
dirs.foreach(d ⇒ loop(d))
sels.foreach(s ⇒ loop(s))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ VariableDefinition(_, tpe, default, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
loop(tpe)
default.foreach(d ⇒ loop(d))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ InlineFragment(cond, dirs, sels, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
cond.foreach(c ⇒ loop(c))
dirs.foreach(d ⇒ loop(d))
sels.foreach(s ⇒ loop(s))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ FragmentSpread(_, dirs, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
dirs.foreach(d ⇒ loop(d))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ NotNullType(ofType, _) ⇒
if (breakOrSkip(onEnter(n))) {
loop(ofType)
breakOrSkip(onLeave(n))
}
case n @ ListType(ofType, _) ⇒
if (breakOrSkip(onEnter(n))) {
loop(ofType)
breakOrSkip(onLeave(n))
}
case n @ Field(_, _, args, dirs, sels, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
args.foreach(d ⇒ loop(d))
dirs.foreach(d ⇒ loop(d))
sels.foreach(s ⇒ loop(s))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ Argument(_, v, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
loop(v)
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ ObjectField(_, v, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
loop(v)
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ Directive(_, args, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
args.foreach(d ⇒ loop(d))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ ListValue(vals, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
vals.foreach(v ⇒ loop(v))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ ObjectValue(fields, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
fields.foreach(f ⇒ loop(f))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ BigDecimalValue(_, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ BooleanValue(_, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ Comment(_, _) ⇒
if (breakOrSkip(onEnter(n))) {
breakOrSkip(onLeave(n))
}
case n @ VariableValue(_, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ EnumValue(_, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ NullValue(comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ NamedType(_, _) ⇒
if (breakOrSkip(onEnter(n))) {
breakOrSkip(onLeave(n))
}
case n @ StringValue(_, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ BigIntValue(_, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ IntValue(_, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ FloatValue(_, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
// IDL schema definition
case n @ ScalarTypeDefinition(_, dirs, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
dirs.foreach(d ⇒ loop(d))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ FieldDefinition(_, fieldType, args, dirs, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
loop(fieldType)
args.foreach(d ⇒ loop(d))
dirs.foreach(d ⇒ loop(d))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ InputValueDefinition(_, valueType, default, dirs, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
loop(valueType)
default.foreach(d ⇒ loop(d))
dirs.foreach(d ⇒ loop(d))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ ObjectTypeDefinition(_, interfaces, fields, dirs, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
interfaces.foreach(d ⇒ loop(d))
fields.foreach(d ⇒ loop(d))
dirs.foreach(d ⇒ loop(d))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ InterfaceTypeDefinition(_, fields, dirs, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
fields.foreach(d ⇒ loop(d))
dirs.foreach(d ⇒ loop(d))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ UnionTypeDefinition(_, types, dirs, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
types.foreach(d ⇒ loop(d))
dirs.foreach(d ⇒ loop(d))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ EnumTypeDefinition(_, values, dirs, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
values.foreach(d ⇒ loop(d))
dirs.foreach(d ⇒ loop(d))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ EnumValueDefinition(_, dirs, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
dirs.foreach(d ⇒ loop(d))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ InputObjectTypeDefinition(_, fields, dirs, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
fields.foreach(d ⇒ loop(d))
dirs.foreach(d ⇒ loop(d))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ TypeExtensionDefinition(definition, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
loop(definition)
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ DirectiveDefinition(_, args, locations, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
args.foreach(d ⇒ loop(d))
locations.foreach(d ⇒ loop(d))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ DirectiveLocation(_, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ SchemaDefinition(ops, dirs, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
ops.foreach(s ⇒ loop(s))
dirs.foreach(s ⇒ loop(s))
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
case n @ OperationTypeDefinition(_, tpe, comment, _) ⇒
if (breakOrSkip(onEnter(n))) {
loop(tpe)
comment.foreach(s ⇒ loop(s))
breakOrSkip(onLeave(n))
}
}
breakable {
loop(doc)
}
}
}
object AstVisitorCommand extends Enumeration {
val Skip, Continue, Break = Value
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy