#![warn(missing_docs)]
use std::borrow::Cow;
use std::fmt::Error as FormatterError;
use std::fmt::{Debug, Display, Formatter};
use std::marker::PhantomData;
use std::time::Duration;
use failure::Fail;
#[cfg(feature = "futures-01")]
use futures_0_1::{Future, IntoFuture};
use http::header::{HeaderMap, HeaderValue, ACCEPT, AUTHORIZATION, CONTENT_TYPE};
use http::status::StatusCode;
use serde::de::DeserializeOwned;
use serde::{Deserialize, Serialize};
use url::{form_urlencoded, Url};
#[cfg(feature = "futures-03")]
mod async_internal;
pub mod basic;
#[cfg(feature = "curl")]
pub mod curl;
pub mod helpers;
#[cfg(any(feature = "reqwest-09", feature = "reqwest-010"))]
pub mod reqwest;
#[cfg(test)]
mod tests;
mod types;
pub use http;
pub use url;
pub use types::{
AccessToken, AuthUrl, AuthorizationCode, ClientId, ClientSecret, CsrfToken, PkceCodeChallenge,
PkceCodeChallengeMethod, PkceCodeVerifier, RedirectUrl, RefreshToken, ResourceOwnerPassword,
ResourceOwnerUsername, ResponseType, Scope, TokenUrl,
};
#[cfg(feature = "futures-03")]
pub use async_internal::{
AsyncClientCredentialsTokenRequest, AsyncCodeTokenRequest, AsyncPasswordTokenRequest,
AsyncRefreshTokenRequest,
};
const CONTENT_TYPE_JSON: &str = "application/json";
const CONTENT_TYPE_FORMENCODED: &str = "application/x-www-form-urlencoded";
#[derive(Clone, Debug)]
pub enum AuthType {
RequestBody,
BasicAuth,
}
#[derive(Clone, Debug)]
pub struct Client<TE, TR, TT>
where
TE: ErrorResponse,
TR: TokenResponse<TT>,
TT: TokenType,
{
client_id: ClientId,
client_secret: Option<ClientSecret>,
auth_url: AuthUrl,
auth_type: AuthType,
token_url: Option<TokenUrl>,
redirect_url: Option<RedirectUrl>,
phantom_te: PhantomData<TE>,
phantom_tr: PhantomData<TR>,
phantom_tt: PhantomData<TT>,
}
impl<TE, TR, TT> Client<TE, TR, TT>
where
TE: ErrorResponse,
TR: TokenResponse<TT>,
TT: TokenType,
{
pub fn new(
client_id: ClientId,
client_secret: Option<ClientSecret>,
auth_url: AuthUrl,
token_url: Option<TokenUrl>,
) -> Self {
Client {
client_id,
client_secret,
auth_url,
auth_type: AuthType::BasicAuth,
token_url,
redirect_url: None,
phantom_te: PhantomData,
phantom_tr: PhantomData,
phantom_tt: PhantomData,
}
}
pub fn set_auth_type(mut self, auth_type: AuthType) -> Self {
self.auth_type = auth_type;
self
}
pub fn set_redirect_url(mut self, redirect_url: RedirectUrl) -> Self {
self.redirect_url = Some(redirect_url);
self
}
pub fn authorize_url<S>(&self, state_fn: S) -> AuthorizationRequest
where
S: FnOnce() -> CsrfToken,
{
AuthorizationRequest {
auth_url: &self.auth_url,
client_id: &self.client_id,
extra_params: Vec::new(),
pkce_challenge: None,
redirect_url: self.redirect_url.as_ref().map(Cow::Borrowed),
response_type: "code".into(),
scopes: Vec::new(),
state: state_fn(),
}
}
pub fn exchange_code(&self, code: AuthorizationCode) -> CodeTokenRequest<TE, TR, TT> {
CodeTokenRequest {
auth_type: &self.auth_type,
client_id: &self.client_id,
client_secret: self.client_secret.as_ref(),
code,
extra_params: Vec::new(),
pkce_verifier: None,
token_url: self.token_url.as_ref(),
redirect_url: self.redirect_url.as_ref(),
_phantom: PhantomData,
}
}
pub fn exchange_password<'a, 'b>(
&'a self,
username: &'b ResourceOwnerUsername,
password: &'b ResourceOwnerPassword,
) -> PasswordTokenRequest<'b, TE, TR, TT>
where
'a: 'b,
{
PasswordTokenRequest::<'b> {
auth_type: &self.auth_type,
client_id: &self.client_id,
client_secret: self.client_secret.as_ref(),
username,
password,
extra_params: Vec::new(),
scopes: Vec::new(),
token_url: self.token_url.as_ref(),
_phantom: PhantomData,
}
}
pub fn exchange_client_credentials(&self) -> ClientCredentialsTokenRequest<TE, TR, TT> {
ClientCredentialsTokenRequest {
auth_type: &self.auth_type,
client_id: &self.client_id,
client_secret: self.client_secret.as_ref(),
extra_params: Vec::new(),
scopes: Vec::new(),
token_url: self.token_url.as_ref(),
_phantom: PhantomData,
}
}
pub fn exchange_refresh_token<'a, 'b>(
&'a self,
refresh_token: &'b RefreshToken,
) -> RefreshTokenRequest<'b, TE, TR, TT>
where
'a: 'b,
{
RefreshTokenRequest {
auth_type: &self.auth_type,
client_id: &self.client_id,
client_secret: self.client_secret.as_ref(),
extra_params: Vec::new(),
refresh_token,
scopes: Vec::new(),
token_url: self.token_url.as_ref(),
_phantom: PhantomData,
}
}
}
#[derive(Debug)]
pub struct AuthorizationRequest<'a> {
auth_url: &'a AuthUrl,
client_id: &'a ClientId,
extra_params: Vec<(Cow<'a, str>, Cow<'a, str>)>,
pkce_challenge: Option<PkceCodeChallenge>,
redirect_url: Option<Cow<'a, RedirectUrl>>,
response_type: Cow<'a, str>,
scopes: Vec<Cow<'a, Scope>>,
state: CsrfToken,
}
impl<'a> AuthorizationRequest<'a> {
pub fn add_scope(mut self, scope: Scope) -> Self {
self.scopes.push(Cow::Owned(scope));
self
}
pub fn add_extra_param<N, V>(mut self, name: N, value: V) -> Self
where
N: Into<Cow<'a, str>>,
V: Into<Cow<'a, str>>,
{
self.extra_params.push((name.into(), value.into()));
self
}
pub fn use_implicit_flow(mut self) -> Self {
self.response_type = "token".into();
self
}
pub fn set_response_type(mut self, response_type: &ResponseType) -> Self {
self.response_type = (&**response_type).to_owned().into();
self
}
pub fn set_pkce_challenge(mut self, pkce_code_challenge: PkceCodeChallenge) -> Self {
self.pkce_challenge = Some(pkce_code_challenge);
self
}
pub fn set_redirect_url(mut self, redirect_url: Cow<'a, RedirectUrl>) -> Self {
self.redirect_url = Some(redirect_url);
self
}
pub fn url(self) -> (Url, CsrfToken) {
let scopes = self
.scopes
.iter()
.map(|s| s.to_string())
.collect::<Vec<_>>()
.join(" ");
let url = {
let mut pairs: Vec<(&str, &str)> = vec![
("response_type", self.response_type.as_ref()),
("client_id", &self.client_id),
("state", self.state.secret()),
];
if let Some(ref pkce_challenge) = self.pkce_challenge {
pairs.push(("code_challenge", &pkce_challenge.as_str()));
pairs.push(("code_challenge_method", &pkce_challenge.method().as_str()));
}
if let Some(ref redirect_url) = self.redirect_url {
pairs.push(("redirect_uri", redirect_url.as_str()));
}
if !scopes.is_empty() {
pairs.push(("scope", &scopes));
}
let mut url: Url = self.auth_url.url().to_owned();
url.query_pairs_mut()
.extend_pairs(pairs.iter().map(|&(k, v)| (k, &v[..])));
url.query_pairs_mut()
.extend_pairs(self.extra_params.iter().cloned());
url
};
(url, self.state)
}
}
#[derive(Clone, Debug)]
pub struct HttpRequest {
pub url: Url,
pub method: http::method::Method,
pub headers: HeaderMap,
pub body: Vec<u8>,
}
#[derive(Clone, Debug)]
pub struct HttpResponse {
pub status_code: http::status::StatusCode,
pub headers: HeaderMap,
pub body: Vec<u8>,
}
#[derive(Debug)]
pub struct CodeTokenRequest<'a, TE, TR, TT>
where
TE: ErrorResponse,
TR: TokenResponse<TT>,
TT: TokenType,
{
auth_type: &'a AuthType,
client_id: &'a ClientId,
client_secret: Option<&'a ClientSecret>,
code: AuthorizationCode,
extra_params: Vec<(Cow<'a, str>, Cow<'a, str>)>,
pkce_verifier: Option<PkceCodeVerifier>,
token_url: Option<&'a TokenUrl>,
redirect_url: Option<&'a RedirectUrl>,
_phantom: PhantomData<(TE, TR, TT)>,
}
impl<'a, TE, TR, TT> CodeTokenRequest<'a, TE, TR, TT>
where
TE: ErrorResponse,
TR: TokenResponse<TT>,
TT: TokenType,
{
pub fn add_extra_param<N, V>(mut self, name: N, value: V) -> Self
where
N: Into<Cow<'a, str>>,
V: Into<Cow<'a, str>>,
{
self.extra_params.push((name.into(), value.into()));
self
}
pub fn set_pkce_verifier(mut self, pkce_verifier: PkceCodeVerifier) -> Self {
self.pkce_verifier = Some(pkce_verifier);
self
}
fn prepare_request<RE>(self) -> Result<HttpRequest, RequestTokenError<RE, TE>>
where
RE: Fail,
{
let mut params = vec![
("grant_type", "authorization_code"),
("code", self.code.secret()),
];
if let Some(ref pkce_verifier) = self.pkce_verifier {
params.push(("code_verifier", pkce_verifier.secret()));
}
Ok(token_request(
self.auth_type,
self.client_id,
self.client_secret,
&self.extra_params,
self.redirect_url,
None,
self.token_url
.ok_or_else(|| RequestTokenError::Other("no token_url provided".to_string()))?,
params,
))
}
pub fn request<F, RE>(self, http_client: F) -> Result<TR, RequestTokenError<RE, TE>>
where
F: FnOnce(HttpRequest) -> Result<HttpResponse, RE>,
RE: Fail,
{
http_client(self.prepare_request()?)
.map_err(RequestTokenError::Request)
.and_then(token_response)
}
}
#[cfg(feature = "futures-01")]
impl<'a, TE, TR, TT> CodeTokenRequest<'a, TE, TR, TT>
where
TE: ErrorResponse + 'static,
TR: TokenResponse<TT>,
TT: TokenType,
{
pub fn request_future<C, F, RE>(
self,
http_client: C,
) -> impl Future<Item = TR, Error = RequestTokenError<RE, TE>>
where
C: FnOnce(HttpRequest) -> F,
F: Future<Item = HttpResponse, Error = RE>,
RE: Fail,
{
self.prepare_request()
.into_future()
.and_then(|http_request| http_client(http_request).map_err(RequestTokenError::Request))
.and_then(|http_response| token_response(http_response).into_future())
}
}
#[derive(Debug)]
pub struct RefreshTokenRequest<'a, TE, TR, TT>
where
TE: ErrorResponse,
TR: TokenResponse<TT>,
TT: TokenType,
{
auth_type: &'a AuthType,
client_id: &'a ClientId,
client_secret: Option<&'a ClientSecret>,
extra_params: Vec<(Cow<'a, str>, Cow<'a, str>)>,
refresh_token: &'a RefreshToken,
scopes: Vec<Cow<'a, Scope>>,
token_url: Option<&'a TokenUrl>,
_phantom: PhantomData<(TE, TR, TT)>,
}
impl<'a, TE, TR, TT> RefreshTokenRequest<'a, TE, TR, TT>
where
TE: ErrorResponse,
TR: TokenResponse<TT>,
TT: TokenType,
{
pub fn add_extra_param<N, V>(mut self, name: N, value: V) -> Self
where
N: Into<Cow<'a, str>>,
V: Into<Cow<'a, str>>,
{
self.extra_params.push((name.into(), value.into()));
self
}
pub fn add_scope(mut self, scope: Scope) -> Self {
self.scopes.push(Cow::Owned(scope));
self
}
pub fn request<F, RE>(self, http_client: F) -> Result<TR, RequestTokenError<RE, TE>>
where
F: FnOnce(HttpRequest) -> Result<HttpResponse, RE>,
RE: Fail,
{
http_client(self.prepare_request()?)
.map_err(RequestTokenError::Request)
.and_then(token_response)
}
fn prepare_request<RE>(&self) -> Result<HttpRequest, RequestTokenError<RE, TE>>
where
RE: Fail,
{
Ok(token_request(
self.auth_type,
self.client_id,
self.client_secret,
&self.extra_params,
None,
Some(&self.scopes),
self.token_url
.ok_or_else(|| RequestTokenError::Other("no token_url provided".to_string()))?,
vec![
("grant_type", "refresh_token"),
("refresh_token", self.refresh_token.secret()),
],
))
}
}
#[cfg(feature = "futures-01")]
impl<'a, TE, TR, TT> RefreshTokenRequest<'a, TE, TR, TT>
where
TE: ErrorResponse + 'static,
TR: TokenResponse<TT>,
TT: TokenType,
{
pub fn request_future<C, F, RE>(
self,
http_client: C,
) -> impl Future<Item = TR, Error = RequestTokenError<RE, TE>>
where
C: FnOnce(HttpRequest) -> F,
F: Future<Item = HttpResponse, Error = RE>,
RE: Fail,
{
self.prepare_request()
.into_future()
.and_then(|http_request| http_client(http_request).map_err(RequestTokenError::Request))
.and_then(|http_response| token_response(http_response).into_future())
}
}
#[derive(Debug)]
pub struct PasswordTokenRequest<'a, TE, TR, TT>
where
TE: ErrorResponse,
TR: TokenResponse<TT>,
TT: TokenType,
{
auth_type: &'a AuthType,
client_id: &'a ClientId,
client_secret: Option<&'a ClientSecret>,
extra_params: Vec<(Cow<'a, str>, Cow<'a, str>)>,
username: &'a ResourceOwnerUsername,
password: &'a ResourceOwnerPassword,
scopes: Vec<Cow<'a, Scope>>,
token_url: Option<&'a TokenUrl>,
_phantom: PhantomData<(TE, TR, TT)>,
}
impl<'a, TE, TR, TT> PasswordTokenRequest<'a, TE, TR, TT>
where
TE: ErrorResponse,
TR: TokenResponse<TT>,
TT: TokenType,
{
pub fn add_extra_param<N, V>(mut self, name: N, value: V) -> Self
where
N: Into<Cow<'a, str>>,
V: Into<Cow<'a, str>>,
{
self.extra_params.push((name.into(), value.into()));
self
}
pub fn add_scope(mut self, scope: Scope) -> Self {
self.scopes.push(Cow::Owned(scope));
self
}
pub fn request<F, RE>(self, http_client: F) -> Result<TR, RequestTokenError<RE, TE>>
where
F: FnOnce(HttpRequest) -> Result<HttpResponse, RE>,
RE: Fail,
{
http_client(self.prepare_request()?)
.map_err(RequestTokenError::Request)
.and_then(token_response)
}
fn prepare_request<RE>(&self) -> Result<HttpRequest, RequestTokenError<RE, TE>>
where
RE: Fail,
{
Ok(token_request(
self.auth_type,
self.client_id,
self.client_secret,
&self.extra_params,
None,
Some(&self.scopes),
self.token_url
.ok_or_else(|| RequestTokenError::Other("no token_url provided".to_string()))?,
vec![
("grant_type", "password"),
("username", self.username),
("password", self.password.secret()),
],
))
}
}
#[cfg(feature = "futures-01")]
impl<'a, TE, TR, TT> PasswordTokenRequest<'a, TE, TR, TT>
where
TE: ErrorResponse + 'static,
TR: TokenResponse<TT>,
TT: TokenType,
{
pub fn request_future<C, F, RE>(
self,
http_client: C,
) -> impl Future<Item = TR, Error = RequestTokenError<RE, TE>>
where
C: FnOnce(HttpRequest) -> F,
F: Future<Item = HttpResponse, Error = RE>,
RE: Fail,
{
self.prepare_request()
.into_future()
.and_then(|http_request| http_client(http_request).map_err(RequestTokenError::Request))
.and_then(|http_response| token_response(http_response).into_future())
}
}
#[derive(Debug)]
pub struct ClientCredentialsTokenRequest<'a, TE, TR, TT>
where
TE: ErrorResponse,
TR: TokenResponse<TT>,
TT: TokenType,
{
auth_type: &'a AuthType,
client_id: &'a ClientId,
client_secret: Option<&'a ClientSecret>,
extra_params: Vec<(Cow<'a, str>, Cow<'a, str>)>,
scopes: Vec<Cow<'a, Scope>>,
token_url: Option<&'a TokenUrl>,
_phantom: PhantomData<(TE, TR, TT)>,
}
impl<'a, TE, TR, TT> ClientCredentialsTokenRequest<'a, TE, TR, TT>
where
TE: ErrorResponse,
TR: TokenResponse<TT>,
TT: TokenType,
{
pub fn add_extra_param<N, V>(mut self, name: N, value: V) -> Self
where
N: Into<Cow<'a, str>>,
V: Into<Cow<'a, str>>,
{
self.extra_params.push((name.into(), value.into()));
self
}
pub fn add_scope(mut self, scope: Scope) -> Self {
self.scopes.push(Cow::Owned(scope));
self
}
pub fn request<F, RE>(self, http_client: F) -> Result<TR, RequestTokenError<RE, TE>>
where
F: FnOnce(HttpRequest) -> Result<HttpResponse, RE>,
RE: Fail,
{
http_client(self.prepare_request()?)
.map_err(RequestTokenError::Request)
.and_then(token_response)
}
fn prepare_request<RE>(&self) -> Result<HttpRequest, RequestTokenError<RE, TE>>
where
RE: Fail,
{
Ok(token_request(
self.auth_type,
self.client_id,
self.client_secret,
&self.extra_params,
None,
Some(&self.scopes),
self.token_url
.ok_or_else(|| RequestTokenError::Other("no token_url provided".to_string()))?,
vec![("grant_type", "client_credentials")],
))
}
}
#[cfg(feature = "futures-01")]
impl<'a, TE, TR, TT> ClientCredentialsTokenRequest<'a, TE, TR, TT>
where
TE: ErrorResponse + 'static,
TR: TokenResponse<TT>,
TT: TokenType,
{
pub fn request_future<C, F, RE>(
self,
http_client: C,
) -> impl Future<Item = TR, Error = RequestTokenError<RE, TE>>
where
C: FnOnce(HttpRequest) -> F,
F: Future<Item = HttpResponse, Error = RE>,
RE: Fail,
{
self.prepare_request()
.into_future()
.and_then(|http_request| http_client(http_request).map_err(RequestTokenError::Request))
.and_then(|http_response| token_response(http_response).into_future())
}
}
#[allow(clippy::too_many_arguments)]
fn token_request<'a>(
auth_type: &'a AuthType,
client_id: &'a ClientId,
client_secret: Option<&'a ClientSecret>,
extra_params: &'a [(Cow<'a, str>, Cow<'a, str>)],
redirect_url: Option<&'a RedirectUrl>,
scopes: Option<&'a Vec<Cow<'a, Scope>>>,
token_url: &'a TokenUrl,
params: Vec<(&'a str, &'a str)>,
) -> HttpRequest {
let mut headers = HeaderMap::new();
headers.append(ACCEPT, HeaderValue::from_static(CONTENT_TYPE_JSON));
headers.append(
CONTENT_TYPE,
HeaderValue::from_static(CONTENT_TYPE_FORMENCODED),
);
let scopes_opt = scopes.and_then(|scopes| {
if !scopes.is_empty() {
Some(
scopes
.iter()
.map(|s| s.to_string())
.collect::<Vec<_>>()
.join(" "),
)
} else {
None
}
});
let mut params: Vec<(&str, &str)> = params;
if let Some(ref scopes) = scopes_opt {
params.push(("scope", scopes));
}
match auth_type {
AuthType::RequestBody => {
params.push(("client_id", client_id));
if let Some(ref client_secret) = client_secret {
params.push(("client_secret", client_secret.secret()));
}
}
AuthType::BasicAuth => {
let urlencoded_id: String =
form_urlencoded::byte_serialize(&client_id.as_bytes()).collect();
let urlencoded_secret = client_secret.map(|secret| {
form_urlencoded::byte_serialize(secret.secret().as_bytes()).collect::<String>()
});
let b64_credential = base64::encode(&format!(
"{}:{}",
&urlencoded_id,
urlencoded_secret
.as_ref()
.map(|secret| secret.as_str())
.unwrap_or("")
));
headers.append(
AUTHORIZATION,
HeaderValue::from_str(&format!("Basic {}", &b64_credential)).unwrap(),
);
}
}
if let Some(ref redirect_url) = redirect_url {
params.push(("redirect_uri", redirect_url.as_ref()));
}
params.extend_from_slice(
extra_params
.iter()
.map(|&(ref k, ref v)| (k.as_ref(), v.as_ref()))
.collect::<Vec<_>>()
.as_slice(),
);
let body = url::form_urlencoded::Serializer::new(String::new())
.extend_pairs(params)
.finish()
.into_bytes();
HttpRequest {
url: token_url.url().to_owned(),
method: http::method::Method::POST,
headers,
body,
}
}
fn token_response<RE, TE, TR, TT>(
http_response: HttpResponse,
) -> Result<TR, RequestTokenError<RE, TE>>
where
RE: Fail,
TE: ErrorResponse,
TR: TokenResponse<TT>,
TT: TokenType,
{
if http_response.status_code != StatusCode::OK {
let reason = http_response.body.as_slice();
if reason.is_empty() {
return Err(RequestTokenError::Other(
"Server returned empty error response".to_string(),
));
} else {
let error = match serde_json::from_slice::<TE>(reason) {
Ok(error) => RequestTokenError::ServerResponse(error),
Err(error) => RequestTokenError::Parse(error, reason.to_vec()),
};
return Err(error);
}
}
http_response
.headers
.get(CONTENT_TYPE)
.map_or(Ok(()), |content_type|
if content_type.to_str().ok().filter(|ct| ct.to_lowercase().starts_with(CONTENT_TYPE_JSON)).is_none() {
Err(
RequestTokenError::Other(
format!(
"Unexpected response Content-Type: {:?}, should be `{}`",
content_type,
CONTENT_TYPE_JSON
)
)
)
} else {
Ok(())
}
)?;
if http_response.body.is_empty() {
Err(RequestTokenError::Other(
"Server returned empty response body".to_string(),
))
} else {
let response_body = http_response.body.as_slice();
serde_json::from_slice(response_body)
.map_err(|e| RequestTokenError::Parse(e, response_body.to_vec()))
}
}
pub trait TokenType: Clone + DeserializeOwned + Debug + PartialEq + Serialize {}
pub trait ExtraTokenFields: DeserializeOwned + Debug + Serialize {}
#[derive(Clone, Debug, Deserialize, PartialEq, Serialize)]
pub struct EmptyExtraTokenFields {}
impl ExtraTokenFields for EmptyExtraTokenFields {}
pub trait TokenResponse<TT>: Debug + DeserializeOwned + Serialize
where
TT: TokenType,
{
fn access_token(&self) -> &AccessToken;
fn token_type(&self) -> &TT;
fn expires_in(&self) -> Option<Duration>;
fn refresh_token(&self) -> Option<&RefreshToken>;
fn scopes(&self) -> Option<&Vec<Scope>>;
}
#[derive(Clone, Debug, Deserialize, Serialize)]
pub struct StandardTokenResponse<EF, TT>
where
EF: ExtraTokenFields,
TT: TokenType,
{
access_token: AccessToken,
#[serde(bound = "TT: TokenType")]
#[serde(deserialize_with = "helpers::deserialize_untagged_enum_case_insensitive")]
token_type: TT,
#[serde(skip_serializing_if = "Option::is_none")]
expires_in: Option<u64>,
#[serde(skip_serializing_if = "Option::is_none")]
refresh_token: Option<RefreshToken>,
#[serde(rename = "scope")]
#[serde(deserialize_with = "helpers::deserialize_space_delimited_vec")]
#[serde(serialize_with = "helpers::serialize_space_delimited_vec")]
#[serde(skip_serializing_if = "Option::is_none")]
#[serde(default)]
scopes: Option<Vec<Scope>>,
#[serde(bound = "EF: ExtraTokenFields")]
#[serde(flatten)]
extra_fields: EF,
}
impl<EF, TT> StandardTokenResponse<EF, TT>
where
EF: ExtraTokenFields,
TT: TokenType,
{
pub fn new(access_token: AccessToken, token_type: TT, extra_fields: EF) -> Self {
Self {
access_token,
token_type,
expires_in: None,
refresh_token: None,
scopes: None,
extra_fields,
}
}
pub fn set_access_token(&mut self, access_token: AccessToken) {
self.access_token = access_token;
}
pub fn set_token_type(&mut self, token_type: TT) {
self.token_type = token_type;
}
pub fn set_expires_in(&mut self, expires_in: Option<&Duration>) {
self.expires_in = expires_in.map(Duration::as_secs);
}
pub fn set_refresh_token(&mut self, refresh_token: Option<RefreshToken>) {
self.refresh_token = refresh_token;
}
pub fn set_scopes(&mut self, scopes: Option<Vec<Scope>>) {
self.scopes = scopes;
}
pub fn extra_fields(&self) -> &EF {
&self.extra_fields
}
pub fn set_extra_fields(&mut self, extra_fields: EF) {
self.extra_fields = extra_fields;
}
}
impl<EF, TT> TokenResponse<TT> for StandardTokenResponse<EF, TT>
where
EF: ExtraTokenFields,
TT: TokenType,
{
fn access_token(&self) -> &AccessToken {
&self.access_token
}
fn token_type(&self) -> &TT {
&self.token_type
}
fn expires_in(&self) -> Option<Duration> {
self.expires_in.map(Duration::from_secs)
}
fn refresh_token(&self) -> Option<&RefreshToken> {
self.refresh_token.as_ref()
}
fn scopes(&self) -> Option<&Vec<Scope>> {
self.scopes.as_ref()
}
}
pub trait ErrorResponse: Debug + DeserializeOwned + Send + Serialize + Sync {}
pub trait ErrorResponseType: Debug + DeserializeOwned + Send + Serialize + Sync {}
#[derive(Clone, Debug, Deserialize, PartialEq, Serialize)]
pub struct StandardErrorResponse<T: ErrorResponseType> {
#[serde(bound = "T: ErrorResponseType")]
error: T,
#[serde(default)]
#[serde(skip_serializing_if = "Option::is_none")]
error_description: Option<String>,
#[serde(default)]
#[serde(skip_serializing_if = "Option::is_none")]
error_uri: Option<String>,
}
impl<T: ErrorResponseType> StandardErrorResponse<T> {
pub fn new(error: T, error_description: Option<String>, error_uri: Option<String>) -> Self {
Self {
error,
error_description,
error_uri,
}
}
pub fn error(&self) -> &T {
&self.error
}
pub fn error_description(&self) -> Option<&String> {
self.error_description.as_ref()
}
pub fn error_uri(&self) -> Option<&String> {
self.error_uri.as_ref()
}
}
impl<T> ErrorResponse for StandardErrorResponse<T> where T: ErrorResponseType + 'static {}
impl<TE> Display for StandardErrorResponse<TE>
where
TE: ErrorResponseType + Display,
{
fn fmt(&self, f: &mut Formatter) -> Result<(), FormatterError> {
let mut formatted = self.error().to_string();
if let Some(error_description) = self.error_description() {
formatted.push_str(": ");
formatted.push_str(error_description);
}
if let Some(error_uri) = self.error_uri() {
formatted.push_str(" / See ");
formatted.push_str(error_uri);
}
write!(f, "{}", formatted)
}
}
#[derive(Debug, Fail)]
pub enum RequestTokenError<RE, T>
where
RE: Fail,
T: ErrorResponse + 'static,
{
#[fail(display = "Server returned error response")]
ServerResponse(T),
#[fail(display = "Request failed")]
Request(#[cause] RE),
#[fail(display = "Failed to parse server response")]
Parse(#[cause] serde_json::error::Error, Vec<u8>),
#[fail(display = "Other error: {}", _0)]
Other(String),
}