Refactor: extract common logic from pr.pickOnVscodeDev and pr.pickOnCodespaces

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

Author: Copilot

src/commands.ts

@@ -694,48 +694,38 @@ export function registerCommands(
 		}
 	}));
 
-	context.subscriptions.push(
-		vscode.commands.registerCommand('pr.pickOnVscodeDev', async (pr: PRNode | RepositoryChangesNode | PullRequestModel) => {
-			if (pr === undefined) {
-				// This is unexpected, but has happened a few times.
-				Logger.error('Unexpectedly received undefined when picking a PR.', logId);
-				return vscode.window.showErrorMessage(vscode.l10n.t('No pull request was selected to checkout, please try again.'));
-			}
-
-			let pullRequestModel: PullRequestModel;
-
-			if (pr instanceof PRNode || pr instanceof RepositoryChangesNode) {
-				pullRequestModel = pr.pullRequestModel;
-			} else {
-				pullRequestModel = pr;
-			}
+	const pickPullRequest = async (pr: PRNode | RepositoryChangesNode | PullRequestModel, linkGenerator: (pr: PullRequestModel) => string, requiresHead: boolean = false) => {
+		if (pr === undefined) {
+			// This is unexpected, but has happened a few times.
+			Logger.error('Unexpectedly received undefined when picking a PR.', logId);
+			return vscode.window.showErrorMessage(vscode.l10n.t('No pull request was selected to checkout, please try again.'));
+		}
 
-			return vscode.env.openExternal(vscode.Uri.parse(vscodeDevPrLink(pullRequestModel)));
-		}),
-	);
+		let pullRequestModel: PullRequestModel;
 
-	context.subscriptions.push(
-		vscode.commands.registerCommand('pr.pickOnCodespaces', async (pr: PRNode | RepositoryChangesNode | PullRequestModel) => {
-			if (pr === undefined) {
-				// This is unexpected, but has happened a few times.
-				Logger.error('Unexpectedly received undefined when picking a PR.', logId);
-				return vscode.window.showErrorMessage(vscode.l10n.t('No pull request was selected to checkout, please try again.'));
-			}
+		if (pr instanceof PRNode || pr instanceof RepositoryChangesNode) {
+			pullRequestModel = pr.pullRequestModel;
+		} else {
+			pullRequestModel = pr;
+		}
 
-			let pullRequestModel: PullRequestModel;
+		if (requiresHead && !pullRequestModel.head) {
+			return vscode.window.showErrorMessage(vscode.l10n.t('Unable to checkout pull request: missing head branch information.'));
+		}
 
-			if (pr instanceof PRNode || pr instanceof RepositoryChangesNode) {
-				pullRequestModel = pr.pullRequestModel;
-			} else {
-				pullRequestModel = pr;
-			}
+		return vscode.env.openExternal(vscode.Uri.parse(linkGenerator(pullRequestModel)));
+	};
 
-			if (!pullRequestModel.head) {
-				return vscode.window.showErrorMessage(vscode.l10n.t('Unable to checkout pull request: missing head branch information.'));
-			}
+	context.subscriptions.push(
+		vscode.commands.registerCommand('pr.pickOnVscodeDev', (pr: PRNode | RepositoryChangesNode | PullRequestModel) =>
+			pickPullRequest(pr, vscodeDevPrLink)
+		),
+	);
 
-			return vscode.env.openExternal(vscode.Uri.parse(codespacesPrLink(pullRequestModel)));
-		}),
+	context.subscriptions.push(
+		vscode.commands.registerCommand('pr.pickOnCodespaces', (pr: PRNode | RepositoryChangesNode | PullRequestModel) =>
+			pickPullRequest(pr, codespacesPrLink, true)
+		),
 	);
 
 	context.subscriptions.push(vscode.commands.registerCommand('pr.checkoutOnVscodeDevFromDescription', async (context: OverviewContext | undefined) => {