Adding chat sessions provider (#7354)

* Adding chat sessions provider

* Review comments

* Missing change

* Review comments

* Revert change

* Update query

Author: osortega

package.json

@@ -14,6 +14,7 @@
 		"activeComment",
 		"chatParticipantAdditions",
 		"chatParticipantPrivate",
+		"chatSessionsProvider",
 		"codiconDecoration",
 		"codeActionRanges",
 		"commentingRangeHint",
@@ -3273,6 +3274,12 @@
 					"command": "review.requestChangesOnDotComDescription",
 					"when": "webviewId == PullRequestOverview && github:reviewCommentMenu && github:reviewCommentRequestChangesOnDotCom"
 				}
+			],
+			"chat/chatSessions": [
+				{
+					"command": "pr.openDescription",
+					"when": "chatSessionType == github.pullRequest.codingAgent"
+				}
 			]
 		},
 		"colors": [

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

@@ -0,0 +1,47 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+declare module 'vscode' {
+	/**
+	 * Provides a list of chat sessions
+	 */
+	export interface ChatSessionsProvider extends Disposable {
+		/**
+		 * Type to identify providers.
+		 */
+		readonly chatSessionType: string;
+
+		/**
+		 * Fired when chat sessions change.
+		 */
+		readonly onDidChangeChatSessionContent: Event<void>;
+
+		/**
+		 * Provide a list of chat sessions.
+		 * */
+		provideChatSessions(token: CancellationToken): Thenable<ChatSessionContent[]>;
+	}
+
+	export interface ChatSessionContent {
+		/**
+		 * Identifies the session
+		 *		 */
+		uri: Uri;
+
+		/**
+		 * Human readable name of the session shown in the UI
+		 */
+		label: string;
+
+		/**
+		 * An icon for the participant shown in UI.
+		 */
+		iconPath?: IconPath;
+	}
+
+	export namespace chat {
+		export function registerChatSessionsProvider(provider: ChatSessionsProvider): Disposable;
+	}
+}

src/commands.ts

@@ -18,6 +18,7 @@ import { ITelemetry } from './common/telemetry';
 import { asTempStorageURI, fromPRUri, fromReviewUri, Schemes, toPRUri } from './common/uri';
 import { formatError } from './common/utils';
 import { EXTENSION_ID } from './constants';
+import { ChatSessionWithPR } from './github/copilotApi';
 import { CopilotRemoteAgentManager, ICopilotRemoteAgentCommandArgs } from './github/copilotRemoteAgent';
 import { FolderRepositoryManager } from './github/folderRepositoryManager';
 import { GitHubRepository } from './github/githubRepository';
@@ -201,6 +202,11 @@ export async function closeAllPrAndReviewEditors() {
 	}
 }
 
+function isChatSessionWithPR(value: any): value is ChatSessionWithPR {
+	const asChatSessionWithPR = value as Partial<ChatSessionWithPR>;
+	return !!asChatSessionWithPR.pullRequest;
+}
+
 export function registerCommands(
 	context: vscode.ExtensionContext,
 	reposManager: RepositoriesManager,
@@ -835,7 +841,7 @@ export function registerCommands(
 		}),
 	);
 
