Support choosing checkout branch method when status is not empty

CHANGELOG.md

@@ -10,6 +10,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 * increase MSRV from 1.81 to 1.82 [[@cruessler](https://github.com/cruessler)]
 
 ### Added
+* support choosing checkout branch method when status is not empty [[@fatpandac](https://github.com/fatpandac)] ([#2404](https://github.com/extrawurst/gitui/issues/2404))
 * Support pre-push hook [[@xlai89](https://github.com/xlai89)] ([#1933](https://github.com/extrawurst/gitui/issues/1933))
 * Message tab supports pageUp and pageDown [[@xlai89](https://github.com/xlai89)] ([#2623](https://github.com/extrawurst/gitui/issues/2623))
 * Files and status tab support pageUp and pageDown [[@fatpandac](https://github.com/fatpandac)] ([#1951](https://github.com/extrawurst/gitui/issues/1951))

asyncgit/src/sync/branch/mod.rs

@@ -104,6 +104,11 @@ impl BranchInfo {
 
 None
 }
+
+/// returns whether branch is local
+pub const fn is_local(&self) -> bool {
+matches!(self.details, BranchDetails::Local(_))
+}
 }
 
 ///

asyncgit/src/sync/status.rs

@@ -280,3 +280,49 @@ pub fn get_status(
 
 Ok(res)
 }
+
+/// discard all changes in the working directory
+pub fn discard_status(repo_path: &RepoPath) -> Result<bool> {
+let repo = repo(repo_path)?;
+let commit = repo.head()?.peel_to_commit()?;
+
+repo.reset(commit.as_object(), git2::ResetType::Hard, None)?;
+
+Ok(true)
+}
+
+#[cfg(test)]
+mod tests {
+use super::*;
+use crate::sync::{commit, stage_add_file, tests::repo_init};
+use std::{fs::File, io::Write, path::Path};
+
+#[test]
+fn test_discard_status() {
+let file_path = Path::new("README.md");
+let (_td, repo) = repo_init().unwrap();
+let root = repo.path().parent().unwrap();
+let repo_path: &RepoPath =
+&root.as_os_str().to_str().unwrap().into();
+
+let mut file = File::create(root.join(file_path)).unwrap();
+
+// initial commit
+stage_add_file(repo_path, file_path).unwrap();
+commit(repo_path, "commit msg").unwrap();
+
+writeln!(file, "Test for discard_status").unwrap();
+
+let statuses =
+get_status(repo_path, StatusType::WorkingDir, None)
+.unwrap();
+assert_eq!(statuses.len(), 1);
+
+discard_status(repo_path).unwrap();
+
+let statuses =
+get_status(repo_path, StatusType::WorkingDir, None)
+.unwrap();
+assert_eq!(statuses.len(), 0);
+}
+}

src/app.rs

@@ -10,16 +10,16 @@ use crate::{
 options::{Options, SharedOptions},
 popup_stack::PopupStack,
 popups::{
-AppOption, BlameFilePopup, BranchListPopup, CommitPopup,
-CompareCommitsPopup, ConfirmPopup, CreateBranchPopup,
-CreateRemotePopup, ExternalEditorPopup, FetchPopup,
-FileRevlogPopup, FuzzyFindPopup, HelpPopup,
-InspectCommitPopup, LogSearchPopupPopup, MsgPopup,
-OptionsPopup, PullPopup, PushPopup, PushTagsPopup,
-RemoteListPopup, RenameBranchPopup, RenameRemotePopup,
-ResetPopup, RevisionFilesPopup, StashMsgPopup,
-SubmodulesListPopup, TagCommitPopup, TagListPopup,
-UpdateRemoteUrlPopup,
+AppOption, BlameFilePopup, BranchListPopup,
+CheckoutOptionPopup, CommitPopup, CompareCommitsPopup,
+ConfirmPopup, CreateBranchPopup, CreateRemotePopup,
+ExternalEditorPopup, FetchPopup, FileRevlogPopup,
+FuzzyFindPopup, HelpPopup, InspectCommitPopup,
+LogSearchPopupPopup, MsgPopup, OptionsPopup, PullPopup,
+PushPopup, PushTagsPopup, RemoteListPopup, RenameBranchPopup,
+RenameRemotePopup, ResetPopup, RevisionFilesPopup,
+StashMsgPopup, SubmodulesListPopup, TagCommitPopup,
+TagListPopup, UpdateRemoteUrlPopup,
 },
 queue::{
 Action, AppTabs, InternalEvent, NeedsUpdate, Queue,
@@ -98,6 +98,7 @@ pub struct App {
 submodule_popup: SubmodulesListPopup,
 tags_popup: TagListPopup,
 reset_popup: ResetPopup,
+checkout_option_popup: CheckoutOptionPopup,
 cmdbar: RefCell<CommandBar>,
 tab: usize,
 revlog: Revlog,
@@ -218,6 +219,7 @@ impl App {
 stashing_tab: Stashing::new(&env),
 stashlist_tab: StashList::new(&env),
 files_tab: FilesTab::new(&env),
+checkout_option_popup: CheckoutOptionPopup::new(&env),
 tab: 0,
 queue: env.queue,
 theme: env.theme,
@@ -493,6 +495,7 @@ impl App {
 fetch_popup,
 tag_commit_popup,
 reset_popup,
+checkout_option_popup,
 create_branch_popup,
 create_remote_popup,
 rename_remote_popup,
@@ -533,6 +536,7 @@ impl App {
 submodule_popup,
 tags_popup,
 reset_popup,
+checkout_option_popup,
 create_branch_popup,
 rename_branch_popup,
 revision_files_popup,
@@ -905,6 +909,9 @@ impl App {
 InternalEvent::CommitSearch(options) => {
 self.revlog.search(options);
 }
+InternalEvent::CheckoutOption(branch) => {
+self.checkout_option_popup.open(branch)?;
+}
 }
 
 Ok(flags)

src/popups/branchlist.rs

@@ -20,8 +20,9 @@ use asyncgit::{
 checkout_remote_branch, BranchDetails, LocalBranch,
 RemoteBranch,
 },
-checkout_branch, get_branches_info, BranchInfo, BranchType,
-CommitId, RepoPathRef, RepoState,
+checkout_branch, get_branches_info,
+status::StatusType,
+BranchInfo, BranchType, CommitId, RepoPathRef, RepoState,
 },
 AsyncGitNotification,
 };
@@ -582,23 +583,36 @@ impl BranchListPopup {
 anyhow::bail!("no valid branch selected");
 }
 
-if self.local {
-checkout_branch(
-&self.repo.borrow(),
-&self.branches[self.selection as usize].name,
-)?;
-self.hide();
+let status = sync::status::get_status(
+&self.repo.borrow(),
+StatusType::WorkingDir,
+None,
+)
+.expect("Could not get status");
+
+let selected_branch = &self.branches[self.selection as usize];
+if status.is_empty() {
+if self.local {
+checkout_branch(
+&self.repo.borrow(),
+&selected_branch.name,
+)?;
+self.hide();
+} else {
+checkout_remote_branch(
+&self.repo.borrow(),
+selected_branch,
+)?;
+self.local = true;
+self.update_branches()?;
+}
+self.queue.push(InternalEvent::Update(NeedsUpdate::ALL));
 } else {
-checkout_remote_branch(
-&self.repo.borrow(),
-&self.branches[self.selection as usize],
-)?;
-self.local = true;
-self.update_branches()?;
+self.queue.push(InternalEvent::CheckoutOption(
+selected_branch.clone(),
+));
 }
 
-self.queue.push(InternalEvent::Update(NeedsUpdate::ALL));
-
 Ok(())
 }