take

1 files changed, 18 insertions, 2 deletions

diff --git a/src/gleam/iterator.gleam b/src/gleam/iterator.gleam
index 53b80d3..6634f23 100644
--- a/src/gleam/iterator.gleam
+++ b/src/gleam/iterator.gleam
@@ -63,7 +63,6 @@ pub fn repeat(x: element) -> Iterator(element) {
   repeatedly(fn() { x })
 }
 
-// TODO: test
 // TODO: document
 pub fn from_list(list: List(element)) -> Iterator(element) {
   unfold(list, fn(acc) {
@@ -96,7 +95,6 @@ pub fn run(iterator) -> Nil {
   fold(iterator, Nil, fn(_, acc) { acc })
 }
 
-// TODO: test
 // TODO: document
 pub fn to_list(iterator: Iterator(element)) -> List(element) {
   iterator
@@ -104,6 +102,24 @@ pub fn to_list(iterator: Iterator(element)) -> List(element) {
   |> list.reverse
 }
 
+fn do_take(iterator, desired, acc) {
+  case desired > 0 {
+    True -> case iterator() {
+      Continue(element, iterator) -> do_take(iterator, desired - 1, [element | acc])
+      Stop -> acc
+    }
+    False -> acc
+  }
+}
+
+// TODO: document
+pub fn take(from iterator: Iterator(e), up_to desired: Int) -> List(e) {
+  iterator
+  |> unopaque
+  |> do_take(_, desired, [])
+  |> list.reverse
+}
+
 // Transforming Iterators
 
 fn do_map(iterator, f) {