import ky from 'ky'

const rand = <T>(arr: T[]): T => arr[Math.floor(Math.random() * arr.length)]

// request gitlab api v4 (graphql)
export function queryGitlab<T = any>(query) {
  const token = pickGitlabToken()
  const headers = {
    authorization: token ? `Bearer ${token}` : undefined,
  }
  const json = { query }
  const endpoint =
    process.env.GITLAB_API_GRAPHQL || 'https://gitlab.com/api/graphql'
  return ky.post(endpoint, { json, headers }).json<T>()
}

export function restGitlab<T = any>(path: string, fullResponse = false) {
  const token = pickGitlabToken()
  const headers = {
    authorization: token ? `Bearer ${token}` : undefined,
  }
  const prefixUrl = process.env.GITLAB_API || 'https://gitlab.com/api/v4'
  return fullResponse ? ky.get(path, { prefixUrl, headers }) : ky.get(path, { prefixUrl, headers }).json<T>()
}

function pickGitlabToken() {
  const { GITLAB_TOKENS } = process.env
  if (!GITLAB_TOKENS) {
    return null
  }
  const tokens = GITLAB_TOKENS.split(',').map(segment => segment.trim())
  return rand(tokens)
}