1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2018 Make.org
4  *
5  * This program is free software: you can redistribute it and/or modify
6  *  it under the terms of the GNU Affero General Public License as
7  *  published by the Free Software Foundation, either version 3 of the
8  *  License, or (at your option) any later version.
9  *
10  *  This program is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  *  GNU Affero General Public License for more details.
14  *
15  *  You should have received a copy of the GNU Affero General Public License
16  *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
17  *
18  */
19 
20 package org.make.core.technical
21 
22 import scala.annotation.tailrec
23 import cats.Monad
24 import cats.data.ValidatedNec
25 import cats.implicits._
26 import org.make.core.{ValidationError, ValidationFailedError}
27 import cats.data.Validated._
28 
29 object ValidatedUtils {
30 
31   implicit class ValidatedNecWithUtils[T](value: ValidatedNec[ValidationError, T]) {
32 
33     def toValidationEither: Either[ValidationFailedError, T] =
34       value.leftMap(ValidationFailedError(_)).toEither
35 
36     @SuppressWarnings(Array("org.wartremover.warts.TryPartial"))
37     def throwIfInvalid(): T =
38       this.toValidationEither.toTry.get
39   }
40 
41   type ValidatedPartial[A] = ValidatedNec[ValidationError, A]
42 
43   implicit def validatedMonad: Monad[ValidatedPartial] = new Monad[ValidatedPartial] {
44 
45     override def pure[A](x: A): ValidatedPartial[A] =
46       x.validNec
47 
48     override def flatMap[A, B](fa: ValidatedPartial[A])(f: A => ValidatedPartial[B]): ValidatedPartial[B] =
49       fa match {
50         case Invalid(e) => Invalid(e)
51         case Valid(a)   => f(a)
52       }
53 
54     @tailrec
55     override def tailRecM[A, B](a: A)(f: A => ValidatedPartial[Either[A, B]]): ValidatedPartial[B] =
56       f(a) match {
57         case e: Invalid[_]      => e
58         case Valid(Left(nextA)) => tailRecM(nextA)(f)
59         case Valid(Right(b))    => b.validNec
60       }
61   }
62 }