2018-07-21 22:11:08 +00:00
|
|
|
#![feature(plugin, decl_macro, use_extern_macros, custom_derive, const_fn)]
|
2016-10-12 07:38:30 +00:00
|
|
|
#![plugin(rocket_codegen)]
|
2016-07-16 04:09:08 +00:00
|
|
|
|
2018-08-07 02:58:07 +00:00
|
|
|
#[macro_use] extern crate rocket;
|
2016-08-02 02:07:36 +00:00
|
|
|
#[macro_use] extern crate diesel;
|
2016-10-04 00:56:43 +00:00
|
|
|
#[macro_use] extern crate serde_derive;
|
2017-01-31 10:01:30 +00:00
|
|
|
extern crate rocket_contrib;
|
2016-08-02 02:07:36 +00:00
|
|
|
|
|
|
|
mod static_files;
|
|
|
|
mod task;
|
2017-05-26 23:44:53 +00:00
|
|
|
#[cfg(test)] mod tests;
|
2016-07-16 04:09:08 +00:00
|
|
|
|
2017-05-26 23:44:53 +00:00
|
|
|
use rocket::Rocket;
|
2016-11-02 17:49:06 +00:00
|
|
|
use rocket::request::{Form, FlashMessage};
|
2016-09-12 01:57:04 +00:00
|
|
|
use rocket::response::{Flash, Redirect};
|
2016-09-22 11:12:07 +00:00
|
|
|
use rocket_contrib::Template;
|
2018-07-21 22:11:08 +00:00
|
|
|
use rocket_contrib::databases::database;
|
|
|
|
use diesel::SqliteConnection;
|
2017-02-03 01:38:36 +00:00
|
|
|
|
2017-05-26 23:44:53 +00:00
|
|
|
use task::{Task, Todo};
|
2016-08-02 02:07:36 +00:00
|
|
|
|
2018-07-21 22:11:08 +00:00
|
|
|
#[database("sqlite_database")]
|
|
|
|
pub struct DbConn(SqliteConnection);
|
|
|
|
|
2016-09-22 11:12:07 +00:00
|
|
|
#[derive(Debug, Serialize)]
|
2016-10-12 07:14:42 +00:00
|
|
|
struct Context<'a, 'b>{ msg: Option<(&'a str, &'b str)>, tasks: Vec<Task> }
|
2016-07-16 04:09:08 +00:00
|
|
|
|
2016-09-22 11:12:07 +00:00
|
|
|
impl<'a, 'b> Context<'a, 'b> {
|
2018-07-21 22:11:08 +00:00
|
|
|
pub fn err(conn: &DbConn, msg: &'a str) -> Context<'static, 'a> {
|
2017-02-03 01:38:36 +00:00
|
|
|
Context{msg: Some(("error", msg)), tasks: Task::all(conn)}
|
2016-09-22 11:12:07 +00:00
|
|
|
}
|
|
|
|
|
2018-07-21 22:11:08 +00:00
|
|
|
pub fn raw(conn: &DbConn, msg: Option<(&'a str, &'b str)>) -> Context<'a, 'b> {
|
2017-02-03 01:38:36 +00:00
|
|
|
Context{msg: msg, tasks: Task::all(conn)}
|
2016-09-22 11:12:07 +00:00
|
|
|
}
|
2016-07-16 04:09:08 +00:00
|
|
|
}
|
|
|
|
|
2016-10-12 07:14:42 +00:00
|
|
|
#[post("/", data = "<todo_form>")]
|
2018-07-21 22:11:08 +00:00
|
|
|
fn new(todo_form: Form<Todo>, conn: DbConn) -> Flash<Redirect> {
|
2016-10-12 07:14:42 +00:00
|
|
|
let todo = todo_form.into_inner();
|
2016-08-02 02:07:36 +00:00
|
|
|
if todo.description.is_empty() {
|
2016-10-12 07:14:42 +00:00
|
|
|
Flash::error(Redirect::to("/"), "Description cannot be empty.")
|
2017-05-26 23:44:53 +00:00
|
|
|
} else if Task::insert(todo, &conn) {
|
2016-10-12 07:14:42 +00:00
|
|
|
Flash::success(Redirect::to("/"), "Todo successfully added.")
|
2016-08-02 02:07:36 +00:00
|
|
|
} else {
|
2016-10-12 07:14:42 +00:00
|
|
|
Flash::error(Redirect::to("/"), "Whoops! The server failed.")
|
2016-08-02 02:07:36 +00:00
|
|
|
}
|
|
|
|
}
|
2016-07-16 04:09:08 +00:00
|
|
|
|
2016-09-25 09:26:15 +00:00
|
|
|
#[put("/<id>")]
|
2018-07-21 22:11:08 +00:00
|
|
|
fn toggle(id: i32, conn: DbConn) -> Result<Redirect, Template> {
|
2017-02-03 01:38:36 +00:00
|
|
|
if Task::toggle_with_id(id, &conn) {
|
2016-09-12 01:57:04 +00:00
|
|
|
Ok(Redirect::to("/"))
|
2016-08-02 02:07:36 +00:00
|
|
|
} else {
|
2017-02-03 01:38:36 +00:00
|
|
|
Err(Template::render("index", &Context::err(&conn, "Couldn't toggle task.")))
|
2016-08-02 02:07:36 +00:00
|
|
|
}
|
2016-07-16 04:09:08 +00:00
|
|
|
}
|
|
|
|
|
2016-09-25 09:26:15 +00:00
|
|
|
#[delete("/<id>")]
|
2018-07-21 22:11:08 +00:00
|
|
|
fn delete(id: i32, conn: DbConn) -> Result<Flash<Redirect>, Template> {
|
2017-02-03 01:38:36 +00:00
|
|
|
if Task::delete_with_id(id, &conn) {
|
2016-09-12 01:57:04 +00:00
|
|
|
Ok(Flash::success(Redirect::to("/"), "Todo was deleted."))
|
2016-08-02 02:07:36 +00:00
|
|
|
} else {
|
2017-02-03 01:38:36 +00:00
|
|
|
Err(Template::render("index", &Context::err(&conn, "Couldn't delete task.")))
|
2016-08-02 02:07:36 +00:00
|
|
|
}
|
2016-07-16 04:09:08 +00:00
|
|
|
}
|
|
|
|
|
2016-09-04 11:06:28 +00:00
|
|
|
#[get("/")]
|
2018-07-21 22:11:08 +00:00
|
|
|
fn index(msg: Option<FlashMessage>, conn: DbConn) -> Template {
|
2016-09-30 04:41:21 +00:00
|
|
|
Template::render("index", &match msg {
|
2017-02-03 01:38:36 +00:00
|
|
|
Some(ref msg) => Context::raw(&conn, Some((msg.name(), msg.msg()))),
|
|
|
|
None => Context::raw(&conn, None),
|
2016-09-22 11:12:07 +00:00
|
|
|
})
|
2016-07-16 04:09:08 +00:00
|
|
|
}
|
|
|
|
|
2018-07-21 22:11:08 +00:00
|
|
|
fn rocket() -> (Rocket, Option<DbConn>) {
|
2017-05-26 23:44:53 +00:00
|
|
|
let rocket = rocket::ignite()
|
2018-07-21 22:11:08 +00:00
|
|
|
.attach(DbConn::fairing())
|
2016-10-04 02:48:33 +00:00
|
|
|
.mount("/", routes![index, static_files::all])
|
2018-07-30 01:24:25 +00:00
|
|
|
.mount("/todo", routes![new, toggle, delete])
|
2017-05-26 23:44:53 +00:00
|
|
|
.attach(Template::fairing());
|
|
|
|
|
2018-07-21 22:11:08 +00:00
|
|
|
let conn = match cfg!(test) {
|
|
|
|
true => DbConn::get_one(&rocket),
|
|
|
|
false => None,
|
|
|
|
};
|
|
|
|
|
2017-05-26 23:44:53 +00:00
|
|
|
(rocket, conn)
|
|
|
|
}
|
|
|
|
|
|
|
|
fn main() {
|
|
|
|
rocket().0.launch();
|
2016-07-16 04:09:08 +00:00
|
|
|
}
|