io.getquill.context.sql.SqlContextMacro.scala Maven / Gradle / Ivy

Go to download
package io.getquill.context.sql

import scala.reflect.macros.whitebox.{ Context => MacroContext }
import io.getquill.ast._
import io.getquill.NamingStrategy
import io.getquill.context.ContextMacro
import io.getquill.context.sql.idiom.SqlIdiom
import io.getquill.util.Messages.RichContext
import io.getquill.quotation.IsDynamic
import io.getquill.LoadNaming
import io.getquill.util.LoadObject

class SqlContextMacro(val c: MacroContext) extends ContextMacro {
  import c.universe.{ Try => _, Literal => _, Ident => _, _ }

  override protected def prepare(ast: Ast, params: List[Ident]) = {
    if (!IsDynamic(ast)) {
      implicit val (d, n) = dialectAndNamingStatic
      val (sql, idents, generated) = Prepare(ast, params)
      c.info(sql)
      probe(sql, d)
      q"($sql, $idents, $generated)"
    } else {
      c.info("Dynamic query")
      q"""
      {
        implicit val (d, n) = $dialectAndNamingDynamic
        io.getquill.context.sql.Prepare($ast, $params)
      }
      """
    }
  }

  private def probe(sql: String, d: SqlIdiom) =
    probeQuery[SqlContext[SqlIdiom, NamingStrategy, Any, Any]](_.probe(d.prepare(sql)))

  private def dialectAndNaming = {
    val (idiom :: n :: _) =
      c.prefix.actualType
        .baseType(c.weakTypeOf[SqlContext[SqlIdiom, NamingStrategy, Any, Any]].typeSymbol)
        .typeArgs
    (idiom, n)
  }

  private def dialectAndNamingDynamic = {
    val (idiom, naming) = dialectAndNaming
    q"(${idiom.typeSymbol.companion}, ${LoadNaming.dynamic(c)(naming)})"
  }

  private def dialectAndNamingStatic = {
    val (idiom, naming) = dialectAndNaming
    (LoadObject[SqlIdiom](c)(idiom), LoadNaming.static(c)(naming))
  }
}