When editing todo, dont clear

Author: forivall

src/modules/confirm_abort.rs

@@ -7,7 +7,7 @@ use crate::{
 	input::{Event, InputOptions, KeyBindings},
 	module::{ExitStatus, Module, State},
 	process::Results,
-	todo_file::TodoFile,
+	todo_file::{TodoFile, State as TodoFileState},
 	view::{RenderContext, ViewData},
 };
 
@@ -34,7 +34,10 @@ impl Module for ConfirmAbort {
 		let mut results = Results::new();
 		match confirmed {
 			Confirmed::Yes => {
-				self.todo_file.lock().set_lines(vec![]);
+				let todo_state = self.todo_file.lock().state().clone();
+				if todo_state != TodoFileState::Edit {
+					self.todo_file.lock().set_lines(vec![]);
+				}
 				results.exit_status(ExitStatus::Good);
 			},
 			Confirmed::No => {
@@ -135,4 +138,24 @@ mod tests {
 			},
 		);
 	}
+
+	#[test]
+	fn handle_event_yes_in_edit() {
+		module_test(
+			&["pick aaa comment"],
+			&[Event::from(MetaEvent::Yes)],
+			|mut test_context| {
+				let mut todo_file = test_context.take_todo_file();
+				todo_file.set_state(TodoFileState::Edit);
+
+				let mut module = create_confirm_abort(todo_file);
+				assert_results!(
+					test_context.handle_event(&mut module),
+					Artifact::Event(Event::from(MetaEvent::Yes)),
+					Artifact::ExitStatus(ExitStatus::Good)
+				);
+				assert!(!module.todo_file.lock().is_empty());
+			},
+		);
+	}
 }

src/todo_file.rs

@@ -88,6 +88,12 @@ impl TodoFile {
 		self.history.reset();
 	}
 
+	/// Set the rebase todo file state.
+	#[inline]
+	pub fn set_state(&mut self, state: State) {
+		self.state = state;
+	}
+
 	/// Load the rebase file from disk.
 	///
 	/// # Errors
@@ -117,7 +123,7 @@ impl TodoFile {
 			})
 			.collect();
 		self.set_lines(lines?);
-		self.state = detect_state(&self.filepath)?;
+		self.set_state(detect_state(&self.filepath)?);
 		Ok(())
 	}