sangria.macros.AstLiftable.scala Maven / Gradle / Ivy

Go to download
package sangria.macros

import sangria.ast._

import scala.reflect.api.Universe
import scala.reflect.macros.blackbox

/**
 * Implements the Liftable type class for AstNode, so we can use them in
 * quasiquotes.
 */
trait AstLiftable {
  val universe: Universe

  import universe._

  implicit def liftSeq[T : Liftable]: Liftable[Seq[T]] = Liftable { seq ⇒
    q"_root_.scala.collection.immutable.Vector(..$seq)"
  }

  implicit def liftPosition: Liftable[org.parboiled2.Position] = Liftable {
    case org.parboiled2.Position(i, l, c) ⇒
      q"_root_.org.parboiled2.Position($i, $l, $c)"
  }

  implicit def liftOperationType: Liftable[OperationType] = Liftable {
    case OperationType.Query ⇒ q"_root_.sangria.ast.OperationType.Query"
    case OperationType.Mutation ⇒ q"_root_.sangria.ast.OperationType.Mutation"
    case OperationType.Subscription ⇒ q"_root_.sangria.ast.OperationType.Subscription"
  }

  implicit def liftType[T <: sangria.ast.Type]: Liftable[T] = Liftable {
    case NamedType(n, p) ⇒ q"_root_.sangria.ast.NamedType($n, $p)"
    case NotNullType(o, p) ⇒ q"_root_.sangria.ast.NotNullType($o, $p)"
    case ListType(o, p) ⇒ q"_root_.sangria.ast.ListType($o, $p)"
  }

  implicit def liftComment: Liftable[Comment] = Liftable {
    case Comment(l, p) ⇒
      q"_root_.sangria.ast.Comment($l, $p)"
  }

  implicit def liftVarDef: Liftable[VariableDefinition] = Liftable {
    case VariableDefinition(n, t, d, c, p) ⇒
      q"_root_.sangria.ast.VariableDefinition($n, $t, $d, $c, $p)"
  }

  implicit def liftInpValDef: Liftable[InputValueDefinition] = Liftable {
    case InputValueDefinition(n, v, de, di, c, p) ⇒
      q"_root_.sangria.ast.InputValueDefinition($n, $v, $de, $di, $c, $p)"
  }

  implicit def liftInpOpTpeDef: Liftable[OperationTypeDefinition] = Liftable {
    case OperationTypeDefinition(o, t, c, p) ⇒
      q"_root_.sangria.ast.OperationTypeDefinition($o, $t, $c, $p)"
  }

  implicit def liftEnumValDef: Liftable[EnumValueDefinition] = Liftable {
    case EnumValueDefinition(n, d, c, p) ⇒
      q"_root_.sangria.ast.EnumValueDefinition($n, $d, $c, $p)"
  }

  implicit def liftFieldDef: Liftable[FieldDefinition] = Liftable {
    case FieldDefinition(n, f, a, d, c, p) ⇒
      q"_root_.sangria.ast.FieldDefinition($n, $f, $a, $d, $c, $p)"
  }

  implicit def liftDirLocDef: Liftable[DirectiveLocation] = Liftable {
    case DirectiveLocation(n, c, p) ⇒
      q"_root_.sangria.ast.DirectiveLocation($n, $c, $p)"
  }

