heroku-cli-util/lib/got.js

Set of helpful CLI utilities

'use strict'

let hush = require('./console').hush

function findProxy (urlParsed) {
  let httpProxy = process.env.HTTP_PROXY || process.env.http_proxy
  let httpsProxy = process.env.HTTPS_PROXY || process.env.https_proxy

  if (urlParsed.protocol === 'https:') {
    return httpsProxy || httpProxy
  } else {
    return httpProxy
  }
}

function findTunnel (urlParsed) {
  let tunnel = require('tunnel-agent')

  if (urlParsed.protocol === 'https:') {
    return tunnel.httpsOverHttp
  } else {
    return tunnel.httpOverHttp
  }
}

function agent (urlParsed, proxyParsed, certs) {
  let tunnelMethod = findTunnel(urlParsed)
  let opts = {
    proxy: {
      host: proxyParsed.hostname,
      port: proxyParsed.port || '8080'
    }
  }

  if (proxyParsed.auth) {
    opts.proxy.proxyAuth = proxyParsed.auth
  }

  if (certs.length > 0) {
    opts.ca = certs
  }

  let tunnelAgent = tunnelMethod(opts)
  if (urlParsed.protocol === 'https:') {
    tunnelAgent.defaultPort = 443
  }
  return tunnelAgent
}

function sslCertFile () {
  return process.env.SSL_CERT_FILE ? [process.env.SSL_CERT_FILE] : []
}

function sslCertDir () {
  let certDir = process.env.SSL_CERT_DIR
  if (certDir) {
    const fs = require('fs')
    const path = require('path')
    return fs.readdirSync(certDir).map(f => path.join(certDir, f))
  } else {
    return []
  }
}

function getCerts () {
  let filenames = sslCertFile().concat(sslCertDir())

  if (filenames.length > 0) {
    hush('Adding the following trusted certificate authorities')
  }

  return filenames.map(function (filename) {
    const fs = require('fs')
    hush('  ' + filename)
    return fs.readFileSync(filename)
  })
}

function addToOpts (url, opts) {
  const urlLib = require('url')

  let urlParsed = urlLib.parse(url)
  let proxy = findProxy(urlParsed)

  let certs = getCerts()

  if (proxy) {
    let proxyParsed = urlLib.parse(proxy)
    opts = Object.assign({}, opts, {agent: agent(urlParsed, proxyParsed, certs)})
  }

  if (certs.length > 0) {
    opts = Object.assign({}, opts, {ca: certs})
  }

  return opts
}

let loadErrors

function got (url, opts) {
  const gotDelegate = require('got')
  loadErrors()
  return gotDelegate(url, addToOpts(url, opts))
}

got.stream = function (url, opts) {
  const gotDelegate = require('got')
  loadErrors()
  return gotDelegate.stream(url, addToOpts(url, opts))
}

const helpers = [
  'get',
  'post',
  'put',
  'patch',
  'head',
  'delete'
]

helpers.forEach(el => {
  got[el] = (url, opts) => got(url, Object.assign({}, opts, {method: el}))
})

helpers.forEach(el => {
  got.stream[el] = function (url, opts) {
    return got.stream(url, Object.assign({}, opts, {method: el}))
  }
})

loadErrors = () => {
  const gotDelegate = require('got')

  const errors = [
    'CacheError',
    'CancelError',
    'GotError',
    'UnsupportedProtocolError',
    'HTTPError',
    'MaxRedirectsError',
    'ParseError',
    'ReadError',
    'RequestError'
  ]

  errors.forEach(el => {
    got[el] = gotDelegate[el]
  })
}

module.exports = got