use axum::http::StatusCode; use axum::response::{IntoResponse, Response}; use export::data::{AppResponseData, NoData}; use thiserror::Error; pub type SpaceResult = Result; #[derive(Debug, Error)] pub enum SpaceError { #[error("bad request: {0}")] BadRequest(String), #[error("unauthorized: {0}")] Unauthorized(String), #[error("forbidden: {0}")] Forbidden(String), #[error("not found: {0}")] NotFound(String), #[error("conflict: {0}")] Conflict(String), #[error("rate limited: {0}")] RateLimited(String), #[error("internal error: {0}")] Internal(String), #[error("database error: {0}")] Sqlx(#[from] db::conn::SqlxError), #[error("uuid: {0}")] Uuid(#[from] uuid::Error), } impl From for SpaceError { fn from(re: axum::extract::rejection::JsonRejection) -> Self { SpaceError::BadRequest(re.body_text()) } } impl From for SpaceError { fn from(re: axum::extract::rejection::PathRejection) -> Self { SpaceError::BadRequest(re.body_text()) } } impl From for SpaceError { fn from(re: axum::extract::rejection::QueryRejection) -> Self { SpaceError::BadRequest(re.body_text()) } } impl SpaceError { fn status(&self) -> StatusCode { match self { SpaceError::BadRequest(_) => StatusCode::BAD_REQUEST, SpaceError::Forbidden(_) => StatusCode::FORBIDDEN, SpaceError::Conflict(_) => StatusCode::CONFLICT, SpaceError::Unauthorized(_) => StatusCode::UNAUTHORIZED, SpaceError::NotFound(_) => StatusCode::NOT_FOUND, SpaceError::RateLimited(_) => StatusCode::TOO_MANY_REQUESTS, SpaceError::Internal(_) | SpaceError::Sqlx(_) | SpaceError::Uuid(_) => { StatusCode::INTERNAL_SERVER_ERROR } } } fn is_user_facing(&self) -> bool { matches!( self, SpaceError::BadRequest(_) | SpaceError::Unauthorized(_) | SpaceError::Forbidden(_) | SpaceError::NotFound(_) | SpaceError::Conflict(_) | SpaceError::RateLimited(_) ) } pub fn error_kind(&self) -> &'static str { match self { SpaceError::BadRequest(_) => "bad_request", SpaceError::Unauthorized(_) => "unauthorized", SpaceError::Forbidden(_) => "forbidden", SpaceError::NotFound(_) => "not_found", SpaceError::Conflict(_) => "conflict", SpaceError::RateLimited(_) => "rate_limited", SpaceError::Internal(_) => "internal_other", SpaceError::Sqlx(_) => "internal_db", SpaceError::Uuid(_) => "internal_uuid", } } } impl IntoResponse for SpaceError { fn into_response(self) -> Response { let status = self.status(); let msg = if self.is_user_facing() { self.to_string() } else { tracing::error!(error = %self, "space internal error"); "internal error".to_string() }; AppResponseData::::err_plain(status.as_u16() as i32, msg).into_response() } }