badgen.net/libs/github.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 github api v3 (rest)
export function restGithub<T = any>(path: string, preview = 'hellcat') {
  const headers = {
    authorization: `token ${pickGithubToken()}`,
    accept: `application/vnd.github.${preview}-preview+json`
  }
  const prefixUrl = 'https://api.github.com/'
  return got.get(path, { prefixUrl, headers }).json<T>()
}

// request github api v4 (graphql)
export function queryGithub<T = any>(query) {
  const headers = {
    authorization: `token ${pickGithubToken()}`,
    accept: 'application/vnd.github.hawkgirl-preview+json'
  }
  const json = { query }
  return got.post('https://api.github.com/graphql', { json, headers }).json<T>()
}

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