mirror of https://github.com/rwf2/Rocket.git
82 lines
1.9 KiB
Rust
82 lines
1.9 KiB
Rust
#![feature(plugin, custom_derive)]
|
|
#![plugin(rocket_macros)]
|
|
|
|
extern crate rocket;
|
|
|
|
mod files;
|
|
|
|
use rocket::Rocket;
|
|
use rocket::response::Redirect;
|
|
use rocket::form::FromFormValue;
|
|
|
|
#[derive(Debug)]
|
|
struct StrongPassword<'r>(&'r str);
|
|
|
|
#[derive(Debug)]
|
|
struct AdultAge(isize);
|
|
|
|
#[derive(FromForm)]
|
|
struct UserLogin<'r> {
|
|
username: &'r str,
|
|
password: Result<StrongPassword<'r>, &'r str>,
|
|
age: Result<AdultAge, &'r str>,
|
|
}
|
|
|
|
impl<'v> FromFormValue<'v> for StrongPassword<'v> {
|
|
type Error = &'static str;
|
|
|
|
fn parse(v: &'v str) -> Result<Self, Self::Error> {
|
|
if v.len() < 8 {
|
|
Err("Too short!")
|
|
} else {
|
|
Ok(StrongPassword(v))
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<'v> FromFormValue<'v> for AdultAge {
|
|
type Error = &'static str;
|
|
|
|
fn parse(v: &'v str) -> Result<Self, Self::Error> {
|
|
let age = match isize::parse(v) {
|
|
Ok(v) => v,
|
|
Err(_) => return Err("Age value is not a number.")
|
|
};
|
|
|
|
match age > 20 {
|
|
true => Ok(AdultAge(age)),
|
|
false => Err("Must be at least 21.")
|
|
}
|
|
}
|
|
}
|
|
|
|
#[route(POST, path = "/login", form = "<user>")]
|
|
fn login(user: UserLogin) -> Result<Redirect, String> {
|
|
if user.age.is_err() {
|
|
return Err(String::from(user.age.unwrap_err()));
|
|
}
|
|
|
|
if user.password.is_err() {
|
|
return Err(String::from(user.password.unwrap_err()));
|
|
}
|
|
|
|
match user.username {
|
|
"Sergio" => match user.password.unwrap().0 {
|
|
"password" => Ok(Redirect::other("/user/Sergio")),
|
|
_ => Err("Wrong password!".to_string())
|
|
},
|
|
_ => Err(format!("Unrecognized user, '{}'.", user.username))
|
|
}
|
|
}
|
|
|
|
#[route(GET, path = "/user/<username>")]
|
|
fn user_page(username: &str) -> String {
|
|
format!("This is {}'s page.", username)
|
|
}
|
|
|
|
fn main() {
|
|
let mut rocket = Rocket::new("localhost", 8000);
|
|
rocket.mount("/", routes![files::index, files::files, user_page, login]);
|
|
rocket.launch();
|
|
}
|