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

(#3621)

Author: sergeibbb

src/plus/startWork/startWork.ts

@@ -55,6 +55,7 @@ interface Context {
 interface State {
 	item?: StartWorkItem;
 	action?: StartWorkAction;
+	inWorktree?: boolean;
 }
 
 type StartWorkStepState<T extends State = State> = RequireSome<StepState<T>, 'item'>;
@@ -136,6 +137,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;
 
 			await updateContextItems(this.container, context);
 
@@ -172,7 +174,9 @@ export class StartWorkCommand extends QuickCommand<State> {
 									name: issue ? slug(`${issue.id}-${issue.title}`) : undefined,
 									suggestNameOnly: true,
 									suggestRepoOnly: true,
+									flags: state.inWorktree ? ['--worktree'] : [],
 								},
+								confirm: !state.inWorktree,
 							},
 							this.pickedVia,
 						);
@@ -186,14 +190,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;