use std::{convert::Infallible, net::SocketAddr, sync::Arc};
use hyper::{
Request,
body::Incoming,
server::conn::{http1, http2},
service::service_fn,
};
use hyper_util::rt::{TokioExecutor, TokioIo};
use tokio::{
io::{AsyncRead, AsyncWrite},
net::TcpListener,
};
use crate::{
client::Client,
group::Group,
middleware::error::DefaultErrorHandler,
request::RequestBody,
response::ResponseWriter,
router::Router,
types::{BoltError, ErrorHandler, Handler, Method, Middleware, Mode},
};
pub mod client;
mod group;
pub mod macros;
pub mod middleware;
pub mod request;
pub mod response;
mod router;
pub mod types;
pub use bolt_web_macro::main;
pub use num_cpus;
pub use paste;
pub use tokio;
trait Io: AsyncRead + AsyncWrite + Unpin {}
impl<T: AsyncRead + AsyncWrite + Unpin> Io for T {}
#[allow(dead_code)]
pub struct Bolt {
router: Router,
error_handler: Arc<dyn ErrorHandler>,
client: Client,
}
#[allow(unused_variables)]
#[allow(dead_code)]
impl Bolt {
pub fn new() -> Self {
Self {
router: Router::new(),
error_handler: Arc::new(DefaultErrorHandler),
client: Client::new(),
}
}
pub async fn run(
&self,
addr: &str,
mode: Mode,
tls: Option<(&str, &str)>,
) -> Result<(), BoltError> {
println!("⚡ A high performance & minimalist web framework in rust.");
println!(
r#"
__ ____
/ /_ ____ / / /_
/ __ \/ __ \/ / __/
/ /_/ / /_/ / / /_
/_.___/\____/_/\__/ v0.2
"#
);
println!(
">> Server running on {}://{}",
if tls.is_some() { "https" } else { "http" },
addr
);
let addr: SocketAddr = addr.parse().unwrap();
let listener = TcpListener::bind(addr).await?;
let router = Arc::new(self.router.clone());
let tls_acceptor = if let Some((pkcs12_path, password)) = tls {
let pkcs12 = std::fs::read(pkcs12_path)?;
let identity = tokio_native_tls::native_tls::Identity::from_pkcs12(&pkcs12, password)?;
Some(Arc::new(tokio_native_tls::TlsAcceptor::from(
tokio_native_tls::native_tls::TlsAcceptor::builder(identity).build()?,
)))
} else {
None
};
loop {
let (stream, _) = listener.accept().await?;
let io: Box<dyn Io + Send> = if let Some(ref acceptor) = tls_acceptor {
Box::new(acceptor.accept(stream).await?)
} else {
Box::new(stream)
};
let io = TokioIo::new(io);
let router = router.clone();
let error_handler = self.error_handler.clone();
let service = service_fn(move |req: Request<Incoming>| {
let router = router.clone();
let error_handler = error_handler.clone();
async move {
let mut req_body = RequestBody::new(req);
let mut res_body = ResponseWriter::new();
let method = match *req_body.method() {
hyper::Method::GET => Method::GET,
hyper::Method::POST => Method::POST,
hyper::Method::PUT => Method::PUT,
hyper::Method::PATCH => Method::PATCH,
hyper::Method::DELETE => Method::DELETE,
hyper::Method::OPTIONS => Method::OPTIONS,
hyper::Method::HEAD => Method::HEAD,
hyper::Method::TRACE => Method::TRACE,
_ => {
res_body.status(405);
res_body.send("Method Not Allowed");
return Ok::<_, Infallible>(res_body.into_response());
}
};
let path = req_body.path().to_string();
let mws = router.collect_middleware(&path, method);
for mw in mws {
mw.run(&mut req_body, &mut res_body).await;
if res_body.has_error() {
break;
}
}
if !res_body.has_error() {
if let Some((handler, params)) = router.find(&path, method) {
req_body.set_params(params);
handler.handle(&mut req_body, &mut res_body).await;
} else {
let method_str = match *req_body.method() {
hyper::Method::GET => "GET",
hyper::Method::POST => "POST",
hyper::Method::PUT => "PUT",
hyper::Method::PATCH => "PATCH",
hyper::Method::DELETE => "DELETE",
hyper::Method::OPTIONS => "OPTIONS",
hyper::Method::HEAD => "HEAD",
hyper::Method::TRACE => "TRACE",
_ => "UNKNOWN",
};
res_body.error(404, &format!("Not Found {} {}", method_str, path));
}
}
if res_body.has_error() {
let msg = res_body.body.clone();
error_handler.run(msg, &mut res_body).await;
}
req_body.cleanup().await;
if req_body.log {
println!(
"[LOG] method={} path={} status={}",
req_body.method(),
path,
res_body.status,
);
}
res_body.strip_header("X-Internal-Request-Start");
Ok::<_, Infallible>(res_body.into_response())
}
});
match mode {
Mode::Http1 => {
tokio::task::spawn(async move {
if let Err(err) = http1::Builder::new().serve_connection(io, service).await
{
eprintln!("Error serving connection: {}", err);
}
});
}
Mode::Http2 => {
tokio::task::spawn(async move {
if let Err(err) = http2::Builder::new(TokioExecutor::new())
.serve_connection(io, service)
.await
{
eprintln!("Error serving connection: {}", err);
}
});
}
}
}
}
fn add_route<H>(&mut self, method: Method, path: &str, handler: H)
where
H: Handler + 'static,
{
self.router.insert(path, method, handler);
}
pub fn get<H>(&mut self, path: &str, handler: H)
where
H: Handler + 'static,
{
self.add_route(Method::GET, path, handler);
}
pub fn post<H>(&mut self, path: &str, handler: H)
where
H: Handler + 'static,
{
self.add_route(Method::POST, path, handler);
}
pub fn put<H>(&mut self, path: &str, handler: H)
where
H: Handler + 'static,
{
self.add_route(Method::PUT, path, handler);
}
pub fn patch<H>(&mut self, path: &str, handler: H)
where
H: Handler + 'static,
{
self.add_route(Method::PATCH, path, handler);
}
pub fn delete<H>(&mut self, path: &str, handler: H)
where
H: Handler + 'static,
{
self.add_route(Method::DELETE, path, handler);
}
pub fn group<'a>(&'a mut self, path: &str) -> Group<'a> {
Group {
prefix: path.to_string(),
app: self,
}
}
pub fn middleware<M>(&mut self, path: &str, method: Option<Method>, middleware_fn: M)
where
M: Middleware + 'static,
{
let mw = Arc::new(middleware_fn);
let full_path = path.to_string();
match method {
Some(m) => self.router.insert_middleware(&full_path, m, mw),
None => {
for m in [
Method::GET,
Method::POST,
Method::PUT,
Method::PATCH,
Method::DELETE,
Method::OPTIONS,
Method::HEAD,
Method::TRACE,
] {
self.router.insert_middleware(&full_path, m, mw.clone());
}
}
}
}
pub fn set_error_handler(&mut self, handler: Arc<dyn ErrorHandler>) {
self.error_handler = handler;
}
}