PR view always expands and fetches "All Open" (#7157)

Fixes #7150

Author: alexr00

src/github/copilotRemoteAgent.ts

@@ -384,6 +384,7 @@ export class CopilotRemoteAgentManager extends Disposable {
 
 		try {
 			const { pull_request } = await capiClient.postRemoteAgentJob(owner, repo, payload);
+			this._onDidCreatePullRequest.fire(pull_request.number);
 			const webviewUri = await toOpenPullRequestWebviewUri({ owner, repo, pullRequestNumber: pull_request.number });
 			const prLlmString = `The remote agent has begun work. The user can track progress by visiting ${pull_request.html_url} or from the PR extension. Format this VS Code webview link so the user can click it to also track progress: ${webviewUri.toString()}`;
 			return {

src/view/prsTreeModel.ts

@@ -29,7 +29,7 @@ export class PrsTreeModel extends Disposable {
 	public readonly onDidChangePrStatus = this._onDidChangePrStatus.event;
 	private readonly _onDidChangeData: vscode.EventEmitter<FolderRepositoryManager | void> = this._register(new vscode.EventEmitter<FolderRepositoryManager | void>());
 	public readonly onDidChangeData = this._onDidChangeData.event;
-	private _expandedQueries: Set<string> = new Set();
+	private _expandedQueries: Set<string> | undefined;
 	private _hasLoaded: boolean = false;
 	private _onLoaded: vscode.EventEmitter<void> = this._register(new vscode.EventEmitter<void>());
 	public readonly onLoaded = this._onLoaded.event;
@@ -78,10 +78,16 @@ export class PrsTreeModel extends Disposable {
 			}
 		}));
 
-		this._expandedQueries = new Set(this._context.workspaceState.get(EXPANDED_QUERIES_STATE, [] as string[]));
+		const expandedQueries = this._context.workspaceState.get(EXPANDED_QUERIES_STATE, undefined);
+		if (expandedQueries) {
+			this._expandedQueries = new Set(expandedQueries);
+		}
 	}
 
 	public updateExpandedQueries(element: TreeNode, isExpanded: boolean) {
+		if (!this._expandedQueries) {
+			this._expandedQueries = new Set();
+		}
 		if ((element instanceof CategoryTreeNode) && element.id) {
 			if (isExpanded) {
 				this._expandedQueries.add(element.id);
@@ -92,8 +98,8 @@ export class PrsTreeModel extends Disposable {
 		}
 	}
 
-	get expandedQueries(): Set<string> {
-		if (this._reposManager.folderManagers.length > 3 && this._expandedQueries.size > 0) {
+	get expandedQueries(): Set<string> | undefined {
+		if (this._reposManager.folderManagers.length > 3 && this._expandedQueries && this._expandedQueries.size > 0) {
 			return new Set();
 		}
 		return this._expandedQueries;

src/view/treeNodes/categoryNode.ts

@@ -170,11 +170,11 @@ export class CategoryTreeNode extends TreeNode implements vscode.TreeItem {
 		if (hasCopilotChanges) {
 			this.collapsibleState = vscode.TreeItemCollapsibleState.Expanded;
 			this.resourceUri = COPILOT_QUERY;
-		} else if ((this._prsTreeModel.expandedQueries.size === 0) && (this.type === PRType.All)) {
+		} else if ((this._prsTreeModel.expandedQueries === undefined) && (this.type === PRType.All)) {
 			this.collapsibleState = vscode.TreeItemCollapsibleState.Expanded;
 		} else {
 			this.collapsibleState =
-				this._prsTreeModel.expandedQueries.has(this.id)
+				this._prsTreeModel.expandedQueries?.has(this.id)
 					? vscode.TreeItemCollapsibleState.Expanded
 					: vscode.TreeItemCollapsibleState.Collapsed;
 		}