Collapse nodes where appropriate

1 files changed, 12 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/mammoth/util/MammothOptionals.java b/src/main/java/org/zwobble/mammoth/util/MammothOptionals.java
index e0e4876..6890f16 100644
--- a/src/main/java/org/zwobble/mammoth/util/MammothOptionals.java
+++ b/src/main/java/org/zwobble/mammoth/util/MammothOptionals.java
@@ -15,4 +15,16 @@ public class MammothOptionals {
             return Optional.empty();
         }
     }
+
+    public static <T1, T2, R> Optional<R> map(
+        Optional<T1> first,
+        Optional<T2> second,
+        BiFunction<T1, T2, R> function)
+    {
+        if (first.isPresent() && second.isPresent()) {
+            return Optional.of(function.apply(first.get(), second.get()));
+        } else {
+            return Optional.empty();
+        }
+    }
 }