badgen.net/libs/gitlab.ts

import got from './got'
import { BadgenError } from './create-badgen-handler'

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 headers = {
    authorization: `Bearer ${pickGitlabToken()}`,
  }
  const json = { query }
  const endpoint =
    process.env.GITLAB_API_GRAPHQL || 'https://gitlab.com/api/graphql'
  return got.post(endpoint, { json, headers }).json<T>()
}

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

function pickGitlabToken() {
  const { GITLAB_TOKENS } = process.env
  if (!GITLAB_TOKENS) {
    throw new BadgenError({ status: 'token required' })
  }
  const tokens = GITLAB_TOKENS.split(',').map(segment => segment.trim())
  return rand(tokens)
}