DONE Refactoring methods: Dedup into single function and fix odds and ends

This commit is contained in:
Joseph Ferano 2023-06-15 14:06:05 +07:00
parent 02ad7c3da6
commit dba1e8a640

View File

@ -88,39 +88,14 @@ impl<'a> State<'a> {
&mut self.columns[self.selected_column_idx] &mut self.columns[self.selected_column_idx]
} }
pub fn select_previous_column(&mut self) -> &Column { pub fn select_previous_column(&mut self) {
self.selected_column_idx = self.selected_column_idx.saturating_sub(1); self.selected_column_idx = self.selected_column_idx.saturating_sub(1);
self.db_conn.set_selected_column(self.selected_column_idx); self.db_conn.set_selected_column(self.selected_column_idx);
&self.columns[self.selected_column_idx]
} }
pub fn select_next_column(&mut self) -> &Column { pub fn select_next_column(&mut self) {
self.selected_column_idx = min(self.selected_column_idx + 1, self.columns.len() - 1); self.selected_column_idx = min(self.selected_column_idx + 1, self.columns.len() - 1);
self.db_conn.set_selected_column(self.selected_column_idx); self.db_conn.set_selected_column(self.selected_column_idx);
&self.columns[self.selected_column_idx]
}
fn move_task_to_column(&mut self, move_next: bool) {
let col_idx = self.selected_column_idx;
let cols_len = self.columns.len();
let column = self.get_selected_column_mut();
let cond = if move_next {
col_idx < cols_len - 1
} else {
col_idx > 0
};
if cond && !column.tasks.is_empty() {
let t = column.tasks.remove(column.selected_task_idx);
self.select_previous_task();
if move_next {
self.select_next_column();
} else {
self.select_previous_column();
}
let col = self.get_selected_column_mut();
col.tasks.push(t);
self.select_last_task();
}
} }
#[must_use] #[must_use]
@ -132,7 +107,11 @@ impl<'a> State<'a> {
#[must_use] #[must_use]
pub fn get_previous_task(&self) -> Option<&Task> { pub fn get_previous_task(&self) -> Option<&Task> {
let column = self.get_selected_column(); let column = self.get_selected_column();
if column.selected_task_idx > 0 {
column.tasks.get(column.selected_task_idx - 1) column.tasks.get(column.selected_task_idx - 1)
} else {
None
}
} }
#[must_use] #[must_use]
@ -148,34 +127,40 @@ impl<'a> State<'a> {
pub fn select_previous_task(&mut self) { pub fn select_previous_task(&mut self) {
let column = self.get_selected_column_mut(); let column = self.get_selected_column_mut();
let task_idx = &mut column.selected_task_idx; column.selected_task_idx = column.selected_task_idx.saturating_sub(1);
*task_idx = task_idx.saturating_sub(1);
let task_idx = column.selected_task_idx; let task_idx = column.selected_task_idx;
let col_id = self.get_selected_column().id; let col_id = column.id;
self.db_conn.set_selected_task_for_column(task_idx, col_id); self.db_conn.set_selected_task_for_column(task_idx, col_id);
} }
pub fn select_next_task(&mut self) { pub fn select_next_task(&mut self) {
let column = self.get_selected_column_mut(); let column = self.get_selected_column_mut();
let task_idx = &mut column.selected_task_idx; column.selected_task_idx = min(
*task_idx = min(*task_idx + 1, column.tasks.len().saturating_sub(1)); column.selected_task_idx + 1,
let task_idx = self.get_selected_column().selected_task_idx; column.tasks.len().saturating_sub(1),
let col_id = self.get_selected_column().id; );
let task_idx = column.selected_task_idx;
let col_id = column.id;
self.db_conn.set_selected_task_for_column(task_idx, col_id); self.db_conn.set_selected_task_for_column(task_idx, col_id);
} }
pub fn select_first_task(&mut self) { pub fn select_first_task(&mut self) {
self.get_selected_column_mut().selected_task_idx = 0; let column = self.get_selected_column_mut();
let task_idx = self.get_selected_column().selected_task_idx; column.selected_task_idx = 0;
let col_id = self.get_selected_column().id;
let task_idx = column.selected_task_idx;
let col_id = column.id;
self.db_conn.set_selected_task_for_column(task_idx, col_id); self.db_conn.set_selected_task_for_column(task_idx, col_id);
} }
pub fn select_last_task(&mut self) { pub fn select_last_task(&mut self) {
let column = self.get_selected_column_mut(); let column = self.get_selected_column_mut();
column.selected_task_idx = column.tasks.len().saturating_sub(1); column.selected_task_idx = column.tasks.len().saturating_sub(1);
let task_idx = column.selected_task_idx; let task_idx = column.selected_task_idx;
let col_id = self.get_selected_column().id; let col_id = column.id;
self.db_conn.set_selected_task_for_column(task_idx, col_id); self.db_conn.set_selected_task_for_column(task_idx, col_id);
} }
@ -188,103 +173,95 @@ impl<'a> State<'a> {
is_edit: true, is_edit: true,
}) })
} }
/// Returns the move task up of this [`State`].
/// pub fn move_task_up(&mut self) {
/// # Panics self.move_task(false);
///
/// We have conditions to ensure this doesn't panic but we still unwrap()
pub fn move_task_up(&mut self) -> bool {
let column = self.get_selected_column_mut();
let task_idx = column.selected_task_idx;
if task_idx > 0 {
column.tasks.swap(task_idx, task_idx - 1);
column.selected_task_idx -= 1;
let col_id = column.id;
let task1_id = self.get_selected_task().unwrap().id;
let task2_id = self.get_next_task().unwrap().id;
self.db_conn.swap_task_order(task1_id, task2_id);
self.db_conn.set_selected_task_for_column(task_idx, col_id);
true
} else {
false
}
} }
pub fn move_task_down(&mut self) {
self.move_task(true);
}
/// Returns the move task down of this [`State`]. /// Returns the move task down of this [`State`].
/// ///
/// # Panics /// # Panics
/// ///
/// We have conditions to ensure this doesn't panic but we still unwrap() /// We have conditions to ensure this doesn't panic but we still unwrap()
pub fn move_task_down(&mut self) -> bool { pub fn move_task(&mut self, is_down: bool) {
let column = self.get_selected_column_mut(); let other_task = if is_down {
if column.selected_task_idx < column.tasks.len().saturating_sub(1) { self.get_next_task()
let task_idx = column.selected_task_idx;
column.tasks.swap(task_idx, task_idx + 1);
column.selected_task_idx += 1;
let col_id = column.id;
let task1_id = self.get_selected_task().unwrap().id;
let task2_id = self.get_previous_task().unwrap().id;
self.db_conn.swap_task_order(task1_id, task2_id);
self.db_conn.set_selected_task_for_column(task_idx, col_id);
true
} else { } else {
false self.get_previous_task()
};
if let (Some(task1), Some(task2)) = (self.get_selected_task(), other_task) {
let t1_id = task1.id;
let t2_id = task2.id;
let column = self.get_selected_column_mut();
let task_idx = column.selected_task_idx;
let other_idx = if is_down { task_idx + 1 } else { task_idx - 1 };
column.tasks.swap(task_idx, other_idx);
if is_down {
column.selected_task_idx += 1;
} else {
column.selected_task_idx -= 1;
}
let col_id = column.id;
self.db_conn.swap_task_order(t2_id, t1_id);
self.db_conn.set_selected_task_for_column(task_idx, col_id);
} }
} }
/// Returns the move task previous column of this [`State`].
///
/// # Panics
///
/// We have conditions to ensure this doesn't panic but we still unwrap()
pub fn move_task_previous_column(&mut self) { pub fn move_task_previous_column(&mut self) {
let first_col = self.get_selected_column_mut();
if first_col.tasks.is_empty() {
return;
}
let task_idx = first_col.selected_task_idx.saturating_sub(1);
let col_id = first_col.id;
self.db_conn.set_selected_task_for_column(task_idx, col_id);
self.move_task_to_column(false); self.move_task_to_column(false);
self.db_conn.move_task_to_column(
self.get_selected_task().unwrap(),
self.get_selected_column(),
);
self.db_conn.set_selected_column(self.selected_column_idx);
} }
/// Returns the move task previous column of this [`State`].
///
/// # Panics
///
/// We have conditions to ensure this doesn't panic but we still unwrap()
pub fn move_task_next_column(&mut self) { pub fn move_task_next_column(&mut self) {
self.move_task_to_column(true);
}
fn move_task_to_column(&mut self, move_right: bool) {
let can_move_right = move_right && self.selected_column_idx < self.columns.len() - 1;
let can_move_left = !move_right && self.selected_column_idx > 0;
let first_col = self.get_selected_column_mut(); let first_col = self.get_selected_column_mut();
if first_col.tasks.is_empty() { if first_col.tasks.is_empty() || !can_move_right && !can_move_left {
return; return;
} }
let task_idx = first_col.selected_task_idx.saturating_sub(1); let t = first_col.tasks.remove(first_col.selected_task_idx);
let col_id = first_col.id;
self.db_conn.set_selected_task_for_column(task_idx, col_id); // Only move it if it was the last task
self.move_task_to_column(true); if first_col.selected_task_idx == first_col.tasks.len() {
self.db_conn.move_task_to_column( self.select_previous_task();
self.get_selected_task().unwrap(), }
self.get_selected_column(),
); if move_right {
self.select_next_column();
} else {
self.select_previous_column();
}
let col = self.get_selected_column_mut();
col.tasks.push(t);
self.select_last_task();
if let Some(task) = self.get_selected_task() {
self.db_conn.move_task_to_column(task, self.get_selected_column());
self.db_conn.set_selected_column(self.selected_column_idx); self.db_conn.set_selected_column(self.selected_column_idx);
} }
}
pub fn add_new_task(&mut self, title: String, description: String) { pub fn add_new_task(&mut self, title: String, description: String) {
let col_id = self.get_selected_column().id; let col_id = self.get_selected_column().id;
let task = self.db_conn.insert_new_task(title, description, col_id); let task = self.db_conn.insert_new_task(title, description, col_id);
self.select_last_task(); self.select_last_task();
let selected_task_idx = self.get_selected_column().selected_task_idx; let selected_task_idx = self.get_selected_column().selected_task_idx;
self.db_conn.set_selected_task_for_column(selected_task_idx, col_id); self.db_conn.set_selected_task_for_column(selected_task_idx, col_id);
self.get_selected_column_mut().tasks.push(task); self.get_selected_column_mut().tasks.push(task);
} }
pub fn edit_task(&mut self, title: String, description: String) { pub fn edit_task(&mut self, title: String, description: String) {
let _column = self.get_selected_column_mut();
if let Some(selected_task) = self.get_selected_task_mut() { if let Some(selected_task) = self.get_selected_task_mut() {
selected_task.title = title; selected_task.title = title;
selected_task.description = description; selected_task.description = description;
@ -301,7 +278,7 @@ impl<'a> State<'a> {
pub fn delete_task(&mut self) { pub fn delete_task(&mut self) {
let column = self.get_selected_column(); let column = self.get_selected_column();
if column.tasks.is_empty() { if column.tasks.is_empty() {
return return;
} }
let task_id = self.get_selected_task().unwrap().id; let task_id = self.get_selected_task().unwrap().id;
let column = self.get_selected_column_mut(); let column = self.get_selected_column_mut();