From 2fc1c3baeef3662258caa64068fbcb25401a4065 Mon Sep 17 00:00:00 2001 From: Zach Smith Date: Fri, 12 May 2017 16:01:44 -0700 Subject: Re-add withAuthToken method to ServiceRequestToken --- src/main/scala/xyz/driver/core/rest.scala | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'src/main/scala') diff --git a/src/main/scala/xyz/driver/core/rest.scala b/src/main/scala/xyz/driver/core/rest.scala index 55755ba..5c4b332 100644 --- a/src/main/scala/xyz/driver/core/rest.scala +++ b/src/main/scala/xyz/driver/core/rest.scala @@ -100,11 +100,18 @@ package rest { 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) + user + ) override def hashCode(): Int = Seq[Any](trackingId, contextHeaders).foldLeft(31)((result, obj) => 31 * result + obj.hashCode()) -- cgit v1.2.3