use err_derive::Error; use hyper::StatusCode; use hyper::{Body, Response}; use garage_util::error::Error as GarageError; #[derive(Debug, Error)] pub enum Error { // Category: internal error #[error(display = "Internal error: {}", _0)] InternalError(#[error(source)] GarageError), #[error(display = "Internal error (Hyper error): {}", _0)] Hyper(#[error(source)] hyper::Error), #[error(display = "Internal error (HTTP error): {}", _0)] HTTP(#[error(source)] http::Error), // Category: cannot process #[error(display = "Forbidden: {}", _0)] Forbidden(String), #[error(display = "Not found")] NotFound, // Category: bad request #[error(display = "Invalid UTF-8: {}", _0)] InvalidUTF8(#[error(source)] std::str::Utf8Error), #[error(display = "Invalid XML: {}", _0)] InvalidXML(#[error(source)] roxmltree::Error), #[error(display = "Invalid header value: {}", _0)] InvalidHeader(#[error(source)] hyper::header::ToStrError), #[error(display = "Invalid HTTP range: {:?}", _0)] InvalidRange(#[error(from)] http_range::HttpRangeParseError), #[error(display = "Bad request: {}", _0)] BadRequest(String), } impl Error { pub fn http_status_code(&self) -> StatusCode { match self { Error::NotFound => StatusCode::NOT_FOUND, Error::Forbidden(_) => StatusCode::FORBIDDEN, Error::InternalError(GarageError::RPC(_)) => StatusCode::SERVICE_UNAVAILABLE, Error::InternalError(_) | Error::Hyper(_) | Error::HTTP(_) => { StatusCode::INTERNAL_SERVER_ERROR } _ => StatusCode::BAD_REQUEST, } } pub fn into_http_response(&self) -> Response { let body: Body = Body::from(format!("{}\n", self)); let mut http_error = Response::new(body); *http_error.status_mut() = self.http_status_code(); http_error } } pub trait OkOrBadRequest { type S2; fn ok_or_bad_request(self, reason: &'static str) -> Self::S2; } impl OkOrBadRequest for Result where E: std::fmt::Display, { type S2 = Result; fn ok_or_bad_request(self, reason: &'static str) -> Result { match self { Ok(x) => Ok(x), Err(e) => Err(Error::BadRequest(format!("{}: {}", reason, e))), } } } impl OkOrBadRequest for Option { type S2 = Result; fn ok_or_bad_request(self, reason: &'static str) -> Result { match self { Some(x) => Ok(x), None => Err(Error::BadRequest(format!("{}", reason))), } } } pub trait OkOrInternalError { type S2; fn ok_or_internal_error(self, reason: &'static str) -> Self::S2; } impl OkOrInternalError for Result where E: std::fmt::Display, { type S2 = Result; fn ok_or_internal_error(self, reason: &'static str) -> Result { match self { Ok(x) => Ok(x), Err(e) => Err(Error::InternalError(GarageError::Message(format!( "{}: {}", reason, e )))), } } } impl OkOrInternalError for Option { type S2 = Result; fn ok_or_internal_error(self, reason: &'static str) -> Result { match self { Some(x) => Ok(x), None => Err(Error::InternalError(GarageError::Message(format!( "{}", reason )))), } } }