#[derive(Debug)]
pub enum GrantError {
RequestFailure(reqwest::Error),
InvalidHeaderValue(reqwest::header::InvalidHeaderValue),
UnexpectedStatus(reqwest::StatusCode),
Status401 {
message: String,
},
Status404 {
message: String,
},
Status400 {
errors: Vec<String>,
},
Status500 {
error: String,
},
}
impl From<reqwest::Error> for GrantError {
fn from(error: reqwest::Error) -> GrantError {
GrantError::RequestFailure(error)
}
}
impl From<reqwest::header::InvalidHeaderValue> for GrantError {
fn from(error: reqwest::header::InvalidHeaderValue) -> GrantError {
GrantError::InvalidHeaderValue(error)
}
}
impl GrantError {
pub fn to_account_endpoint_error(&self) -> Option<crate::model::AccountEndpointError> {
match self {
GrantError::Status401 { message } => Some(crate::model::AccountEndpointError::Unauthorized { message: message.clone() }),
GrantError::Status404 { message } => Some(crate::model::AccountEndpointError::NotFound { message: message.clone() }),
GrantError::Status400 { errors } => Some(crate::model::AccountEndpointError::ArgValidation { errors: errors.clone() }),
GrantError::Status500 { error } => Some(crate::model::AccountEndpointError::Internal { error: error.clone() }),
_ => None
}
}
}
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
struct AccountEndpointErrorUnauthorizedPayload {
pub message: String,
}
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
struct AccountEndpointErrorNotFoundPayload {
pub message: String,
}
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
struct AccountEndpointErrorArgValidationPayload {
pub errors: Vec<String>,
}
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
struct AccountEndpointErrorInternalPayload {
pub error: String,
}
#[async_trait::async_trait]
pub trait Grant {
async fn get_grants(&self, account_id: &str, authorization: &str) -> Result<Vec<crate::model::Role>, GrantError>;
async fn get_grant(&self, account_id: &str, role: &str, authorization: &str) -> Result<crate::model::Role, GrantError>;
async fn put_grant(&self, account_id: &str, role: &str, authorization: &str) -> Result<crate::model::Role, GrantError>;
async fn delete_grant(&self, account_id: &str, role: &str, authorization: &str) -> Result<(), GrantError>;
}
#[derive(Clone, Debug)]
pub struct GrantLive {
pub base_url: reqwest::Url,
pub allow_insecure: bool,
}
#[async_trait::async_trait]
impl Grant for GrantLive {
async fn get_grants(&self, account_id: &str, authorization: &str) -> Result<Vec<crate::model::Role>, GrantError> {
let mut url = self.base_url.clone();
url.path_segments_mut().unwrap()
.push("v1")
.push("accounts")
.push(account_id)
.push("grants");
let mut headers = reqwest::header::HeaderMap::new();
headers.append("authorization", reqwest::header::HeaderValue::from_str(&format!("{authorization}"))?);
{
let headers_vec: Vec<(&str, String)> = headers.iter().map(|(k, v)| crate::hide_authorization(k, v)).collect();
tracing::info!(method="get", url=url.to_string(), headers=?headers_vec, body="<no_body>", "get_grants");
}
let mut builder = reqwest::Client::builder();
if self.allow_insecure {
builder = builder.danger_accept_invalid_certs(true);
}
let client = builder.build()?;
let result = client
.get(url)
.headers(headers)
.send()
.await?;
match result.status().as_u16() {
200 => {
let body = result.json::<Vec<crate::model::Role>>().await?;
Ok(body)
}
401 => {
let body = result.json::<AccountEndpointErrorUnauthorizedPayload>().await?;
Err(GrantError::Status401 { message: body.message })
}
404 => {
let body = result.json::<AccountEndpointErrorNotFoundPayload>().await?;
Err(GrantError::Status404 { message: body.message })
}
400 => {
let body = result.json::<AccountEndpointErrorArgValidationPayload>().await?;
Err(GrantError::Status400 { errors: body.errors })
}
500 => {
let body = result.json::<AccountEndpointErrorInternalPayload>().await?;
Err(GrantError::Status500 { error: body.error })
}
_ => Err(GrantError::UnexpectedStatus(result.status()))
}
}
async fn get_grant(&self, account_id: &str, role: &str, authorization: &str) -> Result<crate::model::Role, GrantError> {
let mut url = self.base_url.clone();
url.path_segments_mut().unwrap()
.push("v1")
.push("accounts")
.push(account_id)
.push("grants")
.push(role);
let mut headers = reqwest::header::HeaderMap::new();
headers.append("authorization", reqwest::header::HeaderValue::from_str(&format!("{authorization}"))?);
{
let headers_vec: Vec<(&str, String)> = headers.iter().map(|(k, v)| crate::hide_authorization(k, v)).collect();
tracing::info!(method="get", url=url.to_string(), headers=?headers_vec, body="<no_body>", "get_grant");
}
let mut builder = reqwest::Client::builder();
if self.allow_insecure {
builder = builder.danger_accept_invalid_certs(true);
}
let client = builder.build()?;
let result = client
.get(url)
.headers(headers)
.send()
.await?;
match result.status().as_u16() {
200 => {
let body = result.json::<crate::model::Role>().await?;
Ok(body)
}
401 => {
let body = result.json::<AccountEndpointErrorUnauthorizedPayload>().await?;
Err(GrantError::Status401 { message: body.message })
}
404 => {
let body = result.json::<AccountEndpointErrorNotFoundPayload>().await?;
Err(GrantError::Status404 { message: body.message })
}
400 => {
let body = result.json::<AccountEndpointErrorArgValidationPayload>().await?;
Err(GrantError::Status400 { errors: body.errors })
}
500 => {
let body = result.json::<AccountEndpointErrorInternalPayload>().await?;
Err(GrantError::Status500 { error: body.error })
}
_ => Err(GrantError::UnexpectedStatus(result.status()))
}
}
async fn put_grant(&self, account_id: &str, role: &str, authorization: &str) -> Result<crate::model::Role, GrantError> {
let mut url = self.base_url.clone();
url.path_segments_mut().unwrap()
.push("v1")
.push("accounts")
.push(account_id)
.push("grants")
.push(role);
let mut headers = reqwest::header::HeaderMap::new();
headers.append("authorization", reqwest::header::HeaderValue::from_str(&format!("{authorization}"))?);
{
let headers_vec: Vec<(&str, String)> = headers.iter().map(|(k, v)| crate::hide_authorization(k, v)).collect();
tracing::info!(method="put", url=url.to_string(), headers=?headers_vec, body="<no_body>", "put_grant");
}
let mut builder = reqwest::Client::builder();
if self.allow_insecure {
builder = builder.danger_accept_invalid_certs(true);
}
let client = builder.build()?;
let result = client
.put(url)
.headers(headers)
.send()
.await?;
match result.status().as_u16() {
200 => {
let body = result.json::<crate::model::Role>().await?;
Ok(body)
}
401 => {
let body = result.json::<AccountEndpointErrorUnauthorizedPayload>().await?;
Err(GrantError::Status401 { message: body.message })
}
404 => {
let body = result.json::<AccountEndpointErrorNotFoundPayload>().await?;
Err(GrantError::Status404 { message: body.message })
}
400 => {
let body = result.json::<AccountEndpointErrorArgValidationPayload>().await?;
Err(GrantError::Status400 { errors: body.errors })
}
500 => {
let body = result.json::<AccountEndpointErrorInternalPayload>().await?;
Err(GrantError::Status500 { error: body.error })
}
_ => Err(GrantError::UnexpectedStatus(result.status()))
}
}
async fn delete_grant(&self, account_id: &str, role: &str, authorization: &str) -> Result<(), GrantError> {
let mut url = self.base_url.clone();
url.path_segments_mut().unwrap()
.push("v1")
.push("accounts")
.push(account_id)
.push("grants")
.push(role);
let mut headers = reqwest::header::HeaderMap::new();
headers.append("authorization", reqwest::header::HeaderValue::from_str(&format!("{authorization}"))?);
{
let headers_vec: Vec<(&str, String)> = headers.iter().map(|(k, v)| crate::hide_authorization(k, v)).collect();
tracing::info!(method="delete", url=url.to_string(), headers=?headers_vec, body="<no_body>", "delete_grant");
}
let mut builder = reqwest::Client::builder();
if self.allow_insecure {
builder = builder.danger_accept_invalid_certs(true);
}
let client = builder.build()?;
let result = client
.delete(url)
.headers(headers)
.send()
.await?;
match result.status().as_u16() {
200 => {
let body = ();
Ok(body)
}
401 => {
let body = result.json::<AccountEndpointErrorUnauthorizedPayload>().await?;
Err(GrantError::Status401 { message: body.message })
}
404 => {
let body = result.json::<AccountEndpointErrorNotFoundPayload>().await?;
Err(GrantError::Status404 { message: body.message })
}
400 => {
let body = result.json::<AccountEndpointErrorArgValidationPayload>().await?;
Err(GrantError::Status400 { errors: body.errors })
}
500 => {
let body = result.json::<AccountEndpointErrorInternalPayload>().await?;
Err(GrantError::Status500 { error: body.error })
}
_ => Err(GrantError::UnexpectedStatus(result.status()))
}
}
}