use hyper::{Body, Request, Response, Server, StatusCode};
use routerify::{Router, RouterService};
use std::fmt;
use std::net::SocketAddr;
#[derive(Debug)]
enum ApiError {
#[allow(dead_code)]
Unauthorized,
Generic(String),
}
impl std::error::Error for ApiError {}
impl fmt::Display for ApiError {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match self {
ApiError::Unauthorized => write!(f, "Unauthorized"),
ApiError::Generic(s) => write!(f, "Generic: {}", s),
}
}
}
async fn home_handler(_: Request<Body>) -> Result<Response<Body>, ApiError> {
Err(ApiError::Generic("Something went wrong!".into()))
}
async fn error_handler(err: routerify::RouteError) -> Response<Body> {
let api_err = err.downcast::<ApiError>().unwrap();
match api_err.as_ref() {
ApiError::Unauthorized => Response::builder()
.status(StatusCode::UNAUTHORIZED)
.body(Body::empty())
.unwrap(),
ApiError::Generic(s) => Response::builder()
.status(StatusCode::INTERNAL_SERVER_ERROR)
.body(Body::from(s.to_string()))
.unwrap(),
}
}
fn router() -> Router<Body, ApiError> {
Router::builder()
.get("/", home_handler)
.err_handler(error_handler)
.build()
.unwrap()
}
#[tokio::main]
async fn main() {
let router = router();
let service = RouterService::new(router).unwrap();
let addr = SocketAddr::from(([127, 0, 0, 1], 3001));
let server = Server::bind(&addr).serve(service);
println!("App is running on: {}", addr);
if let Err(err) = server.await {
eprintln!("Server error: {}", err);
}
}