From 1adda71863c3c757f0b6e5b083135e962abf2a59 Mon Sep 17 00:00:00 2001
From: k-nasa <htilcs1115@gmail.com>
Date: Thu, 17 Oct 2019 18:06:46 +0900
Subject: [PATCH] feat: Add stream::successors

---
 src/stream/mod.rs        |  2 ++
 src/stream/successors.rs | 61 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 63 insertions(+)
 create mode 100644 src/stream/successors.rs

diff --git a/src/stream/mod.rs b/src/stream/mod.rs
index c41ceb68a..ac9336ba2 100644
--- a/src/stream/mod.rs
+++ b/src/stream/mod.rs
@@ -30,6 +30,7 @@ pub use repeat::{repeat, Repeat};
 pub use stream::{
     Chain, Filter, Fuse, Inspect, Scan, Skip, SkipWhile, StepBy, Stream, Take, TakeWhile, Zip,
 };
+pub use successors::{successors, Successors};
 
 pub(crate) mod stream;
 
@@ -37,6 +38,7 @@ mod empty;
 mod from_fn;
 mod once;
 mod repeat;
+mod successors;
 
 cfg_if! {
     if #[cfg(any(feature = "unstable", feature = "docs"))] {
diff --git a/src/stream/successors.rs b/src/stream/successors.rs
new file mode 100644
index 000000000..559554508
--- /dev/null
+++ b/src/stream/successors.rs
@@ -0,0 +1,61 @@
+use std::pin::Pin;
+
+use crate::stream::Stream;
+use crate::task::{Context, Poll};
+
+/// Creates a new stream where each successive item is computed based on the preceding one.
+/// The stream starts with the given first item (if any) and calls the given FnMut(&T) -> Option<T> closure to compute each item’s successor.
+///
+/// # Examples
+///
+/// ```
+/// # async_std::task::block_on(async {
+/// #
+/// use async_std::prelude::*;
+/// use async_std::stream;
+///
+/// let powers_of_10 = stream::successors(Some(1_u16), |n| n.checked_mul(10));
+/// assert_eq!(powers_of_10.collect::<Vec<_>>().await, &[1, 10, 100, 1_000, 10_000]);
+///
+/// #
+/// # })
+/// ```
+pub fn successors<T, F>(first: Option<T>, succ: F) -> Successors<T, F>
+where
+    F: FnMut(&T) -> Option<T>,
+{
+    Successors { next: first, succ }
+}
+
+/// An new stream where each successive item is computed based on the preceding one.
+///
+/// This `struct` is created by the [`successors`] function.
+/// See its documentation for more.
+///
+/// [`successors`]: fn.successors.html
+#[derive(Clone, Debug)]
+pub struct Successors<T, F> {
+    next: Option<T>,
+    succ: F,
+}
+
+impl<T, F> Successors<T, F> {
+    pin_utils::unsafe_unpinned!(next: Option<T>);
+    pin_utils::unsafe_unpinned!(succ: F);
+}
+
+impl<T, F> Stream for Successors<T, F>
+where
+    F: FnMut(&T) -> Option<T>,
+{
+    type Item = T;
+
+    fn poll_next(mut self: Pin<&mut Self>, _: &mut Context<'_>) -> Poll<Option<Self::Item>> {
+        let item = self.as_mut().next().take().map(|item| {
+            *self.as_mut().next() = (self.as_mut().succ())(&item);
+            item
+        });
+
+        Poll::Ready(item)
+    }
+}