use hyper::http::request::Parts; use hyper::http::response::Builder; use hyper::service::{make_service_fn, service_fn}; use hyper::{Body, Method, Server, StatusCode}; use serde::Deserialize; use sputnik::hyper_body::{FormError, SputnikBody}; use sputnik::{html_escape, mime, request::SputnikParts, response::SputnikBuilder}; use std::convert::Infallible; type Response = hyper::Response; #[derive(thiserror::Error, Debug)] enum Error { #[error("page not found")] NotFound(String), #[error("{0}")] FormError(#[from] FormError), } fn render_error(err: Error) -> (StatusCode, String) { match err { Error::NotFound(msg) => (StatusCode::NOT_FOUND, msg), Error::FormError(err) => (StatusCode::BAD_REQUEST, err.to_string()), } } async fn route(req: &mut Parts, body: Body) -> Result { match (&req.method, req.uri.path()) { (&Method::GET, "/form") => Ok(get_form(req)), (&Method::POST, "/form") => post_form(req, body).await, _ => return Err(Error::NotFound("page not found".to_owned())), } } fn get_form(_req: &mut Parts) -> Response { Builder::new() .content_type(mime::TEXT_HTML) .body("
".into()) .unwrap() } #[derive(Deserialize)] struct FormData { text: String, } async fn post_form(_req: &mut Parts, body: Body) -> Result { let msg: FormData = body.into_form().await?; Ok(Builder::new() .content_type(mime::TEXT_HTML) .body(format!("hello {}", html_escape(msg.text)).into()) .unwrap()) } async fn service( req: hyper::Request, ) -> Result, Infallible> { let (mut parts, body) = req.into_parts(); match route(&mut parts, body).await { Ok(mut res) => { for (k, v) in parts.response_headers().iter() { res.headers_mut().append(k, v.clone()); } Ok(res) } Err(err) => { let (code, message) = render_error(err); // you can easily wrap or log errors here Ok(hyper::Response::builder() .status(code) .body(message.into()) .unwrap()) } } } #[tokio::main] async fn main() { let service = make_service_fn(move |_| async move { Ok::<_, hyper::Error>(service_fn(move |req| service(req))) }); let addr = ([127, 0, 0, 1], 8000).into(); let server = Server::bind(&addr).serve(service); println!("Listening on http://{}", addr); server.await; }