-	async function openDescriptionCommand(argument: RepositoryChangesNode | PRNode | IssueModel | undefined) {
+	async function openDescriptionCommand(argument: RepositoryChangesNode | PRNode | IssueModel | ChatSessionWithPR | undefined) {
 		let issueModel: IssueModel | undefined;
 		if (!argument) {
 			const activePullRequests: PullRequestModel[] = reposManager.folderManagers
@@ -852,6 +858,8 @@ export function registerCommands(
 				issueModel = argument.pullRequestModel;
 			} else if (argument instanceof PRNode) {
 				issueModel = argument.pullRequestModel;
+			} else if (isChatSessionWithPR(argument)) {
+				issueModel = argument.pullRequest;
 			} else {
 				issueModel = argument;
 			}

src/extension.ts

@@ -411,6 +411,15 @@ async function deferredActivate(context: vscode.ExtensionContext, showPRControll
 
 	const copilotRemoteAgentManager = new CopilotRemoteAgentManager(credentialStore, reposManager, telemetry);
 	context.subscriptions.push(copilotRemoteAgentManager);
+	context.subscriptions.push(vscode.chat.registerChatSessionsProvider({
+		chatSessionType: 'github.pullRequest.codingAgent',
+		provideChatSessions: async (token) => {
+			return await copilotRemoteAgentManager.provideChatSessions(token);
+		},
+		// Events not used yet, but required by interface.
+		onDidChangeChatSessionContent: new vscode.EventEmitter<void>().event,
+		dispose: () => { }
+	}));
 
 	const prTree = new PullRequestsTreeDataProvider(telemetry, context, reposManager, copilotRemoteAgentManager);
 	context.subscriptions.push(prTree);

src/github/copilotApi.ts

@@ -9,8 +9,11 @@ import * as vscode from 'vscode';
 import { AuthProvider } from '../common/authentication';
 import Logger from '../common/logger';
 import { ITelemetry } from '../common/telemetry';
-import { CredentialStore } from './credentials';
+import { CredentialStore, GitHub } from './credentials';
+import { PRType } from './interface';
 import { LoggingOctokit } from './loggingOctokit';
+import { PullRequestModel } from './pullRequestModel';
+import { RepositoriesManager } from './repositoriesManager';
 import { hasEnterpriseUri } from './utils';
 
 const LEARN_MORE_URL = 'https://docs.github.com/en/copilot/how-tos/agents/copilot-coding-agent';
@@ -35,10 +38,19 @@ export interface RemoteAgentJobResponse {
 	}
 }
 
+export interface ChatSessionWithPR extends vscode.ChatSessionContent {
+	pullRequest: PullRequestModel;
+}
+
 export class CopilotApi {
 	protected static readonly ID = 'copilotApi';
 
-	constructor(private octokit: LoggingOctokit, private token: string, private telemetry: ITelemetry) { }
+	constructor(
+		private octokit: LoggingOctokit,
+		private token: string,
+		private credentialStore: CredentialStore,
+		private telemetry: ITelemetry
+	) { }
 
 	private get baseUrl(): string {
 		return 'https://api.githubcopilot.com';
@@ -173,6 +185,23 @@ export class CopilotApi {
 		return sessions.sessions;
 	}
 
+	public async getAllCodingAgentPRs(repositoriesManager: RepositoriesManager): Promise<PullRequestModel[]> {
+		const hub = this.getHub();
+		const username = (await hub?.currentUser)?.login;
+		if (!username) {
+			Logger.error('Failed to get GitHub username from auth provider', CopilotApi.ID);
+			return [];
+		}
+		const query = `is:open author:copilot-swe-agent[bot] assignee:${username} is:pr`;
+		const allItems = await Promise.all(
+			repositoriesManager.folderManagers.map(async fm => {
+				const result = await fm.getPullRequests(PRType.Query, undefined, query);
+				return result.items;
+			})
+		);
+		return allItems.flat();
+	}
+
 	public async getSessionInfo(sessionId: string): Promise<SessionInfo> {
 		const response = await fetch(`https://api.githubcopilot.com/agents/sessions/${sessionId}`, {
 			method: 'GET',
@@ -201,6 +230,19 @@ export class CopilotApi {
 		}
 		return await logsResponse.text();
 	}
+
+	private getHub(): GitHub | undefined {
+		let authProvider: AuthProvider | undefined;
+		if (this.credentialStore.isAuthenticated(AuthProvider.githubEnterprise) && hasEnterpriseUri()) {
+			authProvider = AuthProvider.githubEnterprise;
+		} else if (this.credentialStore.isAuthenticated(AuthProvider.github)) {
+			authProvider = AuthProvider.github;
+		} else {
+			return;
+		}
+
+		return this.credentialStore.getHub(authProvider);
+	}
 }
 
 
@@ -242,5 +284,5 @@ export async function getCopilotApi(credentialStore: CredentialStore, telemetry:
 	}
 
 	const { token } = await github.octokit.api.auth() as { token: string };
-	return new CopilotApi(github.octokit, token, telemetry);
+	return new CopilotApi(github.octokit, token, credentialStore, telemetry);
 }

src/github/copilotRemoteAgent.ts

@@ -14,7 +14,7 @@ import { CODING_AGENT, CODING_AGENT_AUTO_COMMIT_AND_PUSH, CODING_AGENT_ENABLED }
 import { ITelemetry } from '../common/telemetry';
 import { toOpenPullRequestWebviewUri } from '../common/uri';
 import { OctokitCommon } from './common';
-import { CopilotApi, getCopilotApi, RemoteAgentJobPayload, SessionInfo } from './copilotApi';
+import { ChatSessionWithPR, CopilotApi, getCopilotApi, RemoteAgentJobPayload, SessionInfo } from './copilotApi';
 import { CopilotPRWatcher, CopilotStateModel } from './copilotPrWatcher';
 import { CredentialStore } from './credentials';
 import { FolderRepositoryManager } from './folderRepositoryManager';
@@ -681,4 +681,31 @@ export class CopilotRemoteAgentManager extends Disposable {
 	getCounts(): { total: number; inProgress: number; error: number } {
 		return this._stateModel.getCounts();
 	}
+
+	public async provideChatSessions(token: vscode.CancellationToken): Promise<ChatSessionWithPR[]> {
+		try {
+			const capi = await this.copilotApi;
+			if (!capi) {
+				return [];
+			}
+
+			// Check if the token is already cancelled
+			if (token.isCancellationRequested) {
+				return [];
+			}
+
+			const sessions = await capi.getAllCodingAgentPRs(this.repositoriesManager);
+			return sessions.map(session => {
+				return {
+					uri: vscode.Uri.parse(`github://coding-agent/${session.id}`),
+					label: session.title || `Session ${session.id}`,
+					iconPath: undefined,
+					pullRequest: session
+				};
+			});
+		} catch (error) {
+			Logger.error(`Failed to provide coding agents information: ${error}`, CopilotRemoteAgentManager.ID);
+		}
+		return [];
+	}
 }