mod schema { infer_schema!("db/db.sql"); } use diesel; use diesel::prelude::*; use diesel::sqlite::SqliteConnection; use self::schema::tasks; use self::schema::tasks::dsl::{tasks as all_tasks, completed as task_completed}; fn db() -> SqliteConnection { SqliteConnection::establish("db/db.sql").expect("Failed to connect to db.") } #[table_name = "tasks"] #[derive(Serialize, Queryable, Insertable, FromForm, Debug, Clone)] pub struct Task { id: Option, pub description: String, pub completed: Option } impl Task { pub fn all() -> Vec { all_tasks.order(tasks::id.desc()).load::(&db()).unwrap() } pub fn insert(&self) -> bool { diesel::insert(self).into(tasks::table).execute(&db()).is_ok() } pub fn toggle_with_id(id: i32) -> bool { let task = all_tasks.find(id).get_result::(&db()); if task.is_err() { return false; } let new_status = !task.unwrap().completed.unwrap(); let updated_task = diesel::update(all_tasks.find(id)); updated_task.set(task_completed.eq(new_status)).execute(&db()).is_ok() } pub fn delete_with_id(id: i32) -> bool { diesel::delete(all_tasks.find(id)).execute(&db()).is_ok() } }