Adopt new command on chat context

Part of microsoft/vscode#280796

Author: alexr00

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

@@ -43,6 +43,11 @@ declare module 'vscode' {
 		 * The value of the context item. Can be omitted when returned from one of the `provide` methods if the provider supports `resolveChatContext`.
 		 */
 		value?: string;
+		/**
+		 * An optional command that is executed when the context item is clicked.
+		 * The original context item will be passed as an argument to the command.
+		 */
+		command?: Command;
 	}
 
 	export interface ChatContextProvider<T extends ChatContextItem = ChatContextItem> {

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 { OverviewContext } from './github/views';
+import { PRChatContextItem } from './lm/pullRequestContextProvider';
 import { isNotificationTreeItem, NotificationTreeItem } from './notifications/notificationItem';
 import { NotificationsManager } from './notifications/notificationsManager';
 import { PullRequestsTreeDataProvider } from './view/prsTreeDataProvider';
@@ -50,6 +51,7 @@ import {
 } from './view/treeNodes/fileChangeNode';
 import { PRNode } from './view/treeNodes/pullRequestNode';
 import { RepositoryChangesNode } from './view/treeNodes/repositoryChangesNode';
+import { IssueChatContextItem } from './lm/issueContextProvider';
 
 function ensurePR(folderRepoManager: FolderRepositoryManager, pr?: PRNode): PullRequestModel;
 function ensurePR<TIssue extends Issue, TIssueModel extends IssueModel<TIssue>>(folderRepoManager: FolderRepositoryManager, pr?: TIssueModel): TIssueModel;
@@ -895,7 +897,7 @@ export function registerCommands(
 		}),
 	);
 
-	async function openDescriptionCommand(argument: RepositoryChangesNode | PRNode | IssueModel | CrossChatSessionWithPR | undefined) {
+	async function openDescriptionCommand(argument: RepositoryChangesNode | PRNode | IssueModel | CrossChatSessionWithPR | PRChatContextItem | IssueChatContextItem | undefined) {
 		let issueModel: IssueModel | undefined;
 		if (!argument) {
 			const activePullRequests: PullRequestModel[] = reposManager.folderManagers
@@ -919,6 +921,10 @@ export function registerCommands(
 					number: argument.pullRequestDetails.number,
 					preventDefaultContextMenuItems: true,
 				}))?.pr;
+			} else if (PRChatContextItem.is(argument)) {
+				issueModel = argument.pr;
+			} else if (IssueChatContextItem.is(argument)) {
+				issueModel = argument.issue;
 			} else {
 				issueModel = argument;
 			}

src/lm/issueContextProvider.ts

@@ -11,10 +11,16 @@ import { RepositoriesManager } from '../github/repositoriesManager';
 import { getIssueNumberLabel } from '../github/utils';
 import { IssueQueryResult, StateManager } from '../issues/stateManager';
 
-interface IssueChatContextItem extends vscode.ChatContextItem {
+export interface IssueChatContextItem extends vscode.ChatContextItem {
 	issue: IssueModel;
 }
 
+export namespace IssueChatContextItem {
+	export function is(item: unknown): item is IssueChatContextItem {
+		return (item as IssueChatContextItem).issue !== undefined;
+	}
+}
+
 export class IssueContextProvider implements vscode.ChatContextProvider {
 	constructor(private readonly _stateManager: StateManager,
 		private readonly _reposManager: RepositoriesManager
@@ -66,6 +72,10 @@ export class IssueContextProvider implements vscode.ChatContextProvider {
 			label: `#${issue.number} ${issue.title}`,
 			modelDescription: 'The GitHub issue the user is viewing.',
 			issue,
+			command: {
+				command: 'issue.openDescription',
+				title: vscode.l10n.t('Open Issue')
+			}
 		};
 	}
 

src/lm/pullRequestContextProvider.ts

@@ -12,10 +12,16 @@ import { PullRequestOverviewPanel } from '../github/pullRequestOverview';
 import { RepositoriesManager } from '../github/repositoriesManager';
 import { PrsTreeModel } from '../view/prsTreeModel';
 
-interface PRChatContextItem extends vscode.ChatContextItem {
+export interface PRChatContextItem extends vscode.ChatContextItem {
 	pr?: PullRequestModel;
 }
 
+export namespace PRChatContextItem {
+	export function is(item: unknown): item is PRChatContextItem {
+		return (item as PRChatContextItem).pr !== undefined;
+	}
+}
+
 export class PullRequestContextProvider extends Disposable implements vscode.ChatContextProvider {
 	private readonly _onDidChangeWorkspaceChatContext = new vscode.EventEmitter<void>();
 	readonly onDidChangeWorkspaceChatContext = this._onDidChangeWorkspaceChatContext.event;
@@ -112,6 +118,10 @@ Active pull request (may not be the same as open pull request): ${folderManager.
 			label: `#${pr.number} ${pr.title}`,
 			modelDescription: 'The GitHub pull request the user is viewing.',
 			pr,
+			command: {
+				command: 'pr.openDescription',
+				title: vscode.l10n.t('Open Pull Request')
+			}
 		};
 	}