Merge pull request #508 from danicheg/feither-leftMapOrKeepIn

Add `leftMapOrKeepIn` / `leftFlatMapOrKeepIn` to `F[Either[A,B]]` syntax
typelevel · Jul 28, 2024 · e7ed4a7 · e7ed4a7
2 parents 43d5afd + 9d7b41a
commit e7ed4a7
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/shared/src/main/scala/mouse/feither.scala b/shared/src/main/scala/mouse/feither.scala
@@ -81,6 +81,14 @@ final class FEitherOps[F[_], L, R](private val felr: F[Either[L, R]]) extends An
       case r @ Right(_) => r.asInstanceOf[Right[A, B]]
     }
 
+  def leftFlatMapOrKeepIn[LL >: L, RR >: R](
+    pf: PartialFunction[L, Either[LL, RR]]
+  )(implicit F: Functor[F]): F[Either[LL, RR]] =
+    F.map(felr) {
+      case l @ Left(a)    => pf.applyOrElse(a, (_: L) => l)
+      case r: Right[L, R] => r.asInstanceOf[Right[LL, RR]]
+    }
+
   def leftFlatMapF[A, B >: R](f: L => F[Either[A, B]])(implicit F: Monad[F]): F[Either[A, B]] =
     F.flatMap(felr) {
       case Left(left)   => f(left)
@@ -93,6 +101,12 @@ final class FEitherOps[F[_], L, R](private val felr: F[Either[L, R]]) extends An
       case r @ Right(_) => r.asInstanceOf[Right[A, R]]
     }
 
+  def leftMapOrKeepIn[LL >: L](pf: PartialFunction[L, LL])(implicit F: Functor[F]): F[Either[LL, R]] =
+    F.map(felr) {
+      case Left(a)        => Left(pf.applyOrElse(a, identity[LL]))
+      case r: Right[L, R] => r.asInstanceOf[Right[LL, R]]
+    }
+
   def leftAsIn[B](b: => B)(implicit F: Functor[F]): F[Either[B, R]] =
     leftMapIn(_ => b)
 

diff --git a/shared/src/test/scala/mouse/FEitherSyntaxTest.scala b/shared/src/test/scala/mouse/FEitherSyntaxTest.scala
@@ -104,6 +104,13 @@ class FEitherSyntaxTest extends MouseSuite {
     assertEquals(leftValue.leftFlatMapIn(_ => "".asLeft[Int]), List("".asLeft[Int]))
   }
 
+  test("FEitherSyntax.leftFlatMapOrKeepIn") {
+    assertEquals(rightValue.leftFlatMapOrKeepIn { case "42" => Left("") }, rightValue)
+    assertEquals(leftValue.leftFlatMapOrKeepIn { case "4242" => Right(42) }, leftValue)
+    assertEquals(leftValue.leftFlatMapOrKeepIn { case "42" => Left("") }, List(Left("")))
+    assertEquals(leftValue.leftFlatMapOrKeepIn { case "42" => Right(42) }, rightValue)
+  }
+
   test("FEitherSyntax.leftFlatMapF") {
     assertEquals(rightValue.leftFlatMapF(_ => List("".asLeft[Int])), rightValue)
     assertEquals(leftValue.leftFlatMapF(_ => List("".asLeft[Int])), List("".asLeft[Int]))
@@ -114,6 +121,12 @@ class FEitherSyntaxTest extends MouseSuite {
     assertEquals(leftValue.leftMapIn(_ => ""), List("".asLeft[Int]))
   }
 
+  test("FEitherSyntax.leftMapOrKeepIn") {
+    assertEquals(rightValue.leftMapOrKeepIn { case "42" => "" }, rightValue)
+    assertEquals(leftValue.leftMapOrKeepIn { case "4242" => "" }, leftValue)
+    assertEquals(leftValue.leftMapOrKeepIn { case "42" => "" }, List(Left("")))
+  }
+
   test("FEitherSyntax.leftAsIn") {
     assertEquals(rightValue.leftAsIn(""), rightValue)
     assertEquals(leftValue.leftAsIn(""), List("".asLeft[Int]))