//> using scala "3.3"
//> using dep io.getquill::quill-sql::4.8.0


import io.getquill.*
import io.getquill.ast.*

@main
def main = 
  case class Custom(name: String)
  val names = Seq("a", "b")

  val ctx = new SqlMirrorContext(MirrorSqlDialect, Literal)
  import ctx.*

  def createNameFilters(names: Seq[String])(p: Quoted[Custom]) =
    val nameFilters = names.map(name => quote(p.name == lift(name)))
    // val nameFilters = names.map(name => quote(p.name.like(lift(name))))
    val uuids = names.map(_ => java.util.UUID.randomUUID.toString)

    // this part should be way more robust, a better way of doing that would be to:
    // 1. retrieve all the current `lifts` (filters.map(_.lifts))
    // 2. keep a map of old uuid => new uuid
    // 3. Traverse the AST and replace old uuid by new uuid
    val filters: Seq[Quoted[Boolean]] = nameFilters.zip(uuids).map: (filter, uuid) =>
      filter.copy(
        ast = filter.ast match 
          case infix @ Infix(_, _ @head :: (last: ScalarTag) :: Nil , _, _, _) =>
            infix.copy(params = head :: last.copy(uid = uuid) :: Nil)
          case bop @ BinaryOperation(_, _ , op: ScalarTag) =>
            bop.copy(b = op.copy(uid = uuid))
          case other => other
        ,
        lifts = filter.lifts.map:
          case lift: EagerPlanter[_, _, _] => lift.copy(uid = uuid)
          case other => other
      )

    filters.reduce((l, r) => quote(l || r))

  val filters = createNameFilters(names)
  val query = dynamicQuery[Custom].filter(p => filters(p))

  println(ctx.translate(query))
  // # SELECT v0.name FROM Custom v0 WHERE v0.name = 'a' OR v0.name = 'b'