import * as core from '@actions/core'
import {OctokitOptions} from '@octokit/core/dist-types/types'
import {ThrottlingOptions} from '@octokit/plugin-throttling'
import {RequestRequestOptions} from '@octokit/types'

export type RetryOptions = {
  doNotRetry?: number[]
  enabled?: boolean
}

export function getRetryOptions(
  retries: number,
  exemptStatusCodes: number[],
  defaultOptions: OctokitOptions
): [
  RetryOptions,
  RequestRequestOptions | undefined,
  ThrottlingOptions | undefined
] {
  if (retries <= 0) {
    return [{enabled: false}, defaultOptions.request, undefined]
  }

  const retryOptions: RetryOptions = {
    enabled: true
  }

  if (exemptStatusCodes.length > 0) {
    retryOptions.doNotRetry = exemptStatusCodes
  }

  // The GitHub type has some defaults for `options.request`
  // see: https://github.com/actions/toolkit/blob/4fbc5c941a57249b19562015edbd72add14be93d/packages/github/src/utils.ts#L15
  // We pass these in here so they are not overidden.
  const requestOptions: RequestRequestOptions = {
    ...defaultOptions.request,
    retries
  }

  const throttleOptions: ThrottlingOptions = {
    onRateLimit: (retryAfter, options, octokit, retryCount) => {
      core.debug(
        `Request quota exhausted for request ${options.method} ${options.url}`
      )
      return retryCount < retries
    },
    onSecondaryRateLimit: (retryAfter, options, octokit, retryCount) => {
      core.debug(
        `Secondary quota detected for request ${options.method} ${options.url}`
      )
      return retryCount < retries
    }
  }

  core.debug(
    `GitHub client configured with: (retries: ${
      requestOptions.retries
    }, retry-exempt-status-code: ${
      retryOptions.doNotRetry ?? 'octokit default: [400, 401, 403, 404, 422]'
    })`
  )

  return [retryOptions, requestOptions, throttleOptions]
}

export function parseNumberArray(listString: string): number[] {
  if (!listString) {
    return []
  }

  const split = listString.trim().split(',')
  return split.map(x => parseInt(x))
}