use rocket::{form::*, get, post, response::Redirect, routes, State};
use rocket_auth::{prelude::Error, *};
use rocket_dyn_templates::Template;
use serde_json::json;
use sqlx::*;
use std::result::Result;
use std::*;
#[get("/login")]
fn get_login() -> Template {
Template::render("login", json!({}))
}
#[post("/login", data = "<form>")]
async fn post_login(auth: Auth<'_>, form: Form<Login>) -> Result<Redirect, Error> {
let result = auth.login(&form).await;
println!("login attempt: {:?}", result);
result?;
Ok(Redirect::to("/"))
}
#[get("/signup")]
async fn get_signup() -> Template {
Template::render("signup", json!({}))
}
#[post("/signup", data = "<form>")]
async fn post_signup(auth: Auth<'_>, form: Form<Signup>) -> Result<Redirect, Error> {
auth.signup(&form).await?;
auth.login(&form.into()).await?;
Ok(Redirect::to("/"))
}
#[get("/")]
async fn index(user: Option<User>) -> Template {
Template::render("index", json!({ "user": user }))
}
#[get("/logout")]
fn logout(auth: Auth<'_>) -> Result<Template, Error> {
auth.logout()?;
Ok(Template::render("logout", json!({})))
}
#[get("/delete")]
async fn delete(auth: Auth<'_>) -> Result<Template, Error> {
auth.delete().await?;
Ok(Template::render("deleted", json!({})))
}
#[get("/show_all_users")]
async fn show_all_users(conn: &State<MySqlPool>, user: Option<User>) -> Result<Template, Error> {
let users: Vec<User> = query_as("select * from users;").fetch_all(&**conn).await?;
println!("{:?}", users);
Ok(Template::render(
"users",
json!({"users": users, "user": user}),
))
}
#[tokio::main]
async fn main() -> Result<(), Error> {
let conn = MySqlPool::connect("mysql://test_user:password@localhost/test").await?;
let users: Users = conn.clone().into();
users.create_table().await?;
rocket::build()
.mount(
"/",
routes![
index,
get_login,
post_signup,
get_signup,
post_login,
logout,
delete,
show_all_users
],
)
.manage(conn)
.manage(users)
.attach(Template::fairing())
.launch()
.await
.unwrap();
Ok(())
}