lila/modules/db/src/main/CursorExt.scala

package lila.db

import scala.collection.Factory

import reactivemongo.api._
import reactivemongo.api.bson._

trait CursorExt { self: dsl =>

  // Can be refactor as CursorProducer
  implicit final class ExtendCursor[A: BSONDocumentReader](val c: Cursor[A])(implicit
      ec: scala.concurrent.ExecutionContext
  ) {

    // like collect, but with stopOnError defaulting to false
    def gather[M[_]](upTo: Int = Int.MaxValue)(implicit cbf: Factory[A, M[A]]): Fu[M[A]] =
      c.collect[M](upTo, Cursor.ContOnError[M[A]]())

    private def list(limit: Option[Int]): Fu[List[A]] = gather[List](limit | Int.MaxValue)

    def list(limit: Int): Fu[List[A]] = list(limit.some)

    def list(): Fu[List[A]] = list(none)

    // like headOption, but with stopOnError defaulting to false
    def uno: Fu[Option[A]] =
      c.collect[Iterable](
        1,
        Cursor.ContOnError[Iterable[A]]()
      ).map(_.headOption)
  }
}