Context actions for chat sessions (#7627)

Author: osortega

package.json

@@ -1738,6 +1738,16 @@
 				"title": "%command.pr.refreshChatSessions.title%",
 				"icon": "$(refresh)",
 				"category": "%command.pull.request.category%"
+			},
+			{
+				"command": "pr.closeChatSessionPullRequest",
+				"title": "%command.pr.closeChatSessionPullRequest.title%",
+				"category": "%command.pull.request.category%"
+			},
+			{
+				"command": "pr.cancelCodingAgent",
+				"title": "%command.pr.cancelCodingAgent.title%",
+				"category": "%command.pull.request.category%"
 			}
 		],
 		"viewsWelcome": [
@@ -3379,6 +3389,16 @@
 					"command": "pr.openDescription",
 					"when": "chatSessionType == copilot-swe-agent",
 					"group": "inline"
+				},
+				{
+					"command": "pr.closeChatSessionPullRequest",
+					"when": "chatSessionType == copilot-swe-agent",
+					"group": "context"
+				},
+				{
+					"command": "pr.cancelCodingAgent",
+					"when": "chatSessionType == copilot-swe-agent",
+					"group": "context"
 				}
 			]
 		},

package.nls.json

@@ -321,6 +321,8 @@
 	"command.notifications.configureNotificationsViewlet.title": "Configure...",
 	"command.notification.chatSummarizeNotification.title": "Summarize With Copilot",
 	"command.codingAgent.openSessionLog.title": "Open Coding Agent Session Log",
+	"command.pr.closeChatSessionPullRequest.title": "Close Pull Request",
+	"command.pr.cancelCodingAgent.title": "Cancel Coding Agent",
 	"welcome.github.login.contents": {
 		"message": "You have not yet signed in with GitHub\n[Sign in](command:pr.signin)",
 		"comment": [

src/commands.ts

@@ -946,6 +946,41 @@ export function registerCommands(
 		await openDescription(telemetry, issueModel, descriptionNode, folderManager, revealDescription, !(argument instanceof RepositoryChangesNode), tree.notificationProvider);
 	}
 
+	async function closeChatSessionPullRequest(argument: ChatSessionWithPR) {
+		const pr = argument.pullRequest;
+		if (!pr) {
+			Logger.warn(`No pull request found in chat session`, logId);
+			return;
+		}
+		pr.close();
+		copilotRemoteAgentManager.refreshChatSessions();
+	}
+
+	async function cancelCodingAgent(argument: ChatSessionWithPR) {
+		const pr = argument.pullRequest;
+		if (!pr) {
+			Logger.warn(`No pull request found in chat session`, logId);
+			return;
+		}
+
+		copilotRemoteAgentManager.cancelMostRecentChatSession(pr);
+		// TODO: show a progress icon until the cancelation is finished
+	}
+
+	context.subscriptions.push(
+		vscode.commands.registerCommand(
+			'pr.closeChatSessionPullRequest',
+			closeChatSessionPullRequest
+		)
+	);
+
+	context.subscriptions.push(
+		vscode.commands.registerCommand(
+			'pr.cancelCodingAgent',
+			cancelCodingAgent
+		)
+	);
+
 	context.subscriptions.push(
 		vscode.commands.registerCommand(
 			'pr.openDescription',

src/github/copilotRemoteAgent.ts

@@ -1249,4 +1249,27 @@ export class CopilotRemoteAgentManager extends Disposable {
 	public refreshChatSessions(): void {
 		this._onDidChangeChatSessions.fire();
 	}
+
+	public async cancelMostRecentChatSession(pullRequest: PullRequestModel): Promise<void> {
+		const capi = await this.copilotApi;
+		if (!capi) {
+			Logger.warn(`No Copilot API instance found`);
+			return;
+		}
+
+		const folderManager = this.repositoriesManager.getManagerForIssueModel(pullRequest) ?? this.repositoriesManager.folderManagers[0];
+		if (!folderManager) {
+			Logger.warn(`No folder manager found for pull request`);
+			return;
+		}
+
+		const sessions = await capi.getAllSessions(pullRequest.id);
+		if (sessions.length > 0) {
+			const mostRecentSession = sessions[sessions.length - 1];
+			const folder = folderManager.gitHubRepositories.find(repo => repo.remote.remoteName === pullRequest.remote.remoteName);
+			folder?.cancelWorkflow(mostRecentSession.workflow_run_id);
+		} else {
+			Logger.warn(`No active chat session found for pull request ${pullRequest.id}`);
+		}
+	}
 }