1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
use rocket;
use rocket::request::Form;
use rocket::response::{Flash, Redirect};
use utils;
#[derive(BartDisplay)]
#[template = "templates/users.html"]
pub struct UsersTemplate {
users: Vec<::models::User>
}
implement_responder_for!(UsersTemplate);
#[get("/", format = "text/html")]
fn index(flash: Option<rocket::request::FlashMessage>, conn: utils::DbConn) -> utils::Page<UsersTemplate> {
utils::Page {
title: String::from("Users"),
flash: flash.map_or(None, |f| Some(f.msg().to_string())),
content: UsersTemplate {
users: ::models::User::all(conn).unwrap()
}
}
}
#[derive(BartDisplay)]
#[template = "templates/new_user.html"]
pub struct NewUserTemplate {
user: ::models::NewUser
}
implement_responder_for!(NewUserTemplate);
#[get("/new", format = "text/html")]
fn new(flash: Option<rocket::request::FlashMessage>) -> utils::Page<NewUserTemplate> {
utils::Page {
title: String::from("New user"),
flash: flash.map_or(None, |f| Some(f.msg().to_string())),
content: NewUserTemplate {
user: Default::default()
}
}
}
#[derive(FromForm)]
struct RegisterUserForm {
username: String,
realname: Option<String>,
email: Option<String>,
password: String,
password_confirm: String
}
impl RegisterUserForm {
fn new_user(&self) -> Result<::models::NewUser, &'static str> {
if self.password != self.password_confirm {
Err("Passwords don't match")
}
else {
Ok(::models::NewUser {
username: self.username.clone(),
realname: self.realname.clone(),
email: self.email.clone(),
password: self.password.clone()
})
}
}
}
#[post("/create", data="<user>")]
fn create(user: Form<RegisterUserForm>, route: &rocket::Route, conn: utils::DbConn) -> Flash<Redirect> {
match user.get().new_user() {
Ok(new_user) => {
match ::models::User::create(&new_user, conn) {
Ok(_) => Flash::success(Redirect::to(route.base.path()), "User successfully created!"),
Err(_) => Flash::error(Redirect::to("/"), "Could not create user!")
}
},
Err(msg) => Flash::error(Redirect::to(&format!("{}/new", route.base.path())), &msg)
}
}
pub fn routes() -> Vec<rocket::Route> {
routes![index, new, create]
}
|