com.avsystem.commons.mongo.BsonRef.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of commons-mongo_2.13 Show documentation

AVSystem commons library for Scala

There is a newer version: 2.22.0

package com.avsystem.commons
package mongo

import com.avsystem.commons.mongo.core.ops.{BsonRefFiltering, BsonRefIterableFiltering, BsonRefIterableUpdating, BsonRefSorting, BsonRefUpdating}
import com.avsystem.commons.serialization.RawRef.Field
import com.avsystem.commons.serialization.{GenCodec, GenRef}

case class BsonRef[S, T](path: String, codec: GenCodec[T], getter: S => T) {
  def apply(s: S): T = getter(s)

  def andThen[T0](other: BsonRef[T, T0]): BsonRef[S, T0] = {
    val newPath = List(path, other.path).filter(_.nonEmpty).mkString(BsonRef.BsonKeySeparator)
    BsonRef(newPath, other.codec, getter andThen other.getter)
  }

  def compose[S0](other: BsonRef[S0, S]): BsonRef[S0, T] =
    other andThen this
}
object BsonRef {
  val BsonKeySeparator = "."

  def identity[S](implicit codec: GenCodec[S]): BsonRef[S, S] = BsonRef("", codec, s => s)
  def create[S]: Creator[S] = new Creator[S] {}

  trait Creator[S] {
    type Ref[T] = BsonRef[S, T]

    def ref[T](fun: S => T): BsonRef[S, T] = macro macros.serialization.BsonRefMacros.bsonRef[S, T]
  }

  def apply[S, T](genRef: GenRef[S, T])(implicit codec: GenCodec[T]): BsonRef[S, T] = {
    val path = genRef.rawRef.normalize.map {
      case Field(name) => KeyEscaper.escape(name)
    }.mkString(BsonKeySeparator)

    BsonRef(path, codec, genRef.fun)
  }

  implicit def bsonRefIterableUpdating[S, E: GenCodec, C[T] <: Iterable[T]](bsonRef: BsonRef[S, C[E]]): BsonRefIterableUpdating[E, C] = {
    new BsonRefIterableUpdating[E, C](bsonRef)
  }
  implicit def bsonRefUpdating[S, T](bsonRef: BsonRef[S, T]): BsonRefUpdating[T] = new BsonRefUpdating(bsonRef)
  implicit def bsonRefSorting[S, T](bsonRef: BsonRef[S, T]): BsonRefSorting[T] = new BsonRefSorting(bsonRef)

  implicit def bsonRefIterableFiltering[S, E: GenCodec, C[T] <: Iterable[T]](bsonRef: BsonRef[S, C[E]]): BsonRefIterableFiltering[E, C] = {
    new BsonRefIterableFiltering[E, C](bsonRef)
  }
  implicit def bsonRefFiltering[S, T](bsonRef: BsonRef[S, T]): BsonRefFiltering[T] = new BsonRefFiltering(bsonRef)
}