mirror of https://github.com/rwf2/Rocket.git
117 lines
3.7 KiB
Rust
117 lines
3.7 KiB
Rust
#[macro_use] extern crate rocket;
|
|
#[macro_use] extern crate diesel;
|
|
#[macro_use] extern crate diesel_migrations;
|
|
#[macro_use] extern crate rocket_contrib;
|
|
|
|
#[cfg(test)]
|
|
mod tests;
|
|
mod task;
|
|
|
|
use rocket::{Rocket, Build};
|
|
use rocket::fairing::AdHoc;
|
|
use rocket::request::FlashMessage;
|
|
use rocket::response::{Flash, Redirect};
|
|
use rocket::serde::Serialize;
|
|
use rocket::form::Form;
|
|
use rocket::fs::{FileServer, relative};
|
|
|
|
use rocket_contrib::templates::Template;
|
|
|
|
use crate::task::{Task, Todo};
|
|
|
|
#[database("sqlite_database")]
|
|
pub struct DbConn(diesel::SqliteConnection);
|
|
|
|
#[derive(Debug, Serialize)]
|
|
#[serde(crate = "rocket::serde")]
|
|
struct Context {
|
|
flash: Option<(String, String)>,
|
|
tasks: Vec<Task>
|
|
}
|
|
|
|
impl Context {
|
|
pub async fn err<M: std::fmt::Display>(conn: &DbConn, msg: M) -> Context {
|
|
Context {
|
|
flash: Some(("error".into(), msg.to_string())),
|
|
tasks: Task::all(conn).await.unwrap_or_default()
|
|
}
|
|
}
|
|
|
|
pub async fn raw(conn: &DbConn, flash: Option<(String, String)>) -> Context {
|
|
match Task::all(conn).await {
|
|
Ok(tasks) => Context { flash, tasks },
|
|
Err(e) => {
|
|
error_!("DB Task::all() error: {}", e);
|
|
Context {
|
|
flash: Some(("error".into(), "Fail to access database.".into())),
|
|
tasks: vec![]
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
#[post("/", data = "<todo_form>")]
|
|
async fn new(todo_form: Form<Todo>, conn: DbConn) -> Flash<Redirect> {
|
|
let todo = todo_form.into_inner();
|
|
if todo.description.is_empty() {
|
|
Flash::error(Redirect::to("/"), "Description cannot be empty.")
|
|
} else if let Err(e) = Task::insert(todo, &conn).await {
|
|
error_!("DB insertion error: {}", e);
|
|
Flash::error(Redirect::to("/"), "Todo could not be inserted due an internal error.")
|
|
} else {
|
|
Flash::success(Redirect::to("/"), "Todo successfully added.")
|
|
}
|
|
}
|
|
|
|
#[put("/<id>")]
|
|
async fn toggle(id: i32, conn: DbConn) -> Result<Redirect, Template> {
|
|
match Task::toggle_with_id(id, &conn).await {
|
|
Ok(_) => Ok(Redirect::to("/")),
|
|
Err(e) => {
|
|
error_!("DB toggle({}) error: {}", id, e);
|
|
Err(Template::render("index", Context::err(&conn, "Failed to toggle task.").await))
|
|
}
|
|
}
|
|
}
|
|
|
|
#[delete("/<id>")]
|
|
async fn delete(id: i32, conn: DbConn) -> Result<Flash<Redirect>, Template> {
|
|
match Task::delete_with_id(id, &conn).await {
|
|
Ok(_) => Ok(Flash::success(Redirect::to("/"), "Todo was deleted.")),
|
|
Err(e) => {
|
|
error_!("DB deletion({}) error: {}", id, e);
|
|
Err(Template::render("index", Context::err(&conn, "Failed to delete task.").await))
|
|
}
|
|
}
|
|
}
|
|
|
|
#[get("/")]
|
|
async fn index(flash: Option<FlashMessage<'_>>, conn: DbConn) -> Template {
|
|
let flash = flash.map(FlashMessage::into_inner);
|
|
Template::render("index", Context::raw(&conn, flash).await)
|
|
}
|
|
|
|
async fn run_migrations(rocket: Rocket<Build>) -> Rocket<Build> {
|
|
// This macro from `diesel_migrations` defines an `embedded_migrations`
|
|
// module containing a function named `run`. This allows the example to be
|
|
// run and tested without any outside setup of the database.
|
|
embed_migrations!();
|
|
|
|
let conn = DbConn::get_one(&rocket).await.expect("database connection");
|
|
conn.run(|c| embedded_migrations::run(c)).await.expect("can run migrations");
|
|
|
|
rocket
|
|
}
|
|
|
|
#[launch]
|
|
fn rocket() -> _ {
|
|
rocket::build()
|
|
.attach(DbConn::fairing())
|
|
.attach(Template::fairing())
|
|
.attach(AdHoc::on_ignite("Run Migrations", run_migrations))
|
|
.mount("/", FileServer::from(relative!("static")))
|
|
.mount("/", routes![index])
|
|
.mount("/todo", routes![new, toggle, delete])
|
|
}
|