io.getquill.sources.EncodeParams.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of quill-core_2.11 Show documentation

quill-core

There is a newer version: 4.6.0

package io.getquill.sources

import scala.reflect.macros.whitebox.Context

import io.getquill.ast._
import io.getquill.util.Messages._

object EncodeParams {

  def apply[S](c: Context)(
    params: collection.Map[Ident, (c.Type, c.Tree)],
    raw:    collection.Map[Ident, (c.Tree, c.Tree)]
  )(
    implicit
    s: c.WeakTypeTag[S]
  ): c.Tree =
    expand[S](c) {
      params.map {
        case (ident, (tpe, tree)) =>
          val encoder =
            Encoding.inferEncoder(c)(tpe)(s)
              .getOrElse(c.fail(s"Source doesn't know how do encode '$ident: $tpe'"))
          (ident, (encoder, tree))
      }.toMap ++ raw
    }

  private def expand[S](c: Context)(
    params: collection.Map[Ident, (c.Tree, c.Tree)]
  )(
    implicit
    s: c.WeakTypeTag[S]
  ): c.Tree = {
    import c.universe._
    val encoders =
      for ((ident, (encoder, tree)) <- params) yield {
        q"${ident.name} -> ((row: $s, index: Int) => $encoder(index, $tree, row))"
      }
    if (encoders.isEmpty)
      q"""
        (bindings: List[String]) =>
          (row: $s) => row  
      """
    else
      q"""
      {
        val bindingMap = collection.Map(..$encoders)
        (bindings: List[String]) =>
          (row: $s) =>
            bindings.foldLeft((row, 0)) {
              case ((row, index), binding) =>
                (bindingMap(binding)(row, index), index + 1)
            }._1
      }
      """
  }
}