error-request.ts

import {
  type RequestOptionsWithRequest,
  type RetryPlugin,
  type RetryState,
} from "./types.js";
import { RequestError } from "@octokit/request-error";

export function isRequestError(error: any): error is RequestError {
  return error.request !== undefined;
}

export function defaultShouldRetry(
  state: RetryState,
  error: RequestError | Error,
): boolean {
  if (!isRequestError(error) || !error?.request.request) {
    // address https://github.com/octokit/plugin-retry.js/issues/8
    throw error;
  }

  return error.status >= 400 && !state.doNotRetry.includes(error.status);
}

export async function errorRequest(
  state: RetryState,
  octokit: RetryPlugin,
  error: RequestError | Error,
  options: RequestOptionsWithRequest,
): Promise<any> {
  if (state.shouldRetry(state, error)) {
    const retries: number =
      options.request?.retries != null
        ? options.request.retries
        : state.retries;
    const retryAfter = Math.pow((options.request?.retryCount || 0) + 1, 2);
    throw new RequestError(
      error.message,
      isRequestError(error) ? error.status : 500,
      {
        request: octokit.retry.retryRequest(options, retries, retryAfter),
      },
    );
  }

  // Maybe eventually there will be more cases here

  throw error;
}