feat: support choosing checkout branch method

Allow choosing a checkout branch method when status is not empty

Author: Fatpandac

src/app.rs

@@ -19,7 +19,7 @@ use crate::{
 		RemoteListPopup, RenameBranchPopup, RenameRemotePopup,
 		ResetPopup, RevisionFilesPopup, StashMsgPopup,
 		SubmodulesListPopup, TagCommitPopup, TagListPopup,
-		UpdateRemoteUrlPopup,
+		UpdateRemoteUrlPopup, CheckoutOptionPopup
 	},
 	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,
@@ -904,7 +908,10 @@ impl App {
 			}
 			InternalEvent::CommitSearch(options) => {
 				self.revlog.search(options);
-			}
+			},
+      InternalEvent::CheckoutOption(branch, is_local) => {
+        self.checkout_option_popup.open(branch, is_local)?;
+      }
 		};
 
 		Ok(flags)

src/popups/branchlist.rs

@@ -13,6 +13,7 @@ use crate::{
 	ui::{self, Size},
 };
 use anyhow::Result;
+use asyncgit::sync::status::StatusType;
 use asyncgit::{
 	sync::{
 		self,
@@ -582,23 +583,37 @@ 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,
+		)
+		.unwrap();
+
+		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.local.clone()
+			));
 		}
 
-		self.queue.push(InternalEvent::Update(NeedsUpdate::ALL));
-
 		Ok(())
 	}