Simple monadic structure

simpleMonadicDataType.scala

sealed trait IOToolResponse[+A] {
  def flatMap[B](f: A => IOToolResponse[B]): IOToolResponse[B] = this match {
    case Authenticated(a) => f(a)
    case na: NotAuthenticated => na
  }

  def map[B](f: A => B): IOToolResponse[B] = this.flatMap(a => IOToolResponse.unit(f(a)))
}
object IOToolResponse {
  def unit[A](a: A): IOToolResponse[A] = Authenticated(a)
}
case class Authenticated[A](data: A) extends IOToolResponse[A]
case class NotAuthenticated(code: String, msg: String) extends IOToolResponse[Nothing]