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,
content: UsersTemplate {
users: ::models::User::all(conn).unwrap()
}
}
}
#[derive(BartDisplay)]
#[template = "templates/edit_user.html"]
pub struct EditUserTemplate {
user: ::models::User
}
implement_responder_for!(EditUserTemplate);
#[get("/<id>/edit")]
fn edit(id: i32, conn: utils::DbConn) -> utils::Page<EditUserTemplate> {
utils::Page {
title: String::from("Edit users"),
flash: None,
content: EditUserTemplate {
user: ::models::User::by_id(id, &conn).unwrap()
}
}
}
#[get("/<id>/delete")]
fn delete(id: i32, route: &rocket::Route, conn: utils::DbConn) -> Flash<Redirect> {
match ::models::User::delete(id, conn) {
Ok(user) => Flash::success(
Redirect::to(route.base.path()),
format!("User {} was successfully deleted.", user.username)
),
Err(msg) => Flash::error(
Redirect::to(route.base.path()),
format!("Error deleting user: {}", msg)
)
}
}
#[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,
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()
})
}
}
fn attrs(&self) -> Result<::models::User, &'static str> {
if self.password != self.password_confirm {
Err("Passwords don't match")
}
else {
Ok(::models::User {
id: Default::default(),
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)
}
}
#[post("/<id>/update", data="<user_form>")]
fn update(id: i32, user_form: Form<RegisterUserForm>, route: &rocket::Route, conn: utils::DbConn) -> Flash<Redirect> {
match user_form.get().attrs() {
Ok(attrs) => {
if let Ok(mut user) = ::models::User::by_id(id, &conn) {
if user.username != attrs.username {
user.username = attrs.username.clone();
}
if user.realname != attrs.realname {
user.realname = attrs.realname.clone();
}
if user.email != attrs.email {
user.email = attrs.email.clone();
}
if !attrs.password.is_empty() && user.password != attrs.password {
user.password = attrs.password.clone();
}
match ::models::User::update(&user, &conn) {
Ok(user) => Flash::success(
Redirect::to(route.base.path()),
format!("User {} updated successfully", user.username)
),
Err(msg) => Flash::error(
Redirect::to(route.base.path()),
format!("User {} was not updated! {}", user.username, msg)
)
}
}
else {
Flash::error(
Redirect::to(route.base.path()),
"Can't delete users which does not exist"
)
}
},
Err(msg) => Flash::error(
Redirect::to(route.base.path()),
format!("Invalid form data: {}", msg)
)
}
}
pub fn routes() -> Vec<rocket::Route> {
routes![index, new, create, edit, update, delete]
}