use crate::client::HttpClientExt;
use crate::error::SocialiteError;
use crate::provider::Provider;
use crate::user::SocialiteUser;
use async_trait::async_trait;
use serde_json::Value;
crate::define_provider!(StripeProvider, "read_write");
#[async_trait]
impl Provider for StripeProvider {
fn redirect_url(&self) -> String {
let mut params = crate::provider::build_oauth_params(
&self.client_id,
&self.redirect_url,
&self.scopes,
self.state.as_deref(),
self.pkce_challenge.as_deref(),
);
params.append_pair("response_type", "code");
format!(
"https://connect.stripe.com/oauth/authorize?{}",
params.finish()
)
}
async fn get_user(&self, auth_code: &str) -> Result<SocialiteUser, SocialiteError> {
let token_res = self
.http_client
.post(self.token_url())
.form(&[
("grant_type", "authorization_code"),
("client_secret", self.client_secret.as_str()),
("code", auth_code),
("redirect_uri", self.redirect_url.as_str()),
])
.send()
.await?
.error_for_status()?
.json::<Value>()
.await?;
let stripe_user_id = token_res["stripe_user_id"]
.as_str()
.ok_or_else(|| SocialiteError::Token("Failed to get stripe_user_id".to_string()))?;
let access_token = token_res["access_token"]
.as_str()
.ok_or_else(|| SocialiteError::Token("Failed to get access_token".to_string()))?;
let mut user = self.get_user_from_token(access_token).await?;
if user.id.is_empty() {
user.id = stripe_user_id.to_string();
}
user.refresh_token = token_res["refresh_token"]
.as_str()
.map(|s: &str| s.to_string());
user.expires_in = token_res["expires_in"]
.as_u64()
.or_else(|| token_res["expires_in"].as_i64().map(|v| v as u64));
Ok(user)
}
async fn get_user_from_token(
&self,
access_token: &str,
) -> Result<SocialiteUser, SocialiteError> {
let user_res = self
.http_client
.get("https://api.stripe.com/v1/account")
.header("Authorization", format!("Bearer {}", access_token))
.send()
.await?
.error_for_status()?
.json::<Value>()
.await?;
let name = user_res["business_profile"]["name"]
.as_str()
.or_else(|| user_res["settings"]["dashboard"]["display_name"].as_str())
.unwrap_or("");
Ok(SocialiteUser {
id: user_res["id"].as_str().unwrap_or("").to_string(),
name: name.to_string(),
email: user_res["email"].as_str().map(|s: &str| s.to_string()),
avatar_url: None, raw_data: user_res,
access_token: access_token.to_string(),
refresh_token: None,
expires_in: None,
})
}
fn token_url(&self) -> String {
"https://connect.stripe.com/oauth/token".to_string()
}
async fn refresh_token(
&self,
refresh_token: &str,
) -> Result<SocialiteUser, crate::error::SocialiteError> {
let token_res = self
.http_client
.post(self.token_url())
.form(&[
("client_id", self.client_id.as_str()),
("client_secret", self.client_secret.as_str()),
("refresh_token", refresh_token),
("grant_type", "refresh_token"),
])
.send()
.await?
.error_for_status()?
.json::<serde_json::Value>()
.await?;
if let Some(err) = token_res["error"].as_str() {
let err_desc = token_res["error_description"].as_str().unwrap_or("");
return Err(crate::error::SocialiteError::Token(format!(
"Provider returned error: {} - {}",
err, err_desc
)));
}
let access_token = token_res["access_token"].as_str().ok_or_else(|| {
crate::error::SocialiteError::Token(
"Failed to get access_token during refresh".to_string(),
)
})?;
let mut user = self.get_user_from_token(access_token).await?;
user.refresh_token = token_res["refresh_token"].as_str().map(|s| s.to_string());
user.expires_in = token_res["expires_in"]
.as_u64()
.or_else(|| token_res["expires_in"].as_i64().map(|v| v as u64));
Ok(user)
}
}