Use new copilot status icons (#7121)

Fixes microsoft/vscode-copilot#18537

Author: alexr00

src/github/githubRepository.ts

@@ -146,6 +146,13 @@ export interface GraphQLError {
 	message?: string;
 }
 
+export enum CopilotWorkingStatus {
+	NotCopilotIssue = 'NotCopilotIssue',
+	InProgress = 'InProgress',
+	Error = 'Error',
+	Done = 'Done',
+}
+
 export class GitHubRepository extends Disposable {
 	static ID = 'GitHubRepository';
 	protected _initialized: boolean = false;
@@ -1527,6 +1534,21 @@ export class GitHubRepository extends Disposable {
 		}
 	}
 
+	async copilotWorkingStatus(issueModel: IssueModel): Promise<CopilotWorkingStatus | undefined> {
+		const copilotEvents = await this.getCopilotTimelineEvents(issueModel);
+		if (copilotEvents.length > 0) {
+			const lastEvent = copilotEvents[copilotEvents.length - 1];
+			if (lastEvent.event === Common.EventType.CopilotFinished) {
+				return CopilotWorkingStatus.Done;
+			} else if (lastEvent.event === Common.EventType.CopilotStarted) {
+				return CopilotWorkingStatus.InProgress;
+			} else if (lastEvent.event === Common.EventType.CopilotFinishedError) {
+				return CopilotWorkingStatus.Error;
+			}
+		}
+		return CopilotWorkingStatus.NotCopilotIssue;
+	}
+
 	/**
 	 * Get the status checks of the pull request, those for the last commit.
 	 *

src/view/treeNodes/pullRequestNode.ts

@@ -11,6 +11,7 @@ import Logger from '../../common/logger';
 import { FILE_LIST_LAYOUT, PR_SETTINGS_NAMESPACE, SHOW_PULL_REQUEST_NUMBER_IN_TREE } from '../../common/settingKeys';
 import { createPRNodeUri, DataUri, fromPRUri, Schemes } from '../../common/uri';
 import { FolderRepositoryManager } from '../../github/folderRepositoryManager';
+import { CopilotWorkingStatus } from '../../github/githubRepository';
 import { NotificationProvider } from '../../github/notifications';
 import { IResolvedPullRequestModel, PullRequestModel } from '../../github/pullRequestModel';
 import { InMemFileChangeModel, RemoteFileChangeModel } from '../fileChangeModel';
@@ -259,6 +260,23 @@ export class PRNode extends TreeNode implements vscode.CommentingRangeProvider2
 		});
 	}
 
+	private async _getIcon(): Promise<vscode.Uri | vscode.ThemeIcon> {
+		const copilotWorkingStatus = await this.pullRequestModel.githubRepository.copilotWorkingStatus(this.pullRequestModel);
+		switch (copilotWorkingStatus) {
+			case CopilotWorkingStatus.InProgress:
+				return new vscode.ThemeIcon('copilot-in-progress');
+			case CopilotWorkingStatus.Done:
+				return new vscode.ThemeIcon('copilot-success');
+			case CopilotWorkingStatus.Error:
+				return new vscode.ThemeIcon('copilot-error');
+			case CopilotWorkingStatus.NotCopilotIssue:
+			default:
+				return (await DataUri.avatarCirclesAsImageDataUris(this._folderReposManager.context, [this.pullRequestModel.author], 16, 16))[0]
+					?? new vscode.ThemeIcon('github');
+
+		}
+	}
+
 	async getTreeItem(): Promise<vscode.TreeItem> {
 		const currentBranchIsForThisPR = this.pullRequestModel.equals(this._folderReposManager.activePullRequest);
 
@@ -298,8 +316,7 @@ export class PRNode extends TreeNode implements vscode.CommentingRangeProvider2
 				(currentBranchIsForThisPR ? ':active' : ':nonactive') +
 				(hasNotification ? ':notification' : '') +
 				(((this.pullRequestModel.item.isRemoteHeadDeleted && !this._isLocal) || !this._folderReposManager.isPullRequestAssociatedWithOpenRepository(this.pullRequestModel)) ? '' : ':hasHeadRef'),
-			iconPath: (await DataUri.avatarCirclesAsImageDataUris(this._folderReposManager.context, [this.pullRequestModel.author], 16, 16))[0]
-				?? new vscode.ThemeIcon('github'),
+			iconPath: await this._getIcon(),
 			accessibilityInformation: {
 				label: `${isDraft ? 'Draft ' : ''}Pull request number ${formattedPRNumber}: ${title} by ${login}`
 			},