From cf88923672091ca9f9d43d9b603c6dab0f13650a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jonathan=20Wid=C3=A9n?= <jonathan.loke.widen@gmail.com>
Date: Tue, 2 May 2023 03:53:03 +0200
Subject: [PATCH] Add `flatten_left` and `flatten_right`

---
 src/lib.rs | 47 +++++++++++++++++++++++++++++++++++++++--------
 1 file changed, 39 insertions(+), 8 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index 9a271c3..c526430 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -18,20 +18,13 @@
 #[cfg(any(test, feature = "use_std"))]
 extern crate std;
 
-#[cfg(feature = "serde")]
-pub mod serde_untagged;
-
-#[cfg(feature = "serde")]
-pub mod serde_untagged_optional;
-
-use core::convert::{AsMut, AsRef};
+use core::convert::{identity, AsMut, AsRef};
 use core::fmt;
 use core::future::Future;
 use core::iter;
 use core::ops::Deref;
 use core::ops::DerefMut;
 use core::pin::Pin;
-
 #[cfg(any(test, feature = "use_std"))]
 use std::error::Error;
 #[cfg(any(test, feature = "use_std"))]
@@ -39,6 +32,12 @@ use std::io::{self, BufRead, Read, Seek, SeekFrom, Write};
 
 pub use crate::Either::{Left, Right};
 
+#[cfg(feature = "serde")]
+pub mod serde_untagged;
+
+#[cfg(feature = "serde")]
+pub mod serde_untagged_optional;
+
 /// The enum `Either` with variants `Left` and `Right` is a general purpose
 /// sum type with two cases.
 ///
@@ -858,6 +857,38 @@ impl<T, L, R> Either<(L, T), (R, T)> {
     }
 }
 
+impl<L, R> Either<Either<L, R>, R> {
+    /// Flattens a nested structure where the left value type is an [`Either`].
+    ///
+    /// ```
+    /// use either::*;
+    /// let left: Either<Either<&str, u64>, u64> = Left(Left("foo"));
+    /// assert_eq!(left.flatten_left(), Left("foo"));
+    ///
+    /// let right: Either<Either<&str, u64>, u64> = Right(30);
+    /// assert_eq!(right.flatten_left(), Right(30));
+    /// ```
+    pub fn flatten_left(self) -> Either<L, R> {
+        self.either(identity, Right)
+    }
+}
+
+impl<L, R> Either<L, Either<L, R>> {
+    /// Flattens a nested structure where the right value type is an [`Either`].
+    ///
+    /// ```
+    /// use either::*;
+    /// let left: Either<&str, Either<&str, u64>> = Right(Right(10));
+    /// assert_eq!(left.flatten_right(), Right(10));
+    ///
+    /// let right: Either<&str, Either<&str, u64>> = Left("foo");
+    /// assert_eq!(right.flatten_right(), Left("foo"));
+    /// ```
+    pub fn flatten_right(self) -> Either<L, R> {
+        self.either(Left, identity)
+    }
+}
+
 impl<T> Either<T, T> {
     /// Extract the value of an either over two equivalent types.
     ///