githubPullRequestInfoProvider.ts

import {GetPullRequestService} from './Api/getPullRequestService';
import {mergeableStates, PullRequestInfo} from '../pullrequestinfo';
import {debug} from '@actions/core';
import {promiseRetry} from '../Util/promiseRetry';

export class GithubPullRequestInfoProvider {
    constructor(private getPullRequestService: GetPullRequestService) {}

    public pullRequestInfoFor(
        ownerName: string,
        repoName: string,
        pullRequestNumber: number,
    ): Promise<PullRequestInfo> {
        return promiseRetry<PullRequestInfo>(async (attemptNumber): Promise<PullRequestInfo> => {
            try {
                const {draft, rebaseable, mergeableState, labels} = await this.getPullRequestService.getPullRequest(
                    ownerName,
                    repoName,
                    pullRequestNumber,
                );

                if (attemptNumber < 10 && !draft) {
                    if (mergeableState === 'unknown' || !mergeableStates.includes(mergeableState)) {
                        debug(`mergeableState for pull request #${pullRequestNumber} is 'unknown', retrying.`);
                        throw Error("mergeableState is 'unknown'");
                    }
                    if (rebaseable === null) {
                        debug(
                            `rebaseable for pull request #${pullRequestNumber} is null (not yet computed), retrying.`,
                        );
                        throw Error('rebaseable is null');
                    }
                }

                debug(`rebaseable value for pull request #${pullRequestNumber}: ${String(rebaseable)}`);
                debug(`mergeableState for pull request #${pullRequestNumber}: ${mergeableState}`);

                return {
                    ownerName: ownerName,
                    repoName: repoName,
                    number: pullRequestNumber,
                    draft: draft,
                    rebaseable: rebaseable ?? false,
                    mergeableState: mergeableState,
                    labels: labels,
                };
            } catch (error) {
                debug(
                    `Fetching mergeableState for pull request #${pullRequestNumber} failed: "${String(
                        error,
                    )}", retrying.`,
                );
                throw error;
            }
        });
    }
}