Retrieves Bitbucket PRs where a user is author or reviewer

(#4046)

Author: sergeibbb

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

@@ -12,6 +12,8 @@ import type {
 	SearchedPullRequest,
 } from '../../../git/models/pullRequest';
 import type { RepositoryMetadata } from '../../../git/models/repositoryMetadata';
+import { groupBy } from '../../../system/iterable';
+import { getSettledValue } from '../../../system/promise';
 import type { IntegrationAuthenticationProviderDescriptor } from '../authentication/integrationAuthenticationProvider';
 import type { ProviderAuthenticationSession } from '../authentication/models';
 import type { ResourceDescriptor } from '../integration';
@@ -263,7 +265,6 @@ export class BitbucketIntegration extends HostingIntegration<
 		session: ProviderAuthenticationSession,
 		requestedRepositories?: BitbucketRepositoryDescriptor[],
 	): Promise<SearchedPullRequest[] | undefined> {
-		const api = await this.getProvidersApi();
 		if (requestedRepositories != null) {
 			// TODO: implement repos version
 			return undefined;
@@ -278,17 +279,30 @@ export class BitbucketIntegration extends HostingIntegration<
 		const repos = await this.getProviderProjectsForResources(session, workspaces);
 		if (repos == null || repos.length === 0) return undefined;
 
-		const prs = await api.getPullRequestsForRepos(
-			HostingIntegrationId.Bitbucket,
-			repos.map(repo => ({ namespace: repo.owner, name: repo.name })),
-			{
-				accessToken: session.accessToken,
-			},
+		const api = await this.container.bitbucket;
+		if (!api) return undefined;
+		const prsResult = await Promise.allSettled(
+			repos.map(repo =>
+				api.getUsersPullRequestsForRepo(this, session.accessToken, user.id, repo.owner, repo.name, {
+					baseUrl: this.apiBaseUrl,
+				}),
+			),
 		);
-		return prs.values.map(pr => ({
-			pullRequest: this.fromBitbucketProviderPullRequest(pr),
+		const prs = prsResult
+			.map(r => getSettledValue(r))
+			.filter(r => r != null)
+			.flat();
+
+		const groups = groupBy(prs, pr => (pr.author.id === user.id ? 'authored' : 'other'));
+		const authoredPrs = groups.authored.map(pr => ({
+			pullRequest: pr,
+			reasons: ['authored'],
+		}));
+		const otherPrs = groups.other.map(pr => ({
+			pullRequest: pr,
 			reasons: [],
 		}));
+		return [...authoredPrs, ...otherPrs];
 	}
 
 	protected override async searchProviderMyIssues(
@@ -298,14 +312,6 @@ export class BitbucketIntegration extends HostingIntegration<
 		return Promise.resolve(undefined);
 	}
 
-	private fromBitbucketProviderPullRequest(
-		remotePullRequest: ProviderPullRequest,
-		//		repoDescriptors: BitbucketRemoteRepositoryDescriptor[],
-	): PullRequest {
-		remotePullRequest.graphQLId = remotePullRequest.id;
-		return fromProviderPullRequest(remotePullRequest, this);
-	}
-
 	protected override async providerOnConnect(): Promise<void> {
 		if (this._session == null) return;
 

Diff for: src/plus/integrations/providers/bitbucket/bitbucket-api.ts

@@ -223,6 +223,41 @@ export class BitbucketApi implements Disposable {
 		}
 	}
 
+	public async getUsersPullRequestsForRepo(
+		provider: Provider,
+		token: string,
+		userUuid: string,
+		owner: string,
+		repo: string,
+		options: {
+			baseUrl: string;
+		},
+	): Promise<PullRequest[] | undefined> {
+		const scope = getLogScope();
+
+		const query = encodeURIComponent(`reviewers.uuid="${userUuid}" OR author.uuid="${userUuid}"`);
+		const response = await this.request<{
+			values: BitbucketPullRequest[];
+			pagelen: number;
+			size: number;
+			page: number;
+		}>(
+			provider,
+			token,
+			options.baseUrl,
+			`repositories/${owner}/${repo}/pullrequests?q=${query}&state=OPEN&fields=%2Bvalues.reviewers,%2Bvalues.participants`,
+			{
+				method: 'GET',
+			},
+			scope,
+		);
+
+		if (!response?.values?.length) {
+			return undefined;
+		}
+		return response.values.map(pr => fromBitbucketPullRequest(pr, provider));
+	}
+
 	private async request<T>(
 		provider: Provider,
 		token: string,