Dispatch on a case

finch-dispatch.scala

import io.finch._

sealed trait Req
object Req {
  case class Foo(i: Int) extends Req
  case class Bar(s: String) extends Req
}

def dispatch[A](pf: PartialFunction[Req, Output[A]]): Endpoint[A] = new Endpoint[A] {
  override def apply(i: Input): Endpoint.Result[A] = body.as[Req].apply(i).flatMap {
    case (remainder, output @ Output.Payload(value, _)) if pf.isDefinedAt(value) => 
      output.flatMap(a => pf(a))
    case _ => None
  }
}

val foo: Endpoint[Int] = dispatch {
  case Foo(i) => Ok(i)
}

val bar: Endpoint[String] = dispatch {
  case Bar(s) => Ok(s)
}

val api = post("my" :: "endpoint") :: (foo :+: bar)