use crate::errors::error::Error; use crate::errors::error_vec::ErrorVec; use crate::errors::task_error::TaskError; use crate::models::category::Category; #[cfg(feature = "server")] use crate::models::category::ReoccurrenceInterval; #[cfg(feature = "server")] use crate::models::subtask::Subtask; use crate::models::task::{NewTask, Task, TaskWithSubtasks}; #[cfg(feature = "server")] use crate::server::database_connection::establish_database_connection; #[cfg(feature = "server")] use crate::server::subtasks::restore_subtasks_of_task; #[cfg(feature = "server")] use chrono::{Datelike, Days, Local, Months, NaiveDate}; #[cfg(feature = "server")] use diesel::prelude::*; #[cfg(feature = "server")] use diesel::{ExpressionMethods, OptionalExtension, QueryDsl, RunQueryDsl, SelectableHelper}; use dioxus::prelude::*; #[cfg(feature = "server")] use time::Month; #[cfg(feature = "server")] use validator::Validate; #[server] pub(crate) async fn create_task( new_task: NewTask, ) -> Result>> { use crate::schema::tasks; // TODO: replace with model sanitization (https://github.com/matous-volf/todo-baggins/issues/13) let mut new_task = new_task; new_task.title = new_task.title.trim().to_owned(); new_task .validate() .map_err::, _>(|errors| errors.into())?; let mut connection = establish_database_connection().map_err::, _>(|_| { vec![TaskError::Error(Error::ServerInternal)].into() })?; let new_task = diesel::insert_into(tasks::table) .values(&new_task) .returning(Task::as_returning()) .get_result(&mut connection) .map_err::, _>(|error| vec![error.into()].into())?; Ok(new_task) } #[server] pub(crate) async fn get_task(task_id: i32) -> Result>> { use crate::schema::tasks::dsl::*; let mut connection = establish_database_connection() .map_err::, _>(|_| vec![Error::ServerInternal].into())?; let task = tasks .find(task_id) .select(Task::as_select()) .first(&mut connection) .optional() .map_err::, _>(|_| vec![Error::ServerInternal].into())?; // TODO: Handle not finding the task. Ok(task.unwrap()) } #[server] pub(crate) async fn get_tasks_in_category( filtered_category: Category, ) -> Result, ServerFnError>> { use crate::schema::tasks::dsl::*; let mut connection = establish_database_connection() .map_err::, _>(|_| vec![Error::ServerInternal].into())?; let results = tasks .select(Task::as_select()) .filter(filtered_category.eq_sql_predicate()) .load::(&mut connection) .map_err::, _>(|_| vec![Error::ServerInternal].into())?; Ok(results) } #[server] pub(crate) async fn get_tasks_with_subtasks_in_category( filtered_category: Category, ) -> Result, ServerFnError>> { use crate::schema::tasks; let mut connection = establish_database_connection() .map_err::, _>(|_| vec![Error::ServerInternal].into())?; let tasks_in_category = tasks::table .filter(filtered_category.eq_sql_predicate()) .select(Task::as_select()) .load(&mut connection) .map_err::, _>(|_| vec![Error::ServerInternal].into())?; let subtasks = Subtask::belonging_to(&tasks_in_category) .select(Subtask::as_select()) .load(&mut connection) .map_err::, _>(|_| vec![Error::ServerInternal].into())?; let tasks_with_subtasks = subtasks .grouped_by(&tasks_in_category) .into_iter() .zip(tasks_in_category) .map(|(pages, book)| TaskWithSubtasks::new(book, pages)) .collect(); Ok(tasks_with_subtasks) } #[server] pub(crate) async fn edit_task( task_id: i32, mut new_task: NewTask, ) -> Result>> { use crate::schema::tasks::dsl::*; // TODO: replace with model sanitization (https://github.com/matous-volf/todo-baggins/issues/13) let mut new_task = new_task; new_task.title = new_task.title.trim().to_owned(); new_task .validate() .map_err::, _>(|errors| errors.into())?; let mut connection = establish_database_connection().map_err::, _>(|_| { vec![TaskError::Error(Error::ServerInternal)].into() })?; let updated_task = diesel::update(tasks) .filter(id.eq(task_id)) .set(( title.eq(new_task.title), deadline.eq(new_task.deadline), category.eq(new_task.category), project_id.eq(new_task.project_id), )) .returning(Task::as_returning()) .get_result(&mut connection) .map_err::, _>(|error| vec![error.into()].into())?; Ok(updated_task) } #[server] pub(crate) async fn complete_task(task_id: i32) -> Result>> { let task = get_task(task_id).await?; let mut new_task = NewTask::from(task); if let Category::Calendar { reoccurrence: Some(reoccurrence), date, .. } = &mut new_task.category { match reoccurrence.interval() { ReoccurrenceInterval::Day => *date = *date + Days::new(reoccurrence.length() as u64), ReoccurrenceInterval::Month | ReoccurrenceInterval::Year => { *date = *date + Months::new( reoccurrence.length() * if *(reoccurrence.interval()) == ReoccurrenceInterval::Year { 12 } else { 1 }, ); *date = NaiveDate::from_ymd_opt( date.year(), date.month(), reoccurrence.start_date().day().min( Month::try_from(date.month() as u8) .unwrap() .length(date.year()) as u32, ), ) .unwrap() } } restore_subtasks_of_task(task_id).await?; } else { new_task.category = Category::Done; } let updated_task = edit_task(task_id, new_task) .await .map_err::, _>(|_| vec![Error::ServerInternal].into())?; Ok(updated_task) } // TODO: Get rid of this suppression. //noinspection DuplicatedCode #[server] pub(crate) async fn delete_task(task_id: i32) -> Result<(), ServerFnError>> { use crate::schema::tasks::dsl::*; let mut connection = establish_database_connection() .map_err::, _>(|_| vec![Error::ServerInternal].into())?; diesel::delete(tasks.filter(id.eq(task_id))) .execute(&mut connection) .map_err::, _>(|error| vec![error.into()].into())?; Ok(()) } #[cfg(feature = "server")] pub(crate) async fn trigger_task_updated_at(task_id: i32) -> Result> { use crate::schema::tasks::dsl::*; let mut connection = establish_database_connection() .map_err::, _>(|_| vec![Error::ServerInternal].into())?; let updated_task = diesel::update(tasks) .filter(id.eq(task_id)) .set(updated_at.eq(Local::now().naive_local())) .returning(Task::as_returning()) .get_result(&mut connection) .map_err::, _>(|error| vec![error.into()].into())?; Ok(updated_task) }