syntax.CodecOps.scala Maven / Gradle / Ivy

Go to download
// Copyright (c) 2018-2021 by Rob Norris
// This software is licensed under the MIT License (MIT).
// For more information see LICENSE or https://opensource.org/licenses/MIT

package skunk
package syntax

import cats.syntax.all._
import scala.deriving.Mirror

class CodecOps[A <: Tuple](self: Codec[A]) {

  def *:[B](other: Codec[B]): Codec[B *: A] =
    (other, self).imapN((b, a) => b *: a)(t => (t.head, t.tail))

  def *:[B](other: Decoder[B]): Decoder[B *: A] =
    (other, self).mapN((b, a) => b *: a)

  def *:[B](other: Encoder[B]): Encoder[B *: A] =
    (other, self).contramapN(t => (t.head, t.tail))

  def pimap[P <: Product](
    using m: Mirror.ProductOf[P],
          i: m.MirroredElemTypes =:= A
  ): Codec[P] =
    self.imap(m.fromProduct)(p => i(Tuple.fromProductTyped(p)))

}

class CodecOpsLow[A](self: Codec[A]) {

  def *:[B](other: Codec[B]): Codec[B *: A *: EmptyTuple] =
    other product self

  def *:[B](other: Decoder[B]): Decoder[B *: A *: EmptyTuple] =
    other product self

  def *:[B](other: Encoder[B]): Encoder[B *: A *: EmptyTuple] =
    other product self

}

trait ToCodecOps extends ToCodecOpsLow {
  implicit def toCodecOps[A <: Tuple](self: Codec[A]): CodecOps[A] =
    new CodecOps(self)
}

trait ToCodecOpsLow {
  implicit def toCodecOpsLow[A](self: Codec[A]): CodecOpsLow[A] =
    new CodecOpsLow(self)
}

object codec extends ToCodecOps