Add checkout pull request in worktree option (#8513)

* Initial plan

* initial plan for checkout PR in worktree feature

Co-authored-by: alexr00 <38270282+alexr00@users.noreply.github.com>

* feat: add checkout pull request in worktree option

Co-authored-by: alexr00 <38270282+alexr00@users.noreply.github.com>

* fix: use VS Code Tasks API for reliable cross-platform worktree creation

Co-authored-by: alexr00 <38270282+alexr00@users.noreply.github.com>

* fix: address code review feedback - improve error handling and branch naming

Co-authored-by: alexr00 <38270282+alexr00@users.noreply.github.com>

* argument clean up and keep branch name

* Update git API

* refactor: use git extension API for worktree creation instead of shell execution

Co-authored-by: alexr00 <38270282+alexr00@users.noreply.github.com>

* fix: run fetch and worktree selection in parallel, move info message after progress

Co-authored-by: alexr00 <38270282+alexr00@users.noreply.github.com>

* Fix createWorktree call

* fix: start progress before fetch operation, include fetch in progress scope

Co-authored-by: alexr00 <38270282+alexr00@users.noreply.github.com>

* feat: add 'Checkout in Worktree' option to PR Description checkout button

Agent-Logs-Url: https://github.com/microsoft/vscode-pull-request-github/sessions/113a0bba-a0a7-47d9-baca-dd4dd5fda645

* fix: handle existing local branch when creating worktree

Agent-Logs-Url: https://github.com/microsoft/vscode-pull-request-github/sessions/c7f23b9d-14de-42d2-960a-36aa53e2d704

Co-authored-by: alexr00 <38270282+alexr00@users.noreply.github.com>

* feat: use modal dialog with new window and current window options for worktree

Agent-Logs-Url: https://github.com/microsoft/vscode-pull-request-github/sessions/b2dbdafb-3aa3-4d09-8555-21772c09c7e9

Co-authored-by: alexr00 <38270282+alexr00@users.noreply.github.com>

* feat: add pr.pickInWorktreeFromDescription command for PR overview checkout menu

Agent-Logs-Url: https://github.com/microsoft/vscode-pull-request-github/sessions/32630df4-6da8-4847-8013-e2fb1dabc1bb

Co-authored-by: alexr00 <38270282+alexr00@users.noreply.github.com>

* refactor: extract worktree checkout logic into standalone utility function

Agent-Logs-Url: https://github.com/microsoft/vscode-pull-request-github/sessions/ddf40003-0929-42c3-abfb-5abf98150f00

Co-authored-by: alexr00 <38270282+alexr00@users.noreply.github.com>

* Fix strings

* fix strings

* fix: stop if fetch fails and use commands.openFolder helper

Agent-Logs-Url: https://github.com/microsoft/vscode-pull-request-github/sessions/6af4fefc-4eff-4f7e-902b-52be43a94c15

Co-authored-by: alexr00 <38270282+alexr00@users.noreply.github.com>

* CCR cleanup

---------

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: alexr00 <38270282+alexr00@users.noreply.github.com>

Author: Copilot

package.json

@@ -1013,6 +1013,18 @@
 				"category": "%command.pull.request.category%",
 				"icon": "$(cloud)"
 			},
+			{
+				"command": "pr.pickInWorktree",
+				"title": "%command.pr.pickInWorktreeFromDescription.title%",
+				"category": "%command.pull.request.category%",
+				"icon": "$(folder-library)"
+			},
+			{
+				"command": "pr.pickInWorktreeFromDescription",
+				"title": "%command.pr.pickInWorktree.title%",
+				"category": "%command.pull.request.category%",
+				"icon": "$(folder-library)"
+			},
 			{
 				"command": "pr.exit",
 				"title": "%command.pr.exit.title%",
@@ -2084,6 +2096,14 @@
 					"command": "pr.pickOnCodespaces",
 					"when": "false"
 				},
