Add responseToListResponse to RestService (#199)v1.12.4

* Add responseToListResponse to RestService

* Make pagination optional

1 files changed, 16 insertions, 0 deletions

diff --git a/src/main/scala/xyz/driver/core/rest/RestService.scala b/src/main/scala/xyz/driver/core/rest/RestService.scala
index 8d46d72..91b3550 100644
--- a/src/main/scala/xyz/driver/core/rest/RestService.scala
+++ b/src/main/scala/xyz/driver/core/rest/RestService.scala
@@ -6,6 +6,8 @@ import akka.stream.Materializer
 
 import scala.concurrent.{ExecutionContext, Future}
 import scalaz.{ListT, OptionT}
+import scalaz.syntax.equal._
+import scalaz.Scalaz.stringInstance
 
 trait RestService extends Service {
 
@@ -69,4 +71,18 @@ trait RestService extends Service {
 
   protected def endpointUri(baseUri: Uri, path: String, query: Seq[(String, String)]): Uri =
     baseUri.withPath(Uri.Path(path)).withQuery(Uri.Query(query: _*))
+
+  protected def responseToListResponse[T: JsonFormat](pagination: Option[Pagination])(
+      response: HttpResponse): Future[ListResponse[T]] = {
+    import DefaultJsonProtocol._
+    val resourceCount = response.headers
+      .find(_.name() === ContextHeaders.ResourceCount)
+      .map(_.value().toInt)
+      .getOrElse(0)
+    val meta = ListResponse.Meta(resourceCount, pagination.getOrElse(Pagination(resourceCount, 1)))
+    Unmarshal(response.entity).to[List[T]].map(ListResponse(_, meta))
+  }
+
+  protected def responseToListResponse[T: JsonFormat](pagination: Pagination)(
+      response: HttpResponse): Future[ListResponse[T]] = responseToListResponse(Some(pagination))(response)
 }