lens.scala

// Copyright (c) 2013-2020 Rob Norris and Contributors
// This software is licensed under the MIT License (MIT).
// For more information see LICENSE or https://opensource.org/licenses/MIT

package doobie.util

import cats.data.State

// embarrassigly minimal lens implementation to avoid a lib conflict
object lens {

  final case class Lens[A, B](get: A => B, set: (A, B) => A) {

    def andThen[C](bc: Lens[B, C]): Lens[A, C] =
      Lens(a => bc.get(get(a)), (a, c) => set(a, bc.set(get(a), c)))

    def >=>[C](bc: Lens[B, C]): Lens[A, C] =
      andThen(bc)

    def compose[C](bc: Lens[C, A]): Lens[C, B] =
      bc.andThen(this)

    def modify(a: A, f: B => B): A =
      set(a, f(get(a)))

    def xmapA[AA](f: AA => A, g: A => AA): Lens[AA, B] =
      Lens(aa => get(f(aa)), (aa, b) => g(set(f(aa), b)))

    def xmapB[BB](f: BB => B, g: B => BB): Lens[A, BB] =
      Lens(a => g(get(a)), (a, bb) => set(a, f(bb)))

    def mods(f: B => B): State[A, B] =
      State { a =>
        val b = get(a)
        val b_ = f(b)
        val a_ = set(a, b_)
        (a_, b_)
      }

    def %=(f: B => B): State[A, B] =
      mods(f)

    def :=(b: => B): State[A, B] =
      mods(_ => b)

  }

  type @>[A, B] = Lens[A, B]

}