prompt to retry with force

Author: sergiolms

src/commands/git/branch.ts

@@ -1,12 +1,12 @@
 import { QuickInputButtons } from 'vscode';
 import type { Container } from '../../container';
-import { BranchError } from '../../git/errors';
+import { BranchError, BranchErrorReason } from '../../git/errors';
 import type { GitBranchReference, GitReference } from '../../git/models/reference';
 import { getNameWithoutRemote, getReferenceLabel, isRevisionReference } from '../../git/models/reference';
 import { Repository } from '../../git/models/repository';
 import type { GitWorktree } from '../../git/models/worktree';
 import { getWorktreesByBranch } from '../../git/models/worktree';
-import { showGenericErrorMessage } from '../../messages';
+import { showGenericErrorMessage, showGitBranchNotFullyMergedPrompt } from '../../messages';
 import type { QuickPickItemOfT } from '../../quickpicks/items/common';
 import { createQuickPickSeparator } from '../../quickpicks/items/common';
 import type { FlagsQuickPickItem } from '../../quickpicks/items/flags';
@@ -527,7 +527,23 @@ export class BranchGitCommand extends QuickCommand {
 				} catch (ex) {
 					// TODO likely need some better error handling here
 					Logger.error(ex);
-					return showGenericErrorMessage(ex);
+					if (ex instanceof BranchError && ex.reason === BranchErrorReason.BranchNotFullyMerged) {
+						const shouldRetryWithForce = await showGitBranchNotFullyMergedPrompt(ref.name);
+						if (shouldRetryWithForce) {
+							try {
+								await state.repo.git.deleteBranch(ref, {
+									force: true,
+									remote: state.flags.includes('--remotes'),
+								});
+							} catch (ex) {
+								Logger.error(ex);
+								await showGenericErrorMessage(ex);
+							}
+						}
+						continue;
+					}
+
+					await showGenericErrorMessage(ex);
 				}
 			}
 		}

src/messages.ts

@@ -137,7 +137,7 @@ export function showGitVersionUnsupportedErrorMessage(
 	);
 }
 
-export async function showGitBranchNotFullyMergedPrompt(branchName: string): Promise<MessageItem | undefined> {
+export async function showGitBranchNotFullyMergedPrompt(branchName: string): Promise<boolean> {
 	const confirm = { title: 'Retry with --force flag' };
 	const result = await showMessage(
 		'warn',