From 7fccda0f13f6f8b7f7a1f6e87a648fe1c3396656 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 2 Apr 2016 14:52:21 +0100
Subject: Use JSON to configure CLI

---
 .../java/org/zwobble/couscous/util/ExtraIterables.java  | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

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

diff --git a/src/main/java/org/zwobble/couscous/util/ExtraIterables.java b/src/main/java/org/zwobble/couscous/util/ExtraIterables.java
index cd9b3df..a77869d 100644
--- a/src/main/java/org/zwobble/couscous/util/ExtraIterables.java
+++ b/src/main/java/org/zwobble/couscous/util/ExtraIterables.java
@@ -5,6 +5,7 @@ import com.google.common.collect.Iterables;
 import java.util.Iterator;
 import java.util.function.BiFunction;
 import java.util.function.Function;
+import java.util.function.Supplier;
 import java.util.stream.Stream;
 import java.util.stream.StreamSupport;
 
@@ -105,15 +106,19 @@ public class ExtraIterables {
     }
 
     public static <T, R> Iterable<R> lazyFlatMap(Iterable<T> iterable, Function<T, Iterable<R>> function) {
-        return new Iterable<R>() {
-            @Override
-            public Iterator<R> iterator() {
-                return stream(iterable).flatMap(element -> stream(function.apply(element))).iterator();
-            }
-        };
+        return iterable(() -> stream(iterable).flatMap(element -> stream(function.apply(element))));
     }
 
     public static <T> Stream<T> stream(Iterable<T> iterable) {
         return StreamSupport.stream(iterable.spliterator(), false);
     }
+
+    public static <T, E> Iterable<T> iterable(Supplier<Stream<T>> stream) {
+        return new Iterable<T>() {
+            @Override
+            public Iterator<T> iterator() {
+                return stream.get().iterator();
+            }
+        };
+    }
 }
-- 
cgit v1.2.3