  implicit def liftDefinition[T <: Definition]: Liftable[T] = Liftable {
    case OperationDefinition(o, n, v, d, s, c, p) ⇒
      q"_root_.sangria.ast.OperationDefinition($o, $n, $v, $d, $s, $c, $p)"
    case FragmentDefinition(n, t, d, s, c, p) ⇒
      q"_root_.sangria.ast.FragmentDefinition($n, $t, $d, $s, $c, $p)"

    case DirectiveDefinition(n, a, l, c, p) ⇒
      q"_root_.sangria.ast.DirectiveDefinition($n, $a, $l, $c, $p)"
    case SchemaDefinition(o, d, c, p) ⇒
      q"_root_.sangria.ast.SchemaDefinition($o, $d, $c, $p)"
    case TypeExtensionDefinition(d, c, p) ⇒
      q"_root_.sangria.ast.TypeExtensionDefinition($d, $c, $p)"

    case EnumTypeDefinition(n, v, d, c, p) ⇒
      q"_root_.sangria.ast.EnumTypeDefinition($n, $v, $d, $c, $p)"
    case InputObjectTypeDefinition(n, f, d, c, p) ⇒
      q"_root_.sangria.ast.InputObjectTypeDefinition($n, $f, $d, $c, $p)"
    case InterfaceTypeDefinition(n, f, d, c, p) ⇒
      q"_root_.sangria.ast.InterfaceTypeDefinition($n, $f, $d, $c, $p)"
    case ObjectTypeDefinition(n, i, f, d, c, p) ⇒
      q"_root_.sangria.ast.ObjectTypeDefinition($n, $i, $f, $d, $c, $p)"
    case ScalarTypeDefinition(n, d, c, p) ⇒
      q"_root_.sangria.ast.ScalarTypeDefinition($n, $d, $c, $p)"
    case UnionTypeDefinition(n, t, d, c, p) ⇒
      q"_root_.sangria.ast.UnionTypeDefinition($n, $t, $d, $c, $p)"
  }

  implicit def liftNamedValue[T <: NameValue]: Liftable[T] = Liftable {
    case Argument(n, v, c, p) ⇒ q"_root_.sangria.ast.Argument($n, $v, $c, $p)"
    case ObjectField(n, v, c, p) ⇒ q"_root_.sangria.ast.ObjectField($n, $v, $c, $p)"
  }

  implicit def liftValue[T <: sangria.ast.Value]: Liftable[T] = Liftable {
    case IntValue(v, c, p) ⇒ q"_root_.sangria.ast.IntValue($v, $c, $p)"
    case FloatValue(v, c, p) ⇒ q"_root_.sangria.ast.FloatValue($v, $c, $p)"
    case StringValue(v, c, p) ⇒ q"_root_.sangria.ast.StringValue($v, $c, $p)"
    case BooleanValue(v, c, p) ⇒ q"_root_.sangria.ast.BooleanValue($v, $c, $p)"
    case NullValue(c, p) ⇒ q"_root_.sangria.ast.NullValue($c, $p)"
    case EnumValue(v, c, p) ⇒ q"_root_.sangria.ast.EnumValue($v, $c, $p)"
    case ListValue(v, c, p) ⇒ q"_root_.sangria.ast.ListValue($v, $c, $p)"
    case ObjectValue(f, c, p) ⇒ q"_root_.sangria.ast.ObjectValue($f, $c, $p)"
    case VariableValue(n, c, p) ⇒ q"_root_.sangria.ast.VariableValue($n, $c, $p)"
    case BigIntValue(v, c, p) ⇒
      q"_root_.sangria.ast.BigIntValue(_root_.scala.math.BigInt(${v.toByteArray}), $c, $p)"
    case sangria.ast.BigDecimalValue(v, c, p) ⇒
      q"_root_.sangria.ast.BigDecimalValue(_root_.scala.math.BigDecimal(${v.toString()}), $c, $p)"
  }

  implicit def directive: Liftable[sangria.ast.Directive] = Liftable {
    case Directive(n, a, c, p) ⇒ q"_root_.sangria.ast.Directive($n, $a, $c, $p)"
  }

  implicit def selection[T <: Selection]: Liftable[T] = Liftable {
    case Field(a, n, arg, d, s, c, p) ⇒
      q"_root_.sangria.ast.Field($a, $n, $arg, $d, $s, $c, $p)"
    case FragmentSpread(n, d, c, p) ⇒
      q"_root_.sangria.ast.FragmentSpread($n, $d, $c, $p)"
    case InlineFragment(t, d, s, c, p) ⇒
      q"_root_.sangria.ast.InlineFragment($t, $d, $s, $c, $p)"
  }

  implicit def liftDocument: Liftable[Document] = Liftable {
    case doc @ Document(d, p, _) ⇒ q"_root_.sangria.ast.Document($d, $p, _root_.scala.Some(new _root_.sangria.parser.Parboiled2SourceMapper(_root_.org.parboiled2.ParserInput(${doc.source.get}))))"
  }
}

trait MacroAstLiftable extends AstLiftable {
  val c: blackbox.Context
  val universe: c.universe.type = c.universe
}