1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2018 Make.org
4  *
5  * This program is free software: you can redistribute it and/or modify
6  *  it under the terms of the GNU Affero General Public License as
7  *  published by the Free Software Foundation, either version 3 of the
8  *  License, or (at your option) any later version.
9  *
10  *  This program is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  *  GNU Affero General Public License for more details.
14  *
15  *  You should have received a copy of the GNU Affero General Public License
16  *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
17  *
18  */
19 
20 package org.make.api.technical.directives
21 
22 import akka.http.scaladsl.model.headers.{Authorization, BasicHttpCredentials}
23 import akka.http.scaladsl.server.Directive1
24 import org.make.api.technical.auth.ClientServiceComponent
25 import org.make.api.technical.directives.FutureDirectivesExtensions._
26 import org.make.core.auth.{Client, ClientId}
27 
28 trait ClientDirectives { self: ClientServiceComponent =>
29 
30   def extractClient: Directive1[Option[Client]] = {
31     optionalHeaderValueByType(Authorization).flatMap {
32       case Some(Authorization(BasicHttpCredentials(clientId, clientSecret))) =>
33         val secret = clientSecret match {
34           case ""    => None
35           case other => Some(other)
36         }
37         clientService.getClient(ClientId(clientId), secret).asDirective.flatMap {
38           case Left(e)       => failWith(e)
39           case Right(client) => provide(Some(client))
40         }
41       case _ => provide[Option[Client]](None)
42     }
43   }
44 
45   def extractClientOrDefault: Directive1[Client] = {
46     extractClient.flatMap {
47       case Some(client) => provide(client)
48       case None =>
49         clientService.getDefaultClient().asDirective.flatMap {
50           case Left(e)       => failWith(e)
51           case Right(client) => provide(client)
52         }
53     }
54   }
55 }