From 20fd6bebda70202767d4d150e0f0ae3749172206 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Wed, 13 Jan 2016 20:44:10 +0000
Subject: Extract upToAndIncluding iterable

---
 .../couscous/util/UpToAndIncludingIterable.java    | 38 ++++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 src/main/java/org/zwobble/couscous/util/UpToAndIncludingIterable.java

(limited to 'src/main/java/org/zwobble/couscous/util/UpToAndIncludingIterable.java')

diff --git a/src/main/java/org/zwobble/couscous/util/UpToAndIncludingIterable.java b/src/main/java/org/zwobble/couscous/util/UpToAndIncludingIterable.java
new file mode 100644
index 0000000..aef5d06
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/util/UpToAndIncludingIterable.java
@@ -0,0 +1,38 @@
+package org.zwobble.couscous.util;
+
+import java.util.Iterator;
+import java.util.NoSuchElementException;
+import java.util.function.Predicate;
+
+public class UpToAndIncludingIterable {
+    public static <T> Iterable<T> upToAndIncluding(Iterable<T> iterable, Predicate<T> predicate) {
+        return () -> new UpToIterator<>(iterable.iterator(), predicate);
+    }
+
+    private static class UpToIterator<T> implements Iterator<T> {
+        private final Iterator<T> iterator;
+        private final Predicate<T> predicate;
+        private boolean finished;
+
+        public UpToIterator(Iterator<T> iterator, Predicate<T> predicate) {
+            this.iterator = iterator;
+            this.predicate = predicate;
+        }
+
+        @Override
+        public boolean hasNext() {
+            return !finished && iterator.hasNext();
+        }
+
+        @Override
+        public T next() {
+            if (hasNext()) {
+                T value = iterator.next();
+                finished = predicate.test(value);
+                return value;
+            } else {
+                throw new NoSuchElementException();
+            }
+        }
+    }
+}
-- 
cgit v1.2.3