Selects if user wants to create a branch on a worktree or not

(#3621)

Author: sergeibbb

src/commands/git/branch.ts

@@ -377,7 +377,7 @@ export class BranchGitCommand extends QuickCommand {
 				state.reference = result;
 			}
 
-			if (state.counter < 4 || state.name == null) {
+			if (state.counter < 4 || state.name == null || state.suggestNameOnly) {
 				const result = yield* inputBranchNameStep(state, context, {
 					titleContext: ` from ${getReferenceLabel(state.reference, {
 						capitalize: true,

src/plus/startWork/startWork.ts

@@ -55,6 +55,7 @@ interface Context {
 interface State {
 	item?: StartWorkItem;
 	action?: StartWorkAction;
+	inWorktree?: boolean;
 }
 
 export type StartWorkAction = 'start';
@@ -128,6 +129,7 @@ export class StartWorkCommand extends QuickCommand<State> {
 				const result = yield* this.selectCommandStep(state);
 				if (result === StepResultBreak) continue;
 				state.action = result.action;
+				state.inWorktree = result.inWorktree;
 			}
 
 			if (state.counter < 2 && !state.action) {
@@ -154,7 +156,9 @@ export class StartWorkCommand extends QuickCommand<State> {
 									name: issue ? slug(`${issue.id}-${issue.title}`) : undefined,
 									suggestNameOnly: true,
 									suggestRepoOnly: true,
+									flags: state.inWorktree ? ['--worktree'] : ['--switch'],
 								},
+								confirm: false,
 							},
 							this.pickedVia,
 						);
@@ -173,14 +177,18 @@ export class StartWorkCommand extends QuickCommand<State> {
 		return state.counter < 0 ? StepResultBreak : undefined;
 	}
 
-	private *selectCommandStep(state: StepState<State>): StepResultGenerator<{ action?: StartWorkAction }> {
+	private *selectCommandStep(
+		state: StepState<State>,
+	): StepResultGenerator<{ action?: StartWorkAction; inWorktree?: boolean }> {
 		const step = createPickStep({
 			placeholder: 'Start work by creating a new branch',
 			items: [
 				createQuickPickItemOfT('Create a branch', {
 					action: 'start',
 				}),
+				createQuickPickItemOfT('Create a branch on a worktree', { action: 'start', inWorktree: true }),
 				createQuickPickItemOfT('Create a branch from an issue', {}),
+				createQuickPickItemOfT('Create a branch from an issue on a worktree', { inWorktree: true }),
 			],
 		});
 		const selection: StepSelection<typeof step> = yield step;