package com.github.everpeace

object BiKleisliSample {

  def main(args:Array[String]): Unit = {

    import scalaz.Scalaz._
    import scalaz.Monad._
    import scalaz.Comonad._
    import com.github.everpeace.BiKleislis._
    import com.github.everpeace.DistributiveLaws._

    case class User(name: String)                  // ユーザ：アクセストークンとして
    case class Person(name: String)                // Person: サンプルエンティティ
    case class AccessRejectedError(message:String) // アクセス拒否エラーメッセージ

    // サンプル Person レポジトリ
    object PersonRepository {
      def search(s: String): Person = Person(s)
      def update(p: Person, s: String): Person = Person(s)
    }

    type AccessTokenComonad[X] = (User, X)                           //User直積スタンピングコモナド
    type AuthorizationErrorMonad[X] = Either[AccessRejectedError, X] //エラー直和スタンピングモナド

    // サンプル操作
    val search = (s: String) => PersonRepository.search(s)
    val updateMrY = (p: Person) => PersonRepository.update(p, "Mr.Y")

    // サンプル権限テーブル（どの関数を誰が実行していいかのマップ）
    val authMap: Map[ScalaObject, Map[User, Boolean]]
    = Map(search    -> Map(User("bob") -> true,    //search は bob, alice 両方OK
                           User("alice") -> true),
          updateMrY -> Map(User("bob") -> true))   //update は bob だけOK

    // 操作に対する認可関数(上記権限テーブルを参照するような関数)
    def authorize[X, Y](f: X => Y): AccessTokenComonad[X] => Boolean = {
      case (u, x) => authMap.getOrElse(f, Map.empty).getOrElse(u, false)
    }

    /**
     * 操作に認可チェックをデコレートする。
     * X=>Y がデコレートされると ((User,X)) => Either[AccessRejectedError,Y] になる。
     * (この下の説明についてはBiKleisliSampleTypeError.scala参照)
     * n-arity functionと n-tuple unary functionは型が違うので注意！！
     * デコレータの戻り値を((User,X)) => Either[AccessRejectedError,Y]にしてもよいが、その場合は★☆に型パラメータを
     * 明示的に渡してやる必要がある。
     */
    def withAccessCheck[X, Y](f: X => Y): AccessTokenComonad[X] => AuthorizationErrorMonad[Y] = {
      case (u, x) => if (authorize(f)(u, x)) {
        Right(f(x))
      } else {
        Left(AccessRejectedError(u.name + " is not granted."))
      }
    }

    // BiKleisli 結合
    // (Tuple2Comonad, EitherMonadに対するDistributesはDistributiveLawsでimplicitに定義されている)
    val searchAndUpdateToMrY = ★☆(withAccessCheck(search)) >>> ★☆(withAccessCheck(updateMrY))

    // alice はupdateが許可されていない。
    println(searchAndUpdateToMrY(User("alice"), "alice")) //Left(AccessRejectedError("alice is not granted."))

    // bob は両方認証成功し、処理成功。
    println(searchAndUpdateToMrY(User("bob"), "bob")) //Right(Person("Mr.Y"))

    // BiKleisli は Arrowにもなる
    val sample_arrow = ★☆(withAccessCheck(search)) &&& searchAndUpdateToMrY
    println(sample_arrow(User("bob"),"bob")) //Right((Person("bob"),Person("Mr.Y")))
    println(sample_arrow(User("alice"),"bob")) //Left(AccessRejectedError("alice is not granted."))
  }
}