+				{
+					"command": "pr.pickInWorktree",
+					"when": "false"
+				},
+				{
+					"command": "pr.pickInWorktreeFromDescription",
+					"when": "false"
+				},
 				{
 					"command": "pr.exit",
 					"when": "github:inReviewMode"
@@ -2882,6 +2902,11 @@
 					"when": "view == pr:github && viewItem =~ /pullrequest(:local)?:nonactive/ && (!isWeb || remoteName != codespaces && virtualWorkspace != vscode-vfs)",
 					"group": "1_pullrequest@3"
 				},
+				{
+					"command": "pr.pickInWorktree",
+					"when": "view == pr:github && viewItem =~ /pullrequest(:local)?:nonactive/ && !isWeb",
+					"group": "1_pullrequest@4"
+				},
 				{
 					"command": "pr.openChanges",
 					"when": "view =~ /(pr|prStatus):github/ && viewItem =~ /(pullrequest|description)/",
@@ -3641,13 +3666,18 @@
 					"when": "webviewId == PullRequestOverview && github:checkoutMenu"
 				},
 				{
-					"command": "pr.checkoutOnVscodeDevFromDescription",
+					"command": "pr.pickInWorktreeFromDescription",
 					"group": "checkout@1",
+					"when": "webviewId == PullRequestOverview && github:checkoutMenu && !isWeb"
+				},
+				{
+					"command": "pr.checkoutOnVscodeDevFromDescription",
+					"group": "checkout@2",
 					"when": "webviewId == PullRequestOverview && github:checkoutMenu"
 				},
 				{
 					"command": "pr.checkoutOnCodespacesFromDescription",
-					"group": "checkout@2",
+					"group": "checkout@3",
 					"when": "webviewId == PullRequestOverview && github:checkoutMenu"
 				},
 				{

package.nls.json

@@ -215,6 +215,8 @@
 	"command.pr.openChanges.title": "Open Changes",
 	"command.pr.pickOnVscodeDev.title": "Checkout Pull Request on vscode.dev",
 	"command.pr.pickOnCodespaces.title": "Checkout Pull Request on Codespaces",
+	"command.pr.pickInWorktree.title": "Checkout in Worktree",
+	"command.pr.pickInWorktreeFromDescription.title": "Checkout Pull Request in Worktree",
 	"command.pr.exit.title": "Checkout Default Branch",
 	"command.pr.dismissNotification.title": "Dismiss Notification",
 	"command.pr.markAllCopilotNotificationsAsRead.title": "Dismiss All Copilot Notifications",

src/@types/vscode.proposed.chatParticipantAdditions.d.ts

@@ -453,6 +453,11 @@ declare module 'vscode' {
 		constructor(value: string | MarkdownString);
 	}
 
+	export class ChatResponseInfoPart {
+		value: MarkdownString;
+		constructor(value: string | MarkdownString);
+	}
+
 	export class ChatResponseProgressPart2 extends ChatResponseProgressPart {
 		value: string;
 		task?: (progress: Progress<ChatResponseWarningPart | ChatResponseReferencePart>) => Thenable<string | void>;
@@ -633,6 +638,15 @@ declare module 'vscode' {
 		 */
 		warning(message: string | MarkdownString): void;
 
+		/**
+		 * Push an info banner to this stream. Short-hand for
+		 * `push(new ChatResponseInfoPart(message))`.
+		 *
+		 * @param message An informational message
+		 * @returns This stream.
+		 */
+		info(message: string | MarkdownString): void;
+
 		reference(value: Uri | Location | { variableName: string; value?: Uri | Location }, iconPath?: Uri | ThemeIcon | { light: Uri; dark: Uri }): void;
 
 		reference2(value: Uri | Location | string | { variableName: string; value?: Uri | Location }, iconPath?: Uri | ThemeIcon | { light: Uri; dark: Uri }, options?: { status?: { description: string; kind: ChatResponseReferencePartStatusKind } }): void;

src/@types/vscode.proposed.chatSessionsProvider.d.ts

@@ -76,6 +76,23 @@ declare module 'vscode' {
 		// TODO: Do we need a flag to try auth if needed?
 		provideChatSessionItems(token: CancellationToken): ProviderResult<ChatSessionItem[]>;
 
+		/**
+		 * @deprecated Use {@linkcode ChatSessionItemController.resolveChatSessionItem} instead.
+		 *
+		 * Given a chat session item fill in more data, like {@link ChatSessionItem.timing timing},
+		 * {@link ChatSessionItem.changes changes}, or {@link ChatSessionItem.badge badge}.
+		 *
+		 * The editor will call this when a chat session item becomes visible in the UI, for example
+		 * when the user scrolls to it or when it is first rendered.
+		 *
+		 * @param item A chat session item currently visible in the UI. Treat this as read-only.
+		 * @param token A cancellation token.
+		 * @returns A new {@link ChatSessionItem} instance (or a thenable that resolves to one) with the
+		 * same `resource` as `item` and any additional properties filled in. When no result is returned,
+		 * the given `item` is left unchanged.
+		 */
+		resolveChatSessionItem?: (item: ChatSessionItem, token: CancellationToken) => ProviderResult<ChatSessionItem>;
+
 		// #region Unstable parts of API
 
 		/**
@@ -100,11 +117,6 @@ declare module 'vscode' {
 			readonly command?: string;
 		};
 
-		/**
-		 * @deprecated Use `inputState` instead
-		 */
-		readonly sessionOptions: ReadonlyArray<{ optionId: string; value: string | ChatSessionProviderOptionItem }>;
-
 		readonly inputState: ChatSessionInputState;
 	}
 
@@ -199,6 +211,26 @@ declare module 'vscode' {
 		 */
 		getChatSessionInputState?: ChatSessionControllerGetInputState;
 
+		/**
+		 * Called to fill in more data on a chat session item, like {@link ChatSessionItem.timing timing},
+		 * {@link ChatSessionItem.changes changes}, or {@link ChatSessionItem.badge badge}.
+		 *
+		 * The editor will call this when a chat session item becomes visible in the UI, for example
+		 * when the user scrolls to it or when it is first rendered.
+		 *
+		 * The editor will only resolve a chat session item once, unless the item is updated via
+		 * {@link ChatSessionItemCollection.add add} or {@link ChatSessionItemCollection.replace replace},
+		 * which invalidates the resolve cache.
+		 *
+		 * The handler should update the item in the {@link ChatSessionItemController.items items collection} via
+		 * {@link ChatSessionItemCollection.add add}. The editor picks up the updated item from
+		 * the collection after the returned thenable resolves.
+		 *
+		 * @param item A chat session item currently visible in the UI.
+		 * @param token A cancellation token.
+		 */
+		resolveChatSessionItem?: (item: ChatSessionItem, token: CancellationToken) => Thenable<void>;
+
 		/**
 		 * Create a new managed ChatSessionInputState object.
 		 */
@@ -505,11 +537,6 @@ declare module 'vscode' {
 		 */
 		provideChatSessionContent(resource: Uri, token: CancellationToken, context: {
 			readonly inputState: ChatSessionInputState;
-
-			/**
-			 * @deprecated Use `inputState` instead
-			 */
-			readonly sessionOptions: ReadonlyArray<{ optionId: string; value: string | ChatSessionProviderOptionItem }>;
 		}): Thenable<ChatSession> | ChatSession;
 
 		/**
@@ -621,6 +648,16 @@ declare module 'vscode' {
 		 * Only one item per option group should be marked as default.
 		 */
 		readonly default?: boolean;
+
+		/**
+		 * Optional slash-command alias (without leading `/`) that selects this option
+		 * when the user submits `/<slashCommand>`. Does not send a chat request; only
+		 * updates the selection so the next prompt runs with this option active.
+		 *
+		 * Scoped to chat sessions owned by the contributing provider. Names must be
+		 * unique across the provider's groups; on conflict, the first declared wins.
+		 */
+		readonly slashCommand?: string;
 	}
 
 	/**
@@ -678,6 +715,22 @@ declare module 'vscode' {
 		 * `{ inputState: ChatSessionInputState; sessionResource: Uri | undefined }` that they can use to determine which session and options they are being invoked for.
 		 */
 		readonly commands?: Command[];
+
+		/**
+		 * Optional kind that hints how this option group should be presented in the UI.
+		 *
+		 * - `'permissions'`: The group represents tool-approval permissions for the session.
+		 *   The editor will not render this group as its own picker. Instead, its items
+		 *   replace the built-in items in the chat permission picker for the session,
+		 *   and the user's selection is reported back through the standard
+		 *   {@link ChatSessionContentProvider.handleChatSessionOptionsChange} flow.
+		 *   At most one option group per provider may use this kind; if more than one is
+		 *   declared, the first one (in declaration order) is used. The group is invisible
+		 *   if the chat permission picker itself is hidden by other `when` clauses.
+		 *
+		 * When omitted, the group is rendered as a standalone picker as usual.
+		 */
+		readonly kind?: 'permissions';
 	}
 
 	export interface ChatSessionProviderOptions {

src/commands.ts

@@ -32,6 +32,7 @@ import { chooseItem } from './github/quickPicks';
 import { RepositoriesManager } from './github/repositoriesManager';
 import { codespacesPrLink, getIssuesUrl, getPullsUrl, isInCodespaces, ISSUE_OR_URL_EXPRESSION, parseIssueExpressionOutput, vscodeDevPrLink } from './github/utils';
 import { BaseContext, OverviewContext } from './github/views';
+import { checkoutPRInWorktree } from './github/worktree';
 import { IssueChatContextItem } from './lm/issueContextProvider';
 import { PRChatContextItem } from './lm/pullRequestContextProvider';
 import { isNotificationTreeItem, NotificationTreeItem } from './notifications/notificationItem';
@@ -825,6 +826,47 @@ export function registerCommands(
 		),
 	);
 
+	context.subscriptions.push(
+		vscode.commands.registerCommand('pr.pickInWorktree', async (pr: PRNode | PullRequestModel | unknown) => {
+			if (pr === undefined) {
+				Logger.error('Unexpectedly received undefined when picking a PR for worktree checkout.', logId);
+				return vscode.window.showErrorMessage(vscode.l10n.t('No pull request was selected to checkout, please try again.'));
+			}
+
+			let pullRequestModel: PullRequestModel;
+			let repository: Repository | undefined;
+
+			if (pr instanceof PRNode) {
+				pullRequestModel = pr.pullRequestModel;
+				repository = pr.repository;
+			} else if (pr instanceof PullRequestModel) {
+				pullRequestModel = pr;
+			} else {
+				Logger.error('Unexpectedly received unknown type when picking a PR for worktree checkout.', logId);
+				return vscode.window.showErrorMessage(vscode.l10n.t('No pull request was selected to checkout, please try again.'));
+			}
+
+			// Get the folder manager to access the repository
+			const folderManager = reposManager.getManagerForIssueModel(pullRequestModel);
+			if (!folderManager) {
+				return vscode.window.showErrorMessage(vscode.l10n.t('Unable to find repository for this pull request.'));
+			}
+
+			return checkoutPRInWorktree(telemetry, folderManager, pullRequestModel, repository);
+		}),
+	);
+
+	context.subscriptions.push(vscode.commands.registerCommand('pr.pickInWorktreeFromDescription', async (ctx: BaseContext | undefined) => {
+		if (!ctx) {
+			return vscode.window.showErrorMessage(vscode.l10n.t('No pull request context provided for checkout.'));
+		}
+		const resolved = await resolvePr(ctx);
+		if (!resolved) {
+			return vscode.window.showErrorMessage(vscode.l10n.t('Unable to resolve pull request for checkout.'));
+		}
+		return checkoutPRInWorktree(telemetry, resolved.folderManager, resolved.pr, undefined);
+	}));
+
 	context.subscriptions.push(vscode.commands.registerCommand('pr.checkoutOnVscodeDevFromDescription', async (context: BaseContext | undefined) => {
 		if (!context) {
 			return vscode.window.showErrorMessage(vscode.l10n.t('No pull request context provided for checkout.'));