anorm.macros.ValueToStatement.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of anorm_3 Show documentation
anorm
The newest version!
/*
 * Copyright (C) from 2022 The Play Framework Contributors , 2011-2021 Lightbend Inc. 
 */

package anorm.macros

import scala.quoted.{ Expr, Quotes, Type }

import anorm.Macro.debugEnabled
import anorm.ToStatement

private[anorm] trait ValueToStatement {
  protected def valueToStatementImpl[A <: AnyVal](using q: Quotes, tpe: Type[A]): Expr[ToStatement[A]] = {

    import q.reflect.*

    val aTpr = TypeRepr.of[A](using tpe)
    val ctor = aTpr.typeSymbol.primaryConstructor

    ctor.paramSymss match {
      case List(v) :: Nil => {
        v.tree match {
          case vd: ValDef => {
            val tpr = vd.tpt.tpe

            tpr.asType match {
              case vtpe @ '[t] =>
                Expr.summon[ToStatement[t]] match {
                  case Some(ts) => {
                    def inner(a: Expr[A]) = {
                      val term = asTerm(a)

                      term
                        .select(term.symbol.fieldMember(v.name))
                        .asExprOf[t](using vtpe)
                    }

                    val generated = '{
                      new ToStatement[A] {
                        def set(s: java.sql.PreparedStatement, i: Int, a: A): Unit =
                          ${ ts }.set(s, i, ${ inner('a) })
                      }
                    }

                    if (debugEnabled) {
                      report.info(s"ToStatement for $tpe: ${generated.show}")
                    }

                    generated
                  }

                  case _ =>
                    report.errorAndAbort(
                      s"Instance not found: ${classOf[ToStatement[_]].getName}[${tpr.typeSymbol.fullName}]"
                    )
                }

            }
          }

          case _ =>
            report.errorAndAbort(
              s"Constructor parameter expected, found: ${v}"
            )
        }
      }

      case _ =>
        report.errorAndAbort(
          s"Cannot resolve value ToStatement for '${aTpr.typeSymbol.name}'"
        )
    }
  }
}