jrd/spectrum
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Handle POST requests

Browse Source
This commit is contained in:
Jared Burce
2020-03-18 17:13:05 -07:00
parent 6684075cc5
commit 161a042123
3 changed files with 49 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
Cargo.toml
View File

@@ -6,6 +6,7 @@ edition = "2018"
[dependencies] [dependencies]
futures = "0.3" futures = "0.3"
log = "0.4"
pretty_env_logger = "0.4" pretty_env_logger = "0.4"
tokio = { version = "0.2", features = ["macros"] } tokio = { version = "0.2", features = ["macros"] }
warp = "0.2" warp = "0.2"

47
src/main.rs
View File

@@ -1,14 +1,51 @@
use warp::Filter; use std::collections::HashMap;
#[tokio::main(basic_scheduler)] use log::info;
use warp::{Filter, Rejection, Reply};
use warp::http::StatusCode;
#[tokio::main]
async fn main() { async fn main() {
pretty_env_logger::init(); pretty_env_logger::init();
let root = warp::get() // GET /
.and(warp::path::end()) let root = warp::path::end()
.and(warp::get())
.and(warp::fs::file("./static/index.html")); .and(warp::fs::file("./static/index.html"));
let routes = root; // POST /name name={name}
let namechange = warp::path("name")
.and(warp::path::end())
.and(warp::post())
.and(warp::body::content_length_limit(1024 * 16))
.and(warp::body::form())
.and_then(|form: HashMap<String, String>| async move {
let name = match form.get("name") {
None => return Err(warp::reject::custom(BadName)),
Some(name) if name.is_empty() ||
name.chars().count() > 32 => return Err(warp::reject::custom(BadName)),
Some(name) => name
};
info!("POST /name as \"{}\"", name);
Ok(warp::reply::with_header(StatusCode::SEE_OTHER,
warp::http::header::LOCATION,
"/"))
})
.recover(handle_reject);
let routes = root
.or(namechange);
warp::serve(routes).run(([127, 0, 0, 1], 8060)).await; warp::serve(routes).run(([127, 0, 0, 1], 8060)).await;
} }
#[derive(Debug)]
struct BadName;
impl warp::reject::Reject for BadName {}
async fn handle_reject(err: Rejection) -> Result<impl Reply, Rejection> {
match err.find() {
Some(BadName) => Ok(warp::reply::with_status(warp::reply(), StatusCode::BAD_REQUEST)),
_ => Err(err)
}
}

8
static/index.html
View File

@@ -1,9 +1,13 @@
<!DOCTYPE html> <!DOCTYPE html>
<html> <html lang="en">
<head> <head>
<meta http-equiv="Content-Type" content="text/html; charset=utf-8"/>
<title>Hello, World!</title> <title>Hello, World!</title>
</head> </head>
<body> <body>
Hello, Warp! <form action="name" method="post">
<input name="name" type="text">
<input type="submit" value="Set Name">
</form>
</body> </body>
</html> </html>