Как бросить исключение из пользовательской директивы?

Question

Все вызовы API моей службы - это HTTP POST с параметрами, передаваемыми в многочастном теле. В настоящее время моя аутентификация выглядит так:

formField("token".as[String]) { token => 
    authorize(isAuthorized(token)) { 
     .... my other routes here 
    } 
} 

Но это выглядит слишком подробным. В идеале я хотел бы иметь что-то вроде:

myAuthorization { 
    .... my other routes here 
} 

Итак, я пишу это так:

def myAuthorization: Directive0 = 
    formField("token".as[String]).require(isAuthorized(_)) 

Но как бы вы написать myAuthorization так, что он бросает AuthorizationFailedRejection с в запросе не token?

Answer 1

Мне нравится писать директивы по распылению с экстракторами. Упрощенный пример одного из моих проектов:

def loggedInUser: Directive[Option[User] :: HNil] = headerValue { 
    case Cookie(cookies) => cookies.find(_.name === usrCookie) ∘ extractUserFromCookie 
    case _ => None 
} | reject(NoUserLoggedInRejection) 

def authOpt(usrOpt: Option[User], usr: String): Directive0 = 
    authorize(usrOpt.filter(_.login ≠ usr).isEmpty) 

А затем в файле маршрутизации можно использовать экстракторы:

pathPrefix("path") { 
    loggedInUser { user => 
     authOpt(user, Clients.client) { 
     path("path") { 
      get { 
      complete { 
       html.page() // i'm using Twirl as a fe engine 
      } 
      } 
     } 
     } 
    } 

Если пользователь не вошел в систему, то он выдает отказ которым обрабатывается seperatly:

implicit val customRejectionHandlers = RejectionHandler { 
    case NoUserLoggedInRejection :: _ => 
    ctx => ctx.redirect("/auth", StatusCodes.SeeOther) 
} 

это не самый лучший в случае примера безопасности, но я думаю, что это достаточно четкое

Добавлено

Пример с headerValue директивой от комментария:

// Lets define a extractor for the Host header from the request 
val hostExtractor: HttpHeader => Option[String] = { 
    case Host(host, port) => Some(host) 
    case _ => None 
} 

// Then some path 
val route = { 
    path("somePath") { 
    post { 
     headerValue(hostExtractor) { host => 
     complete(host) 
     } 
    } 
    } 
} 

Большинство таких директив реализуются через extract директивы, которая принимает функцию типа: RequestContext ⇒ T и возвращает T. Например headerValue также реализуется через extract:

extract(_.request.headers.mapFind(protectedF)).flatMap { 
    case Some(Right(a))  ⇒ provide(a) 
    case Some(Left(rejection)) ⇒ reject(rejection) 
    case None     ⇒ reject 
} 

Answer 2

На самом деле я понял, что ошибка поля формы может ввести в заблуждение клиента. А вот как я возвращаю аутентификацию отказа, если данный маркер не был найден в базе данных

def myAuthorization = formField(FieldDefMagnet("token".as[String])).flatMap[User :: HNil]((token: String) => { 
    getUserByToken(token) match { 
    case Some(user) => provide(user) 
    case None => reject(AuthorizationFailedRejection) 
    } 
}) 

Я надеюсь, что это поможет другим людям.