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,
}

#[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.set_path(&format!("v1/accounts/{account_id}/grants"));

        let mut headers = reqwest::header::HeaderMap::new();
        headers.append("authorization", reqwest::header::HeaderValue::from_str(&format!("{authorization}"))?);
        let result = reqwest::Client::builder()
            .build()?
            .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.set_path(&format!("v1/accounts/{account_id}/grants/{role}"));

        let mut headers = reqwest::header::HeaderMap::new();
        headers.append("authorization", reqwest::header::HeaderValue::from_str(&format!("{authorization}"))?);
        let result = reqwest::Client::builder()
            .build()?
            .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.set_path(&format!("v1/accounts/{account_id}/grants/{role}"));

        let mut headers = reqwest::header::HeaderMap::new();
        headers.append("authorization", reqwest::header::HeaderValue::from_str(&format!("{authorization}"))?);
        let result = reqwest::Client::builder()
            .build()?
            .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.set_path(&format!("v1/accounts/{account_id}/grants/{role}"));

        let mut headers = reqwest::header::HeaderMap::new();
        headers.append("authorization", reqwest::header::HeaderValue::from_str(&format!("{authorization}"))?);
        let result = reqwest::Client::builder()
            .build()?
            .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()))
        }
    }
}