2019-09-10 00:56:50 +00:00
|
|
|
use hyper::{Body, Method, Request, Response, Server, StatusCode};
|
|
|
|
|
use std::error::Error as StdError;
|
|
|
|
|
|
2019-09-10 09:30:36 +00:00
|
|
|
#[derive(PartialEq, Debug, Clone)]
|
2019-09-10 05:35:54 +00:00
|
|
|
pub enum ApiError {
|
2019-09-10 00:56:50 +00:00
|
|
|
MethodNotAllowed(String),
|
|
|
|
|
ServerError(String),
|
|
|
|
|
NotImplemented(String),
|
|
|
|
|
InvalidQueryParams(String),
|
|
|
|
|
NotFound(String),
|
|
|
|
|
ImATeapot(String), // Just in case.
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
pub type ApiResult = Result<Response<Body>, ApiError>;
|
|
|
|
|
|
2019-09-10 05:35:54 +00:00
|
|
|
impl ApiError {
|
2019-09-10 09:30:36 +00:00
|
|
|
pub fn status_code(self) -> (StatusCode, String) {
|
2019-09-10 05:35:54 +00:00
|
|
|
match self {
|
2019-09-10 00:56:50 +00:00
|
|
|
ApiError::MethodNotAllowed(desc) => (StatusCode::METHOD_NOT_ALLOWED, desc),
|
|
|
|
|
ApiError::ServerError(desc) => (StatusCode::INTERNAL_SERVER_ERROR, desc),
|
|
|
|
|
ApiError::NotImplemented(desc) => (StatusCode::NOT_IMPLEMENTED, desc),
|
|
|
|
|
ApiError::InvalidQueryParams(desc) => (StatusCode::BAD_REQUEST, desc),
|
|
|
|
|
ApiError::NotFound(desc) => (StatusCode::NOT_FOUND, desc),
|
|
|
|
|
ApiError::ImATeapot(desc) => (StatusCode::IM_A_TEAPOT, desc),
|
2019-09-10 05:35:54 +00:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl Into<Response<Body>> for ApiError {
|
|
|
|
|
fn into(self) -> Response<Body> {
|
|
|
|
|
let status_code = self.status_code();
|
2019-09-10 00:56:50 +00:00
|
|
|
Response::builder()
|
2019-09-10 09:30:36 +00:00
|
|
|
.status(status_code.0)
|
|
|
|
|
.header("content-type", "text/plain")
|
|
|
|
|
.body(Body::from(status_code.1))
|
|
|
|
|
.expect("Response should always be created.")
|
2019-09-10 00:56:50 +00:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl From<store::Error> for ApiError {
|
|
|
|
|
fn from(e: store::Error) -> ApiError {
|
|
|
|
|
ApiError::ServerError(format!("Database error: {:?}", e))
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl From<types::BeaconStateError> for ApiError {
|
|
|
|
|
fn from(e: types::BeaconStateError) -> ApiError {
|
|
|
|
|
ApiError::ServerError(format!("BeaconState error: {:?}", e))
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl From<state_processing::per_slot_processing::Error> for ApiError {
|
|
|
|
|
fn from(e: state_processing::per_slot_processing::Error) -> ApiError {
|
|
|
|
|
ApiError::ServerError(format!("PerSlotProcessing error: {:?}", e))
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2019-09-10 05:35:54 +00:00
|
|
|
impl StdError for ApiError {
|
|
|
|
|
fn cause(&self) -> Option<&StdError> {
|
|
|
|
|
None
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl std::fmt::Display for ApiError {
|
|
|
|
|
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
|
2019-09-10 09:30:36 +00:00
|
|
|
let status = self.clone().status_code();
|
2019-09-10 05:35:54 +00:00
|
|
|
write!(f, "{:?}: {:?}", status.0, status.1)
|
|
|
|
|
}
|
2019-09-10 00:56:50 +00:00
|
|
|
}
|
