1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2021 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.auth
21 
22 import io.circe.{Decoder, Encoder}
23 import io.swagger.annotations.ApiModelProperty
24 import org.make.api.technical.auth.MakeDataHandler.{CreatedAtParameter, RefreshTokenExpirationParameter}
25 import scalaoauth2.provider.AccessToken
26 
27 import scala.annotation.meta.field
28 
29 final case class TokenResponse(
30   @(ApiModelProperty @field)(name = "token_type", required = true)
31   tokenType: String,
32   @(ApiModelProperty @field)(name = "access_token", example = "11111111-2222-3333-4444-555555555555", required = true)
33   accessToken: String,
34   @(ApiModelProperty @field)(name = "expires_in", example = "300", required = true)
35   expiresIn: Long,
36   @(ApiModelProperty @field)(name = "refresh_token", example = "11111111-2222-3333-4444-555555555555")
37   refreshToken: Option[String],
38   @(ApiModelProperty @field)(name = "refresh_expires_in", example = "1500", dataType = "int")
39   refreshExpiresIn: Option[Long],
40   @(ApiModelProperty @field)(name = "created_at", dataType = "dateTime", required = true)
41   createdAt: String
42 )
43 
44 object TokenResponse {
45   def fromAccessToken(tokenResult: AccessToken): TokenResponse = {
46     TokenResponse(
47       "Bearer",
48       tokenResult.token,
49       tokenResult.expiresIn.getOrElse(1L),
50       tokenResult.refreshToken,
51       tokenResult.params.get(RefreshTokenExpirationParameter).map(_.toLong),
52       tokenResult.params.getOrElse(CreatedAtParameter, "")
53     )
54   }
55 
56   implicit val encoder: Encoder[TokenResponse] = {
57     Encoder.forProduct6("token_type", "access_token", "expires_in", "refresh_token", "refresh_expires_in", "created_at") {
58       response =>
59         (
60           response.tokenType,
61           response.accessToken,
62           response.expiresIn,
63           response.refreshToken,
64           response.refreshExpiresIn,
65           response.createdAt
66         )
67     }
68   }
69 
70   implicit val decoder: Decoder[TokenResponse] = {
71     Decoder.forProduct6(
72       "token_type",
73       "access_token",
74       "expires_in",
75       "refresh_token",
76       "refresh_expires_in",
77       "created_at"
78     )(TokenResponse.apply)
79   }
80 }