Retrieves a pull request from Azure

(#3977, #3996)

Author: sergeibbb

Diff for: src/plus/integrations/providers/azure/azure.ts

@@ -22,8 +22,14 @@ import { Logger } from '../../../../system/logger';
 import type { LogScope } from '../../../../system/logger.scope';
 import { getLogScope } from '../../../../system/logger.scope';
 import { maybeStopWatch } from '../../../../system/stopwatch';
-import type { AzureWorkItemState, AzureWorkItemStateCategory, WorkItem } from './models';
-import { azureWorkItemsStateCategoryToState, isClosedAzureWorkItemStateCategory } from './models';
+import type { AzurePullRequest, AzureWorkItemState, AzureWorkItemStateCategory, WorkItem } from './models';
+import {
+	azurePullRequestStatusToState,
+	azureWorkItemsStateCategoryToState,
+	getPullRequestUrl,
+	isClosedAzurePullRequestStatus,
+	isClosedAzureWorkItemStateCategory,
+} from './models';
 
 export class AzureDevOpsApi implements Disposable {
 	private readonly _disposable: Disposable;
@@ -71,7 +77,7 @@ export class AzureDevOpsApi implements Disposable {
 		},
 	): Promise<IssueOrPullRequest | undefined> {
 		const scope = getLogScope();
-		const [projectName] = repo.split('/');
+		const [projectName, _, repoName] = repo.split('/');
 
 		try {
 			// Try to get the Work item (wit) first with specific fields
@@ -112,6 +118,39 @@ export class AzureDevOpsApi implements Disposable {
 					url: issueResult._links.html.href,
 				};
 			}
+		} catch (ex) {
+			if (ex.original?.status !== 404) {
+				Logger.error(ex, scope);
+				return undefined;
+			}
+		}
+
+		try {
+			const prResult = await this.request<AzurePullRequest>(
+				provider,
+				token,
+				options?.baseUrl,
+				`${owner}/${projectName}/_apis/git/repositories/${repoName}/pullRequests/${id}`,
+				{
+					method: 'GET',
+				},
+				scope,
+			);
+
+			if (prResult != null) {
+				return {
+					id: prResult.pullRequestId.toString(),
+					type: 'pullrequest',
+					nodeId: prResult.pullRequestId.toString(), // prResult.artifactId maybe?
+					provider: provider,
+					createdDate: new Date(prResult.creationDate),
+					updatedDate: new Date(prResult.creationDate),
+					state: azurePullRequestStatusToState(prResult.status),
+					closed: isClosedAzurePullRequestStatus(prResult.status),
+					title: prResult.title,
+					url: getPullRequestUrl(options.baseUrl, owner, projectName, repoName, prResult.pullRequestId),
+				};
+			}
 
 			return undefined;
 		} catch (ex) {

Diff for: src/plus/integrations/providers/azure/models.ts

@@ -75,3 +75,70 @@ export interface AzureWorkItemState {
 	color: string;
 	category: AzureWorkItemStateCategory;
 }
+
+export type AzurePullRequestStatus = 'abandoned' | 'active' | 'completed' | 'notSet';
+export function azurePullRequestStatusToState(status: AzurePullRequestStatus): IssueOrPullRequestState {
+	switch (status) {
+		case 'abandoned':
+			return 'closed';
+		case 'completed':
+			return 'merged';
+		case 'active':
+		case 'notSet':
+		default:
+			return 'opened';
+	}
+}
+export function isClosedAzurePullRequestStatus(status: AzurePullRequestStatus): boolean {
+	return azurePullRequestStatusToState(status) !== 'opened';
+}
+
+export function getPullRequestUrl(
+	baseUrl: string,
+	owner: string,
+	projectName: string,
+	repoName: string,
+	pullRequestId: number,
+): string {
+	return `${baseUrl}/${owner}/${projectName}/_git/${repoName}/pullrequest/${pullRequestId}`;
+}
+
+export interface AzurePullRequest {
+	repository: unknown;
+	pullRequestId: number;
+	codeReviewId: number;
+	status: AzurePullRequestStatus;
+	createdBy: AzureUser;
+	creationDate: string;
+	closedDate: string;
+	title: string;
+	description: string;
+	sourceRefName: string;
+	targetRefName: string;
+	isDraft: boolean;
+	mergeId: string;
+	lastMergeSourceCommit: {
+		commitId: string;
+		url: string;
+	};
+	lastMergeTargetCommit: {
+		commitId: string;
+		url: string;
+	};
+	reviewers: unknown[];
+	url: string;
+	_links: {
+		self: AzureLink;
+		repository: AzureLink;
+		workItems: AzureLink;
+		sourceBranch: AzureLink;
+		targetBranch: AzureLink;
+		statuses: AzureLink;
+		sourceCommit: AzureLink;
+		targetCommit: AzureLink;
+		createdBy: AzureLink;
+		iterations: AzureLink;
+	};
+	supportsIterations: boolean;
+	artifactId: string;
+}