Add fold to ParseResult

lucidsoftware · Jun 3, 2016 · 5adbeec · 5adbeec
1 parent 824774e
commit 5adbeec
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/xtract-core/src/main/scala/com/lucidchart/open/xtract/ParseResult.scala b/xtract-core/src/main/scala/com/lucidchart/open/xtract/ParseResult.scala
@@ -218,6 +218,19 @@ sealed trait ParseResult[+A] {
     f(get)
   }
 
+  /**
+   * Handle both success and failure.
+   * @param onFailure Function to handle a [[ParseFailure]].
+   * @param onSuccess Function to process a successful parse.
+   */
+  def fold[B](onFailure: Seq[ParseError] => B)(onSuccess: A => B): B = {
+    if (isSuccessful) {
+      onSuccess(get)
+    } else {
+      onFailure(errors)
+    }
+  }
+
   /**
    * Apply a PartialFunction to the result value.
    * If the `pf` isn't defined for the result value, a [[ParseFailure]] with the `otherwise` error