97 lines
2.6 KiB
Rust
97 lines
2.6 KiB
Rust
use std::{error::Error, fmt, io};
|
|
|
|
use axum::{
|
|
http::StatusCode,
|
|
response::{IntoResponse, Response},
|
|
};
|
|
|
|
pub type Result<T> = std::result::Result<T, ServerError>;
|
|
|
|
#[derive(Debug)]
|
|
pub enum ServerError {
|
|
IO(io::Error),
|
|
Axum(axum::Error),
|
|
Db(sea_orm::DbErr),
|
|
Status(StatusCode),
|
|
Archive(libarchive::error::ArchiveError),
|
|
Figment(figment::Error),
|
|
Unit,
|
|
}
|
|
|
|
impl fmt::Display for ServerError {
|
|
fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
match self {
|
|
ServerError::IO(err) => write!(fmt, "{}", err),
|
|
ServerError::Axum(err) => write!(fmt, "{}", err),
|
|
ServerError::Status(status) => write!(fmt, "{}", status),
|
|
ServerError::Db(err) => write!(fmt, "{}", err),
|
|
ServerError::Archive(err) => write!(fmt, "{}", err),
|
|
ServerError::Figment(err) => write!(fmt, "{}", err),
|
|
ServerError::Unit => Ok(()),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Error for ServerError {}
|
|
|
|
impl IntoResponse for ServerError {
|
|
fn into_response(self) -> Response {
|
|
tracing::error!("{:?}", self);
|
|
|
|
match self {
|
|
ServerError::IO(_) => StatusCode::INTERNAL_SERVER_ERROR.into_response(),
|
|
ServerError::Axum(_) => StatusCode::INTERNAL_SERVER_ERROR.into_response(),
|
|
ServerError::Status(status) => status.into_response(),
|
|
ServerError::Db(sea_orm::DbErr::RecordNotFound(_)) => {
|
|
StatusCode::NOT_FOUND.into_response()
|
|
}
|
|
ServerError::Db(_)
|
|
| ServerError::Archive(_)
|
|
| ServerError::Figment(_)
|
|
| ServerError::Unit => StatusCode::INTERNAL_SERVER_ERROR.into_response(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl From<io::Error> for ServerError {
|
|
fn from(err: io::Error) -> Self {
|
|
ServerError::IO(err)
|
|
}
|
|
}
|
|
|
|
impl From<axum::Error> for ServerError {
|
|
fn from(err: axum::Error) -> Self {
|
|
ServerError::Axum(err)
|
|
}
|
|
}
|
|
|
|
impl From<tokio::task::JoinError> for ServerError {
|
|
fn from(err: tokio::task::JoinError) -> Self {
|
|
ServerError::IO(err.into())
|
|
}
|
|
}
|
|
|
|
impl From<StatusCode> for ServerError {
|
|
fn from(status: StatusCode) -> Self {
|
|
Self::Status(status)
|
|
}
|
|
}
|
|
|
|
impl From<sea_orm::DbErr> for ServerError {
|
|
fn from(err: sea_orm::DbErr) -> Self {
|
|
ServerError::Db(err)
|
|
}
|
|
}
|
|
|
|
impl From<libarchive::error::ArchiveError> for ServerError {
|
|
fn from(err: libarchive::error::ArchiveError) -> Self {
|
|
ServerError::Archive(err)
|
|
}
|
|
}
|
|
|
|
impl From<figment::Error> for ServerError {
|
|
fn from(err: figment::Error) -> Self {
|
|
ServerError::Figment(err)
|
|
}
|
|
}
|