calasql-query_2.13.0.1.9.source-code.TableMacro.scala Maven / Gradle / Ivy

Go to download
package scalasql.query

import scalasql.core.Sc
import scala.language.experimental.macros

object TableMacros {
  type Expr[T] = scalasql.core.Expr[T]
  def applyImpl[V[_[_]]](
      c: scala.reflect.macros.blackbox.Context
  )(implicit caseClassType: c.WeakTypeTag[V[Any]]): c.Expr[Table.Metadata[V]] = {
    import c.universe._

    val tableRef = TermName(c.freshName("tableRef"))
    val constructor = weakTypeOf[V[Any]].members.find(_.isConstructor).head
    val constructorParameters = constructor.info.paramLists.head

    def isTypeParamType(param: Symbol) = {
      param.info.typeSymbol.toString != caseClassType.tpe.typeParams.head.toString
    }

    val columnParams = for (param <- constructorParameters) yield {
      val name = param.name

      if (isTypeParamType(param)) {
        q"implicitly[scalasql.Table.ImplicitMetadata[${param.info.typeSymbol}]].value.vExpr($tableRef, dialect)"
      } else {
        q"""
          new _root_.scalasql.Column[${param.info.typeArgs.head}](
            $tableRef,
            _root_.scalasql.Table.columnNameOverride($tableRef.value)(${name.toString})
          )
        """
      }
    }

    def subParam(paramInfo: Type, tpe: Type) = {
      paramInfo.substituteTypes(
        List(constructor.info.resultType.typeArgs.head.typeSymbol),
        List(
          tpe
            .asInstanceOf[ExistentialType]
            .underlying
            .asInstanceOf[TypeRef]
            .sym
            .info
        )
      )
    }

    val queryables = for (param <- constructorParameters) yield {
      val tpe = subParam(param.info, typeOf[Sc[_]])
      val tpe2 = subParam(param.info, typeOf[TableMacros.Expr[_]])
      q"implicitly[_root_.scalasql.Queryable.Row[$tpe2, $tpe]]"
    }

    val constructParams = for ((param, i) <- constructorParameters.zipWithIndex) yield {
      val tpe = subParam(param.info, typeOf[Sc[_]])
      val tpe2 = subParam(param.info, typeOf[TableMacros.Expr[_]])
      q"queryable[$tpe2, $tpe]($i).construct(args): _root_.scalasql.Sc[$tpe]"
    }

    val deconstructParams = for ((param, i) <- constructorParameters.zipWithIndex) yield {
      val tpe = subParam(param.info, typeOf[Sc[_]])
      val tpe2 = subParam(param.info, typeOf[TableMacros.Expr[_]])
      q"queryable[$tpe2, $tpe]($i).deconstruct(r.${TermName(param.name.toString)})"
    }

    val flattenLists = for (param <- constructorParameters) yield {
      if (isTypeParamType(param)) {
        q"implicitly[scalasql.Table.ImplicitMetadata[${param.info.typeSymbol}]].value.walkLabels0()"
      } else {
        val name = param.name
        q"_root_.scala.List(${name.toString})"
      }
    }

    val flattenExprs = for ((param, i) <- constructorParameters.zipWithIndex) yield {
      val tpe = subParam(param.info, typeOf[Sc[_]])
      val tpe2 = subParam(param.info, typeOf[TableMacros.Expr[_]])
      q"queryable[$tpe2, $tpe]($i).walkExprs(table.${TermName(param.name.toString)})"
    }

    import compat._
    val typeRef = caseClassType.tpe.resultType.asInstanceOf[TypeRef]
    val exprRef = TypeRef(
      pre = typeRef.pre,
      sym = typeRef.sym,
      args = weakTypeOf[V[TableMacros.Expr]].typeArgs
    )
    val idRef = TypeRef(
      pre = typeRef.pre,
      sym = typeRef.sym,
      args = weakTypeOf[V[Sc]].typeArgs
    )
    c.Expr[Table.Metadata[V]](q"""{

    new _root_.scalasql.query.Table.Metadata(
      (dialect, n) => {
        import dialect._;
        n match{ case ..${queryables.zipWithIndex.map { case (q, i) => cq"$i => $q" }} }
      },
      () => ${flattenLists.reduceLeft((l, r) => q"$l ++ $r")},
      (walkLabels0, dialect, queryable) => {
        import dialect._

        new _root_.scalasql.query.Table.Internal.TableQueryable(
          walkLabels0,
          (table: $exprRef) => ${flattenExprs.reduceLeft((l, r) => q"$l ++ $r")},
          construct0 = (args: _root_.scalasql.Queryable.ResultSetIterator) => new $caseClassType(..$constructParams),
          deconstruct0 = (r: $idRef) => new $caseClassType(..$deconstructParams)
        )
      },
      ($tableRef: _root_.scalasql.query.TableRef, dialect, queryable) => {
        import dialect._

        new $caseClassType(..$columnParams)
      }
    )
    }""")
  }

}
trait TableMacros {
  implicit def initTableMetadata[V[_[_]]]: Table.Metadata[V] = macro TableMacros.applyImpl[V]
}