WIP: add index action for git-revise

forivall · forivall · commit a2536be938ed · 2024-03-15T15:47:33.000-07:00
diff --git a/readme/customization.md b/readme/customization.md
@@ -113,6 +113,7 @@ Most keys can be changed to any printable character or supported special charact
 | `inputActionPick`           | p         | String | Key for setting action to pick                      |
 | `inputActionReword`         | r         | String | Key for setting action to reword                    |
 | `inputActionSquash`         | s         | String | Key for setting action to squash                    |
+| `inputActionIndex`          | i         | String | Key for setting action to index (git-revise)        |
 | `inputConfirmNo`            | n         | String | Key for rejecting a confirmation                    |
 | `inputConfirmYes`           | y         | String | Key for confirming a confirmation                   |
 | `inputEdit`                 | E         | String | Key for entering edit mode                          |
diff --git a/src/config/key_bindings.rs b/src/config/key_bindings.rs
@@ -26,6 +26,8 @@ pub(crate) struct KeyBindings {
 	pub(crate) action_edit: Vec<String>,
 	/// Key bindings for the fixup action.
 	pub(crate) action_fixup: Vec<String>,
+	/// Key bindings for the index action.
+	pub(crate) action_index: Vec<String>,
 	/// Key bindings for the pick action.
 	pub(crate) action_pick: Vec<String>,
 	/// Key bindings for the reword action.
@@ -129,6 +131,7 @@ impl KeyBindings {
 			action_drop: get_input(git_config, "interactive-rebase-tool.inputActionDrop", "d")?,
 			action_edit: get_input(git_config, "interactive-rebase-tool.inputActionEdit", "e")?,
 			action_fixup: get_input(git_config, "interactive-rebase-tool.inputActionFixup", "f")?,
+			action_index: get_input(git_config, "interactive-rebase-tool.inputActionIndex", "i")?,
 			action_pick: get_input(git_config, "interactive-rebase-tool.inputActionPick", "p")?,
 			action_reword: get_input(git_config, "interactive-rebase-tool.inputActionReword", "r")?,
 			action_squash: get_input(git_config, "interactive-rebase-tool.inputActionSquash", "s")?,
@@ -243,6 +246,7 @@ mod tests {
 		config_test!(action_drop, "inputActionDrop", "d");
 		config_test!(action_edit, "inputActionEdit", "e");
 		config_test!(action_fixup, "inputActionFixup", "f");
+		config_test!(action_index, "inputActionIndex", "i");
 		config_test!(action_pick, "inputActionPick", "p");
 		config_test!(action_reword, "inputActionReword", "r");
 		config_test!(action_squash, "inputActionSquash", "s");
diff --git a/src/config/theme.rs b/src/config/theme.rs
@@ -39,6 +39,8 @@ pub(crate) struct Theme {
 	pub(crate) color_action_exec: Color,
 	/// The color for the fixup action.
 	pub(crate) color_action_fixup: Color,
+	/// The color for the fixup action.
+	pub(crate) color_action_index: Color,
 	/// The color for the pick action.
 	pub(crate) color_action_pick: Color,
 	/// The color for the reword action.
@@ -87,6 +89,7 @@ impl Theme {
 			color_action_edit: get_color(git_config, "interactive-rebase-tool.editColor", Color::LightBlue)?,
 			color_action_exec: get_color(git_config, "interactive-rebase-tool.execColor", Color::LightWhite)?,
 			color_action_fixup: get_color(git_config, "interactive-rebase-tool.fixupColor", Color::LightMagenta)?,
+			color_action_index: get_color(git_config, "interactive-rebase-tool.indexColor", Color::DarkGreen)?,
 			color_action_pick: get_color(git_config, "interactive-rebase-tool.pickColor", Color::LightGreen)?,
 			color_action_reword: get_color(git_config, "interactive-rebase-tool.rewordColor", Color::LightYellow)?,
 			color_action_squash: get_color(git_config, "interactive-rebase-tool.squashColor", Color::LightCyan)?,
@@ -204,6 +207,7 @@ mod tests {
 		config_test!(color_action_edit, "editColor", Color::LightBlue);
 		config_test!(color_action_exec, "execColor", Color::LightWhite);
 		config_test!(color_action_fixup, "fixupColor", Color::LightMagenta);
+		config_test!(color_action_index, "indexColor", Color::DarkGreen);
 		config_test!(color_action_pick, "pickColor", Color::LightGreen);
 		config_test!(color_action_reword, "rewordColor", Color::LightYellow);
 		config_test!(color_action_squash, "squashColor", Color::LightCyan);
diff --git a/src/display.rs b/src/display.rs
@@ -38,6 +38,7 @@ pub(crate) struct Display<T: Tui> {
 	action_edit: (Colors, Colors),
 	action_exec: (Colors, Colors),
 	action_fixup: (Colors, Colors),
+	action_index: (Colors, Colors),
 	action_label: (Colors, Colors),
 	action_merge: (Colors, Colors),
 	action_pick: (Colors, Colors),
@@ -101,6 +102,12 @@ impl<T: Tui> Display<T> {
 			theme.color_background,
 			theme.color_selected_background,
 		);
+		let action_index = register_selectable_color_pairs(
+			color_mode,
+			theme.color_action_index,
+			theme.color_background,
+			theme.color_selected_background,
+		);
 		let action_pick = register_selectable_color_pairs(
 			color_mode,
 			theme.color_action_pick,
@@ -180,6 +187,7 @@ impl<T: Tui> Display<T> {
 			action_edit,
 			action_exec,
 			action_fixup,
+			action_index,
 			action_label,
 			action_merge,
 			action_pick,
@@ -240,6 +248,7 @@ impl<T: Tui> Display<T> {
 					DisplayColor::ActionEdit => self.action_edit.1,
 					DisplayColor::ActionExec => self.action_exec.1,
 					DisplayColor::ActionFixup => self.action_fixup.1,
+					DisplayColor::ActionIndex => self.action_index.1,
 					DisplayColor::ActionPick => self.action_pick.1,
 					DisplayColor::ActionReword => self.action_reword.1,
 					DisplayColor::ActionSquash => self.action_squash.1,
@@ -263,6 +272,7 @@ impl<T: Tui> Display<T> {
 					DisplayColor::ActionEdit => self.action_edit.0,
 					DisplayColor::ActionExec => self.action_exec.0,
 					DisplayColor::ActionFixup => self.action_fixup.0,
+					DisplayColor::ActionIndex => self.action_index.0,
 					DisplayColor::ActionPick => self.action_pick.0,
 					DisplayColor::ActionReword => self.action_reword.0,
 					DisplayColor::ActionSquash => self.action_squash.0,
@@ -427,6 +437,13 @@ mod tests {
 		CrosstermColor::Magenta,
 		CrosstermColor::AnsiValue(237)
 	)]
+	#[case::action_index(DisplayColor::ActionIndex, false, CrosstermColor::DarkGreen, CrosstermColor::Reset)]
+	#[case::action_index_selected(
+		DisplayColor::ActionIndex,
+		true,
+		CrosstermColor::DarkGreen,
+		CrosstermColor::AnsiValue(237)
+	)]
 	#[case::action_pick(DisplayColor::ActionPick, false, CrosstermColor::Green, CrosstermColor::Reset)]
 	#[case::action_pick_selected(
 		DisplayColor::ActionPick,
diff --git a/src/display/display_color.rs b/src/display/display_color.rs
@@ -12,6 +12,8 @@ pub(crate) enum DisplayColor {
 	ActionExec,
 	/// The color for the fixup action.
 	ActionFixup,
+	/// The color for the index action.
+	ActionIndex,
 	/// The color for the pick action.
 	ActionPick,
 	/// The color for the reword action.
diff --git a/src/input/key_bindings.rs b/src/input/key_bindings.rs
@@ -46,6 +46,8 @@ pub(crate) struct KeyBindings {
 	pub(crate) action_edit: Vec<Event>,
 	/// Key bindings for the fixup action.
 	pub(crate) action_fixup: Vec<Event>,
+	/// Key bindings for the index action.
+	pub(crate) action_index: Vec<Event>,
 	/// Key bindings for the pick action.
 	pub(crate) action_pick: Vec<Event>,
 	/// Key bindings for the reword action.
@@ -124,6 +126,7 @@ impl KeyBindings {
 			action_drop: map_keybindings(&key_bindings.action_drop),
 			action_edit: map_keybindings(&key_bindings.action_edit),
 			action_fixup: map_keybindings(&key_bindings.action_fixup),
+			action_index: map_keybindings(&key_bindings.action_index),
 			action_pick: map_keybindings(&key_bindings.action_pick),
 			action_reword: map_keybindings(&key_bindings.action_reword),
 			action_squash: map_keybindings(&key_bindings.action_squash),
diff --git a/src/input/standard_event.rs b/src/input/standard_event.rs
@@ -55,6 +55,8 @@ pub(crate) enum StandardEvent {
 	ActionEdit,
 	/// The fixup action meta event.
 	ActionFixup,
+	/// The index (git-revise) action meta event.
+	ActionIndex,
 	/// The pick action meta event.
 	ActionPick,
 	/// The reword action meta event.
diff --git a/src/modules/list.rs b/src/modules/list.rs
@@ -557,6 +557,7 @@ impl List {
 			e if key_bindings.action_drop.contains(&e) => Event::from(StandardEvent::ActionDrop),
 			e if key_bindings.action_edit.contains(&e) => Event::from(StandardEvent::ActionEdit),
 			e if key_bindings.action_fixup.contains(&e) => Event::from(StandardEvent::ActionFixup),
+			e if key_bindings.action_index.contains(&e) => Event::from(StandardEvent::ActionIndex),
 			e if key_bindings.action_pick.contains(&e) => Event::from(StandardEvent::ActionPick),
 			e if key_bindings.action_reword.contains(&e) => Event::from(StandardEvent::ActionReword),
 			e if key_bindings.action_squash.contains(&e) => Event::from(StandardEvent::ActionSquash),
@@ -645,6 +646,7 @@ impl List {
 					StandardEvent::ActionDrop => self.set_selected_line_action(Action::Drop),
 					StandardEvent::ActionEdit => self.set_selected_line_action(Action::Edit),
 					StandardEvent::ActionFixup => self.set_selected_line_action(Action::Fixup),
+					StandardEvent::ActionIndex => self.set_selected_line_action(Action::Index),
 					StandardEvent::ActionPick => self.set_selected_line_action(Action::Pick),
 					StandardEvent::ActionReword => self.set_selected_line_action(Action::Reword),
 					StandardEvent::ActionSquash => self.set_selected_line_action(Action::Squash),
diff --git a/src/modules/list/search.rs b/src/modules/list/search.rs
@@ -62,6 +62,7 @@ impl Searchable for Search {
 				Action::Drop
 				| Action::Edit
 				| Action::Fixup
+				| Action::Index
 				| Action::Pick
 				| Action::Reword
 				| Action::Squash
@@ -72,6 +73,7 @@ impl Searchable for Search {
 				Action::Drop
 				| Action::Edit
 				| Action::Fixup
+				| Action::Index
 				| Action::Pick
 				| Action::Reword
 				| Action::Squash
diff --git a/src/modules/list/tests/help.rs b/src/modules/list/tests/help.rs
@@ -40,6 +40,7 @@ fn normal_mode_help() {
 				" s       |Set selected commits to be squashed",
 				" f       |Set selected commits to be fixed-up",
 				" d       |Set selected commits to be dropped",
+				" i       |Set selected commits to be staged in the index (git-revise)",
 				" E       |Edit an exec, label, reset or merge action's content",
 				" I       |Insert a new line",
 				" Delete  |Completely remove the selected lines",
@@ -105,6 +106,7 @@ fn visual_mode_help() {
 				" s       |Set selected commits to be squashed",
 				" f       |Set selected commits to be fixed-up",
 				" d       |Set selected commits to be dropped",
+				" i       |Set selected commits to be staged in the index (git-revise)",
 				" Delete  |Completely remove the selected lines",
 				" Controlz|Undo the last change",
 				" Controly|Redo the previous undone change",
diff --git a/src/modules/list/utils.rs b/src/modules/list/utils.rs
@@ -127,6 +127,11 @@ fn build_help_lines(key_bindings: &KeyBindings, selector: HelpLinesSelector) ->
 			"Set selected commits to be dropped",
 			HelpLinesSelector::Common,
 		),
+		(
+			&key_bindings.action_index,
+			"Set selected commits to be staged in the index (git-revise)",
+			HelpLinesSelector::Common,
+		),
 		(
 			&key_bindings.edit,
 			"Edit an exec, label, reset or merge action's content",
@@ -189,6 +194,7 @@ const fn get_action_color(action: Action) -> DisplayColor {
 		Action::Edit => DisplayColor::ActionEdit,
 		Action::Exec => DisplayColor::ActionExec,
 		Action::Fixup => DisplayColor::ActionFixup,
+		Action::Index => DisplayColor::ActionIndex,
 		Action::Pick => DisplayColor::ActionPick,
 		Action::Reword => DisplayColor::ActionReword,
 		Action::Squash => DisplayColor::ActionSquash,
@@ -209,7 +215,7 @@ pub(super) fn get_line_action_maximum_width(todo_file: &TodoFile) -> usize {
 			// allow these to overflow their bounds
 			&Action::Exec | &Action::UpdateRef => 0,
 			&Action::Drop | &Action::Edit | &Action::Noop | &Action::Pick => 4,
-			&Action::Break | &Action::Label | &Action::Reset | &Action::Merge => 5,
+			&Action::Break | &Action::Label | &Action::Reset | &Action::Merge | &Action::Index => 5,
 			&Action::Fixup => {
 				if line.option().is_some() {
 					8 // "fixup -C" = 8
@@ -297,7 +303,7 @@ pub(super) fn get_todo_line_segments(
 
 	// render hash
 	match *action {
-		Action::Drop | Action::Edit | Action::Fixup | Action::Pick | Action::Reword | Action::Squash => {
+		Action::Drop | Action::Edit | Action::Fixup | Action::Index | Action::Pick | Action::Reword | Action::Squash => {
 			let action_width = if is_full_width { 8 } else { 3 };
 			let max_index = cmp::min(line.get_hash().len(), action_width);
 			let search_hash_match = search_match.map_or(false, |m| m.hash());
diff --git a/src/todo_file.rs b/src/todo_file.rs
@@ -155,6 +155,7 @@ impl TodoFile {
 								Action::Drop
 								| Action::Fixup
 								| Action::Edit
+								| Action::Index
 								| Action::Pick
 								| Action::Reword
 								| Action::Squash => {
diff --git a/src/todo_file/action.rs b/src/todo_file/action.rs
@@ -16,6 +16,8 @@ pub(crate) enum Action {
 	Exec,
 	/// A fixup action.
 	Fixup,
+	/// A index action for git-revise.
+	Index,
 	/// A noop action.
 	Noop,
 	/// A pick action.
@@ -44,6 +46,7 @@ impl Action {
 			Self::Edit => "e",
 			Self::Exec => "x",
 			Self::Fixup => "f",
+			Self::Index => "i",
 			Self::Label => "l",
 			Self::Merge => "m",
 			Self::Noop => "n",
@@ -60,7 +63,7 @@ impl Action {
 	pub(crate) const fn is_static(self) -> bool {
 		match self {
 			Self::Break | Self::Exec | Self::Noop | Self::Reset | Self::Label | Self::Merge | Self::UpdateRef => true,
-			Self::Drop | Self::Edit | Self::Fixup | Self::Pick | Self::Reword | Self::Squash => false,
+			Self::Drop | Self::Edit | Self::Index | Self::Fixup | Self::Pick | Self::Reword | Self::Squash => false,
 		}
 	}
 }
@@ -73,6 +76,7 @@ impl Display for Action {
 			Self::Edit => "edit",
 			Self::Exec => "exec",
 			Self::Fixup => "fixup",
+			Self::Index => "index",
 			Self::Label => "label",
 			Self::Merge => "merge",
 			Self::Noop => "noop",
@@ -91,10 +95,12 @@ impl TryFrom<&str> for Action {
 	fn try_from(s: &str) -> Result<Self, Self::Error> {
 		match s {
 			"break" | "b" => Ok(Self::Break),
+			// "cut" => Ok(Self::Cut),
 			"drop" | "d" => Ok(Self::Drop),
 			"edit" | "e" => Ok(Self::Edit),
 			"exec" | "x" => Ok(Self::Exec),
 			"fixup" | "f" => Ok(Self::Fixup),
+			"index" => Ok(Self::Index),
 			"noop" | "n" => Ok(Self::Noop),
 			"pick" | "p" => Ok(Self::Pick),
 			"reword" | "r" => Ok(Self::Reword),
diff --git a/src/todo_file/line.rs b/src/todo_file/line.rs
@@ -95,7 +95,7 @@ impl Line {
 		Ok(match action {
 			Action::Noop => Self::new_noop(),
 			Action::Break => Self::new_break(),
-			Action::Pick | Action::Reword | Action::Edit | Action::Squash | Action::Drop => {
+			Action::Pick | Action::Reword | Action::Edit | Action::Squash | Action::Drop | Action::Index => {
 				Self::new(action, line_parser.next()?, line_parser.take_remaining(), None)
 			},
 			Action::Fixup => {
@@ -190,7 +190,7 @@ impl Line {
 	#[must_use]
 	pub(crate) const fn is_editable(&self) -> bool {
 		match self.action {
-			Action::Exec | Action::Label | Action::Reset | Action::Merge | Action::UpdateRef => true,
+			Action::Exec | Action::Index | Action::Label | Action::Reset | Action::Merge | Action::UpdateRef => true,
 			Action::Break
 			| Action::Drop
 			| Action::Edit
@@ -212,7 +212,7 @@ impl Line {
 	#[must_use]
 	pub(crate) fn to_text(&self) -> String {
 		match self.action {
-			Action::Drop | Action::Edit | Action::Fixup | Action::Pick | Action::Reword | Action::Squash => {
+			Action::Drop | Action::Edit | Action::Fixup | Action::Index | Action::Pick | Action::Reword | Action::Squash => {
 				if let Some(opt) = self.option.as_ref() {
 					format!("{} {opt} {} {}", self.action, self.hash, self.content)
 				}
