Add "Mark All as Read" context menu for Copilot on My Behalf PRs (#7859)

* Initial plan

* Initial exploration and plan for Copilot PR mark all as read feature

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

* Implement Mark All as Read feature for Copilot notifications

* Improve contextValue assignment logic for better readability

* Changes before error encountered

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

* Code review fixes: use makeKey, existing getNotificationsCount method, and repo getter

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

* Make it work

---------

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

@@ -948,6 +948,12 @@
 				"title": "%command.pr.dismissNotification.title%",
 				"category": "%command.pull.request.category%"
 			},
+			{
+				"command": "pr.markAllCopilotNotificationsAsRead",
+				"title": "Mark All as Read",
+				"category": "%command.pull.request.category%",
+				"enablement": "viewItem == copilot-query-with-notifications"
+			},
 			{
 				"command": "pr.merge",
 				"title": "%command.pr.merge.title%",
@@ -1998,6 +2004,10 @@
 					"command": "pr.dismissNotification",
 					"when": "false"
 				},
+				{
+					"command": "pr.markAllCopilotNotificationsAsRead",
+					"when": "false"
+				},
 				{
 					"command": "pr.resetViewedFiles",
 					"when": "github:inReviewMode"
@@ -2786,6 +2796,11 @@
 					"when": "view == pr:github && viewItem =~ /pullrequest(.*):notification/",
 					"group": "4_pullrequest@5"
 				},
+				{
+					"command": "pr.markAllCopilotNotificationsAsRead",
+					"when": "view == pr:github && viewItem =~ /copilot-query/",
+					"group": "0_category@1"
+				},
 				{
 					"command": "issue.chatSummarizeIssue",
 					"when": "view == pr:github && viewItem =~ /pullrequest/ && github.copilot-chat.activated && config.githubPullRequests.experimental.chat",

src/commands.ts

@@ -917,6 +917,14 @@ export function registerCommands(
 		}),
 	);
 
+	context.subscriptions.push(
+		vscode.commands.registerCommand('pr.markAllCopilotNotificationsAsRead', node => {
+			if (node instanceof CategoryTreeNode && node.isCopilot && node.repo) {
+				copilotRemoteAgentManager.clearAllNotifications(node.repo.owner, node.repo.repositoryName);
+			}
+		}),
+	);
+
 	async function openDescriptionCommand(argument: RepositoryChangesNode | PRNode | IssueModel | ChatSessionWithPR | undefined) {
 		let issueModel: IssueModel | undefined;
 		if (!argument) {

src/github/copilotPrWatcher.ts

@@ -107,6 +107,41 @@ export class CopilotStateModel extends Disposable {
 		}
 	}
 
+	clearAllNotifications(owner?: string, repo?: string): void {
+		if (this._showNotification.size > 0) {
+			const items: PullRequestModel[] = [];
+
+			// If owner and repo are specified, only clear notifications for that repo
+			if (owner && repo) {
+				const keysToRemove: string[] = [];
+				const prefix = `${this.makeKey(owner, repo)}#`;
+				for (const key of this._showNotification.keys()) {
+					if (key.startsWith(prefix)) {
+						const item = this._states.get(key)?.item;
+						if (item) {
+							items.push(item);
+						}
+						keysToRemove.push(key);
+					}
+				}
+				keysToRemove.forEach(key => this._showNotification.delete(key));
+			} else {
+				// Clear all notifications
+				for (const key of this._showNotification.keys()) {
+					const item = this._states.get(key)?.item;
+					if (item) {
+						items.push(item);
+					}
+				}
+				this._showNotification.clear();
+			}
+
+			if (items.length > 0) {
+				this._onDidChangeNotifications.fire(items);
+			}
+		}
+	}
+
 	get notifications(): ReadonlySet<string> {
 		return this._showNotification;
 	}

src/github/copilotRemoteAgent.ts

@@ -881,6 +881,11 @@ export class CopilotRemoteAgentManager extends Disposable {
 		return this._stateModel.notifications.size;
 	}
 
+
+	public clearAllNotifications(owner?: string, repo?: string): void {
+		this._stateModel.clearAllNotifications(owner, repo);
+	}
+
 	hasNotification(owner: string, repo: string, pullRequestNumber?: number): boolean {
 		if (pullRequestNumber !== undefined) {
 			const key = this._stateModel.makeKey(owner, repo, pullRequestNumber);

src/view/treeNodes/categoryNode.ts

@@ -179,7 +179,7 @@ export class CategoryTreeNode extends TreeNode implements vscode.TreeItem {
 		}
 
 		if (this._categoryQuery) {
-			this.contextValue = 'query';
+			this.contextValue = this.isCopilot ? 'copilot-query' : 'query';
 		}
 
 		if (this.isCopilot) {
@@ -195,6 +195,10 @@ export class CategoryTreeNode extends TreeNode implements vscode.TreeItem {
 		}
 	}
 
+	get repo(): RemoteInfo | undefined {
+		return this._repo;
+	}
+
 	private _getDescription(): string | undefined {
 		if (!this.isCopilot || !this._repo) {
 			return undefined;
@@ -328,6 +332,15 @@ export class CategoryTreeNode extends TreeNode implements vscode.TreeItem {
 			this._repo = await extractRepoFromQuery(this.folderRepoManager, this._categoryQuery);
 		}
 		this.resourceUri = toQueryUri({ remote: this._repo, isCopilot: this.isCopilot });
+
+		// Update contextValue based on current notification state
+		if (this._categoryQuery) {
+			const hasNotifications = this.isCopilot && this._repo && this._copilotManager.getNotificationsCount(this._repo.owner, this._repo.repositoryName) > 0;
+			this.contextValue = this.isCopilot ?
+				(hasNotifications ? 'copilot-query-with-notifications' : 'copilot-query') :
+				'query';
+		}
+
 		return this;
 	}
 }