use core::fmt;
use hyper::{
header::{InvalidHeaderName, InvalidHeaderValue},
Error as RequestError,
};
use serde::Deserialize;
use std::{convert::Infallible, error::Error as StdError};
use std::{fmt::Display, result};
pub type Result<T> = result::Result<T, Error>;
#[derive(Debug)]
pub struct ValueError(String);
impl ValueError {
pub fn new<T: Into<String>>(value: T) -> Self {
Self(value.into())
}
}
impl Display for ValueError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "value error: {}", self.0)
}
}
impl StdError for ValueError {}
impl From<&str> for ValueError {
fn from(err: &str) -> Self {
Self(err.to_string())
}
}
impl From<InvalidHeaderValue> for ValueError {
fn from(err: InvalidHeaderValue) -> Self {
return ValueError(err.to_string());
}
}
impl From<InvalidHeaderName> for ValueError {
fn from(err: InvalidHeaderName) -> Self {
return ValueError(err.to_string());
}
}
impl From<Infallible> for ValueError {
fn from(err: Infallible) -> Self {
return ValueError(err.to_string());
}
}
#[derive(Debug)]
pub struct XmlError(String);
impl Display for XmlError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "xmlerror: {}", self.0)
}
}
impl StdError for XmlError {}
impl From<crate::xml::error::Error> for XmlError {
fn from(err: crate::xml::error::Error) -> Self {
Self(err.to_string())
}
}
#[derive(Debug, Deserialize)]
#[serde(rename_all = "PascalCase", rename="Error")]
pub struct S3Error {
pub code: String,
pub message: String,
#[serde(default)]
pub resource: String,
pub request_id: String,
pub host_id: Option<String>,
pub bucket_name: Option<String>,
pub object_name: Option<String>,
}
impl std::fmt::Display for S3Error {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "S3Error: {}", self.message)
}
}
impl StdError for S3Error {}
impl TryFrom<&[u8]> for S3Error {
type Error = XmlError;
fn try_from(res: &[u8]) -> std::result::Result<Self, Self::Error> {
return Ok(crate::xml::de::from_reader(res)?);
}
}
impl TryFrom<&str> for S3Error {
type Error = XmlError;
fn try_from(value: &str) -> std::result::Result<Self, Self::Error> {
value.as_bytes().try_into()
}
}
#[derive(Debug)]
pub enum Error {
ValueError(String),
RequestError(RequestError),
XmlError(XmlError),
S3Error(S3Error),
HttpError(reqwest::Error),
UnknownResponse(reqwest::Response),
MessageDecodeError(String),
SelectObejectError(String),
IoError(std::io::Error),
}
impl StdError for Error {
fn source(&self) -> Option<&(dyn std::error::Error + 'static)> {
match self {
Error::RequestError(e) => e.source(),
Error::S3Error(e) => e.source(),
_ => None,
}
}
}
#[rustfmt::skip]
impl fmt::Display for Error {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match &self {
Error::ValueError(e) => write!(f, "{}", e),
Error::RequestError(e) => write!(f, "{}", e),
Error::XmlError(e) => write!(f, "{}", e),
Error::S3Error(e) => write!(f, "{}", e),
Error::HttpError(e) => write!(f, "{}", e),
Error::UnknownResponse(e) => write!(f, "Unexpected HTTP responses, status: {}", e.status()),
Error::MessageDecodeError(e)=> write!(f, "{}", e),
Error::SelectObejectError(e)=> write!(f, "{}", e),
Error::IoError(e) => write!(f, "{}", e),
}
}
}
impl From<S3Error> for Error {
fn from(err: S3Error) -> Self {
Error::S3Error(err)
}
}
impl<T: Into<ValueError>> From<T> for Error {
fn from(err: T) -> Self {
Error::ValueError(err.into().0)
}
}
impl From<XmlError> for Error {
fn from(err: XmlError) -> Self {
Error::XmlError(err)
}
}
impl From<RequestError> for Error {
fn from(err: RequestError) -> Self {
Error::RequestError(err)
}
}
impl From<std::io::Error> for Error {
fn from(err: std::io::Error) -> Self {
std::io::Error::new(std::io::ErrorKind::UnexpectedEof, "");
Error::IoError(err)
}
}
impl From<reqwest::Error> for Error {
fn from(err: reqwest::Error) -> Self {
if err.is_builder() {
return Self::ValueError(err.to_string());
}
Self::HttpError(err)
}
}
impl From<reqwest::Response> for Error {
fn from(err: reqwest::Response) -> Self {
Self::UnknownResponse(err)
}
}
impl From<crate::xml::error::Error> for Error {
fn from(err: crate::xml::error::Error) -> Self {
Error::XmlError(err.into())
}
}
#[cfg(test)]
mod tests {
use super::S3Error;
use crate::error::XmlError;
#[test]
fn test_s3_error() {
let res = r#"<?xml version="1.0" encoding="UTF-8"?>
<Error>
<Code>NoSuchKey</Code>
<Message>The resource you requested does not exist</Message>
<Resource>/mybucket/myfoto.jpg</Resource>
<RequestId>4442587FB7D0A2F9</RequestId>
</Error>"#;
let result: std::result::Result<S3Error, XmlError> = res.as_bytes().try_into();
assert!(result.is_ok());
println!("{:?}", result);
}
}