AstralRinth/apps/daedalus_client/src/error.rs

use tracing_error::InstrumentError;

#[derive(thiserror::Error, Debug)]
pub enum ErrorKind {
    #[error("Daedalus Error: {0}")]
    Daedalus(#[from] daedalus::Error),
    #[error("Invalid input: {0}")]
    InvalidInput(String),
    #[error("Error while managing asynchronous tasks")]
    TaskError(#[from] tokio::task::JoinError),
    #[error("Error while deserializing JSON: {0}")]
    SerdeJSON(#[from] serde_json::Error),
    #[error("Error while deserializing XML: {0}")]
    SerdeXML(#[from] serde_xml_rs::Error),
    #[error(
        "Failed to validate file checksum at url {url} with hash {hash} after {tries} tries"
    )]
    ChecksumFailure {
        hash: String,
        url: String,
        tries: u32,
    },
    #[error("Unable to fetch {item}")]
    Fetch { inner: reqwest::Error, item: String },
    #[error("Error while uploading file to S3: {file}")]
    S3 {
        inner: Box<s3::error::S3Error>,
        file: String,
    },
    #[error("Error acquiring semaphore: {0}")]
    Acquire(#[from] tokio::sync::AcquireError),
    #[error("Tracing error: {0}")]
    Tracing(#[from] tracing::subscriber::SetGlobalDefaultError),
    #[error("Zip error: {0}")]
    Zip(#[from] async_zip::error::ZipError),
}

#[derive(Debug)]
pub struct Error {
    pub source: tracing_error::TracedError<ErrorKind>,
}

impl std::fmt::Display for Error {
    fn fmt(&self, fmt: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(fmt, "{}", self.source)
    }
}

impl<E: Into<ErrorKind>> From<E> for Error {
    fn from(source: E) -> Self {
        let error = Into::<ErrorKind>::into(source);

        Self {
            source: error.in_current_span(),
        }
    }
}

impl ErrorKind {
    pub fn as_error(self) -> Error {
        self.into()
    }
}

pub type Result<T> = core::result::Result<T, Error>;