kyo.kyo.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kyo-core-opt_3 Show documentation

kyo-core-opt

There is a newer version: 0.8.5

import scala.language.higherKinds

package object kyo {

  type NotGiven[T] = scala.util.NotGiven[T]

  type >[+T, -S] >: T // = T | Kyo[_, _, _, T, S]

  extension [T, S](v: T > S)(using NotGiven[Any => S]) {

    inline
    def flatMap[U, S2]( inline f: T => U > S2): U > (S with S2) =
      kyo.core.transform(v)(f)

    inline
    def map[U, S2]( inline f: T => U > S2): U > (S with S2) =
      flatMap(f)

    inline
    def unit: Unit > S =
      map(_ => ())

    inline
    def withFilter( inline p: T => Boolean): T > S =
      map(v => if (!p(v)) throw new MatchError(v) else v)

    inline
    def flatten[U, S2](implicit ev: T => U > S2): U > (S with S2) =
      flatMap(ev)

    inline
    def andThen[U, S2](f: => U > S2)(implicit ev: T => Unit): U > (S with S2) =
      flatMap(_ => f)

    inline
    def repeat(i: Int)(implicit ev: T => Unit): Unit > S =
      if (i <= 0) () else andThen(repeat(i - 1))
  }

  extension [T, S](v: T > Any) {
    inline
    def pure(implicit ev: Any => S): T =
      v.asInstanceOf[T]
  }

  def zip[T1, T2, S](v1: T1 > S, v2: T2 > S): (T1, T2) > S =
    v1.map(t1 => v2.map(t2 => (t1, t2)))

  def zip[T1, T2, T3, S](v1: T1 > S, v2: T2 > S, v3: T3 > S): (T1, T2, T3) > S =
    v1.map(t1 => v2.map(t2 => v3.map(t3 => (t1, t2, t3))))

  def zip[T1, T2, T3, T4, S](
      v1: T1 > S,
      v2: T2 > S,
      v3: T3 > S,
      v4: T4 > S
  ): (T1, T2, T3, T4) > S =
    v1.map(t1 => v2.map(t2 => v3.map(t3 => v4.map(t4 => (t1, t2, t3, t4)))))

}