Move use of listWorkflowRunsForRepo into GitHubRepository and use restPaginate (#7155)

Fixes #7154

Author: alexr00

src/github/copilotApi.ts

@@ -8,10 +8,8 @@ import JSZip from 'jszip';
 import * as vscode from 'vscode';
 import { AuthProvider } from '../common/authentication';
 import Logger from '../common/logger';
-import { OctokitCommon } from './common';
 import { CredentialStore } from './credentials';
 import { LoggingOctokit } from './loggingOctokit';
-import { PullRequestModel } from './pullRequestModel';
 import { hasEnterpriseUri } from './utils';
 
 const LEARN_MORE_URL = 'https://docs.github.com/en/copilot/how-tos/agents/copilot-coding-agent';
@@ -109,37 +107,6 @@ export class CopilotApi {
 		}
 	}
 
-	public async getWorkflowRunsFromAction(pullRequest: PullRequestModel): Promise<OctokitCommon.ListWorkflowRunsForRepo> {
-		const createdDate = new Date(pullRequest.createdAt);
-		const created = `>=${createdDate.getFullYear()}-${String(createdDate.getMonth() + 1).padStart(2, '0')}-${String(createdDate.getDate()).padStart(2, '0')}`;
-		const allRuns: any[] = [];
-		let page = 1;
-		let hasMore = true;
-		const per_page = 100;
-		while (hasMore) {
-			const runs = await this.octokit.api.actions.listWorkflowRunsForRepo({
-				owner: pullRequest.remote.owner,
-				repo: pullRequest.remote.repositoryName,
-				event: 'dynamic',
-				created,
-				per_page,
-				page
-			});
-			if (runs.status !== 200) {
-				throw new Error(`Failed to fetch workflow runs: ${runs.status}`);
-			}
-			if (Array.isArray(runs.data.workflow_runs)) {
-				allRuns.push(...runs.data.workflow_runs);
-				hasMore = runs.data.total_count > allRuns.length;
-				page++;
-			} else {
-				hasMore = false;
-			}
-		}
-		// Return only the workflow_runs array for compatibility
-		return allRuns as unknown as OctokitCommon.ListWorkflowRunsForRepo;
-	}
-
 	public async getLogsFromZipUrl(logsUrl: string): Promise<string[]> {
 		const logsZip = await fetch(logsUrl, {
 			headers: {

src/github/copilotRemoteAgent.ts

@@ -416,7 +416,7 @@ export class CopilotRemoteAgentManager extends Disposable {
 		if (!capi) {
 			return;
 		}
-		const runs = await capi.getWorkflowRunsFromAction(pullRequest);
+		const runs = await pullRequest.githubRepository.getWorkflowRunsFromAction(pullRequest.createdAt);
 		const padawanRuns = runs
 			.filter(run => run.path && run.path.startsWith('dynamic/copilot-swe-agent'))
 			.filter(run => run.pull_requests?.some(pr => pr.id === pullRequest.id));

src/github/githubRepository.ts

@@ -850,6 +850,20 @@ export class GitHubRepository extends Disposable {
 		}
 	}
 
+	public async getWorkflowRunsFromAction(fromDate: string): Promise<OctokitCommon.ListWorkflowRunsForRepo> {
+		const { octokit, remote } = await this.ensure();
+		const createdDate = new Date(fromDate);
+		const created = `>=${createdDate.getFullYear()}-${String(createdDate.getMonth() + 1).padStart(2, '0')}-${String(createdDate.getDate()).padStart(2, '0')}`;
+		const allRuns = await restPaginate<typeof octokit.api.actions.listWorkflowRunsForRepo, OctokitCommon.ListWorkflowRunsForRepo[0]>(octokit.api.actions.listWorkflowRunsForRepo, {
+			owner: remote.owner,
+			repo: remote.repositoryName,
+			event: 'dynamic',
+			created
+		});
+
+		return allRuns;
+	}
+
 	async fork(): Promise<string | undefined> {
 		try {
 			Logger.debug(`Fork repository`, this.id);