use std::any;
use std::error::Error;
use thiserror::Error;
use crate::api::paged::PaginationError;
use crate::auth::AuthError;
#[derive(Debug, Error)]
#[non_exhaustive]
pub enum BodyError {
#[error("failed to URL encode form parameters: {}", source)]
UrlEncoded {
#[from]
source: serde_urlencoded::ser::Error,
},
}
#[derive(Debug, Error)]
#[non_exhaustive]
pub enum ApiError<E>
where
E: Error + Send + Sync + 'static,
{
#[error("missing auth error: {}", source)]
Auth {
source: AuthError,
},
#[error("pagination error: {}", source)]
Pagination {
source: PaginationError,
},
#[error("client error: {}", source)]
Client {
source: E,
},
#[error("failed to parse URL: {}", source)]
UrlParse {
#[from]
source: url::ParseError,
},
#[error("failed to create request body: {}", source)]
Body {
#[from]
source: BodyError,
},
#[error("could not parse JSON response: {}", source)]
Json {
#[from]
source: serde_json::Error,
},
#[error("marketstack server error: {}", msg)]
Marketstack {
msg: String,
},
#[error("marketstack internal server error: {}", status)]
MarketstackService {
status: http::StatusCode,
data: Vec<u8>,
},
#[error("marketstack server error: {:?}", obj)]
MarketstackObject {
obj: serde_json::Value,
},
#[error("marketstack server error: {:?}", obj)]
MarketstackUnrecognized {
obj: serde_json::Value,
},
#[error("could not parse {} data from JSON: {}", typename, source)]
DataType {
source: serde_json::Error,
typename: &'static str,
},
}
impl<E> ApiError<E>
where
E: Error + Send + Sync + 'static,
{
pub fn client(source: E) -> Self {
ApiError::Client { source }
}
pub fn map_client<F, W>(self, f: F) -> ApiError<W>
where
F: FnOnce(E) -> W,
W: Error + Send + Sync + 'static,
{
match self {
Self::Auth { source } => ApiError::Auth { source },
Self::Client { source } => ApiError::client(f(source)),
Self::UrlParse { source } => ApiError::UrlParse { source },
Self::Body { source } => ApiError::Body { source },
Self::Json { source } => ApiError::Json { source },
Self::Marketstack { msg } => ApiError::Marketstack { msg },
Self::MarketstackService { status, data } => {
ApiError::MarketstackService { status, data }
}
Self::MarketstackObject { obj } => ApiError::MarketstackObject { obj },
Self::MarketstackUnrecognized { obj } => ApiError::MarketstackUnrecognized { obj },
Self::DataType { source, typename } => ApiError::DataType { source, typename },
Self::Pagination { source } => ApiError::Pagination { source },
}
}
pub(crate) fn auth_error() -> Self {
Self::Auth {
source: AuthError::MissingAuth,
}
}
pub(crate) fn server_error(status: http::StatusCode, body: &bytes::Bytes) -> Self {
Self::MarketstackService {
status,
data: body.into_iter().copied().collect(),
}
}
pub(crate) fn from_marketstack(value: serde_json::Value) -> Self {
let error_value = value
.pointer("/message")
.or_else(|| value.pointer("/error"));
if let Some(error_value) = error_value {
if let Some(msg) = error_value.as_str() {
ApiError::Marketstack { msg: msg.into() }
} else {
ApiError::MarketstackObject {
obj: error_value.clone(),
}
}
} else {
ApiError::MarketstackUnrecognized { obj: value }
}
}
pub(crate) fn data_type<T>(source: serde_json::Error) -> Self {
ApiError::DataType {
source,
typename: any::type_name::<T>(),
}
}
}
#[cfg(test)]
mod tests {
use core::panic;
use serde_json::json;
use thiserror::Error;
use crate::api::ApiError;
#[derive(Debug, Error)]
#[error("my error")]
enum MyError {}
#[test]
fn marketstack_error_error() {
let obj = json! {{
"error": "error contents"
}};
let err: ApiError<MyError> = ApiError::from_marketstack(obj);
if let ApiError::Marketstack { msg } = err {
assert_eq!(msg, "error contents");
} else {
panic!("unexpected error: {}", err);
}
}
#[test]
fn marketstack_error_message_string() {
let obj = json!({
"message": "error contents"
});
let err: ApiError<MyError> = ApiError::from_marketstack(obj);
if let ApiError::Marketstack { msg } = err {
assert_eq!(msg, "error contents");
} else {
panic!("unexpected error: {}", err);
}
}
#[test]
fn marketstack_error_object() {
let err_obj = json!({
"blah": "foo",
});
let obj = json!({
"message": err_obj
});
let err: ApiError<MyError> = ApiError::from_marketstack(obj);
if let ApiError::MarketstackObject { obj } = err {
assert_eq!(obj, err_obj);
} else {
panic!("unexpected error: {}", err);
}
}
#[test]
fn marketstack_error_message_unrecognized() {
let err_obj = json!({
"some_weird_key": "an even weirder value",
});
let err: ApiError<MyError> = ApiError::from_marketstack(err_obj.clone());
if let ApiError::MarketstackUnrecognized { obj } = err {
assert_eq!(obj, err_obj);
} else {
panic!("unexpected error: {}", err);
}
}
}