48 lines
1.6 KiB
Rust
48 lines
1.6 KiB
Rust
use actix_web::{cookie::time::error, http::StatusCode, HttpResponse, ResponseError};
|
|
use sea_orm::TransactionError;
|
|
use thiserror::Error;
|
|
|
|
#[derive(Error, Debug)]
|
|
pub enum ApiError {
|
|
#[error("Database Error: {0}")]
|
|
Database(#[from] sea_orm::DbErr),
|
|
#[error("Unauthorized")]
|
|
Unauthorized,
|
|
#[error("Not Found")]
|
|
NotFound,
|
|
#[error("Bad Request: {0}")]
|
|
BadRequest(String),
|
|
#[error("Validation Error: {0}")]
|
|
ValidationError(#[from] validator::ValidationErrors),
|
|
#[error("Argon2 Error: {0}")]
|
|
Argon2Error(String),
|
|
#[error("Session insert error: {0}")]
|
|
SessionInsertError(#[from] actix_session::SessionInsertError),
|
|
}
|
|
|
|
impl ResponseError for ApiError {
|
|
fn status_code(&self) -> StatusCode {
|
|
match self {
|
|
ApiError::Database(..) => StatusCode::INTERNAL_SERVER_ERROR,
|
|
ApiError::NotFound => StatusCode::NOT_FOUND,
|
|
ApiError::Unauthorized => StatusCode::UNAUTHORIZED,
|
|
ApiError::BadRequest(..) => StatusCode::BAD_REQUEST,
|
|
ApiError::ValidationError(..) => StatusCode::BAD_REQUEST,
|
|
ApiError::Argon2Error(..) => StatusCode::INTERNAL_SERVER_ERROR,
|
|
ApiError::SessionInsertError(..) => StatusCode::INTERNAL_SERVER_ERROR,
|
|
}
|
|
}
|
|
|
|
fn error_response(&self) -> HttpResponse {
|
|
HttpResponse::build(self.status_code()).body(self.to_string())
|
|
}
|
|
}
|
|
|
|
impl From<TransactionError<sea_orm::DbErr>> for ApiError {
|
|
fn from(value: TransactionError<sea_orm::DbErr>) -> Self {
|
|
Self::Database(match value {
|
|
TransactionError::Connection(e) => e,
|
|
TransactionError::Transaction(e) => e,
|
|
})
|
|
}
|
|
}
|