From 2c524be93052fc6d57359a00fd60d957099885c6 Mon Sep 17 00:00:00 2001
From: Zach Smith <zach@driver.xyz>
Date: Wed, 27 Sep 2017 22:20:53 -0700
Subject: Refactor rest package into separate files

---
 .../driver/core/rest/ServiceRequestContext.scala   | 39 ++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 src/main/scala/xyz/driver/core/rest/ServiceRequestContext.scala

(limited to 'src/main/scala/xyz/driver/core/rest/ServiceRequestContext.scala')

diff --git a/src/main/scala/xyz/driver/core/rest/ServiceRequestContext.scala b/src/main/scala/xyz/driver/core/rest/ServiceRequestContext.scala
new file mode 100644
index 0000000..5235da6
--- /dev/null
+++ b/src/main/scala/xyz/driver/core/rest/ServiceRequestContext.scala
@@ -0,0 +1,39 @@
+package xyz.driver.core.rest
+
+import xyz.driver.core.auth.{AuthToken, PermissionsToken, User}
+import xyz.driver.core.generators
+
+import scalaz.Scalaz.{mapEqual, stringInstance}
+import scalaz.syntax.equal._
+
+class ServiceRequestContext(val trackingId: String = generators.nextUuid().toString,
+                            val contextHeaders: Map[String, String] = Map.empty[String, String]) {
+  def authToken: Option[AuthToken] =
+    contextHeaders.get(AuthProvider.AuthenticationTokenHeader).map(AuthToken.apply)
+
+  def permissionsToken: Option[PermissionsToken] =
+    contextHeaders.get(AuthProvider.PermissionsTokenHeader).map(PermissionsToken.apply)
+
+  def withAuthToken(authToken: AuthToken): ServiceRequestContext =
+    new ServiceRequestContext(
+      trackingId,
+      contextHeaders.updated(AuthProvider.AuthenticationTokenHeader, authToken.value)
+    )
+
+  def withAuthenticatedUser[U <: User](authToken: AuthToken, user: U): AuthorizedServiceRequestContext[U] =
+    new AuthorizedServiceRequestContext(
+      trackingId,
+      contextHeaders.updated(AuthProvider.AuthenticationTokenHeader, authToken.value),
+      user
+    )
+
+  override def hashCode(): Int =
+    Seq[Any](trackingId, contextHeaders).foldLeft(31)((result, obj) => 31 * result + obj.hashCode())
+
+  override def equals(obj: Any): Boolean = obj match {
+    case ctx: ServiceRequestContext => trackingId === ctx.trackingId && contextHeaders === ctx.contextHeaders
+    case _                          => false
+  }
+
+  override def toString: String = s"ServiceRequestContext($trackingId, $contextHeaders)"
+}
-- 
cgit v1.2.3