heroku-certs/lib/domains.js

heroku ssl plugin

'use strict'

let _ = require('lodash')
let co = require('co')
let cli = require('heroku-cli-util')
let psl = require('psl')

function type (domain) {
  return psl.parse(domain.hostname).subdomain === null ? 'ALIAS/ANAME' : 'CNAME'
}

function * waitForDomains (context, heroku) {
  function someNull (domains) {
    return _.some(domains, (domain) => domain.kind === 'custom' && !domain.cname)
  }

  function apiRequest (context, heroku) {
    return heroku.request({
      path: `/apps/${context.app}/domains`
    })
  }

  let apiDomains = yield apiRequest(context, heroku)

  if (someNull(apiDomains)) {
    yield cli.action('Waiting for stable domains to be created', co(function * () {
      const wait = require('co-wait')

      let i = 0
      do {
        // trying 30 times was easier for me to test that setTimeout
        if (i >= 30) {
          throw new Error('Timed out while waiting for stable domains to be created')
        }

        yield wait(1000)
        apiDomains = yield apiRequest(context, heroku)

        i++
      } while (someNull(apiDomains))
    }))
  }

  return apiDomains
}

function printDomains (domainsTable, msg) {
  domainsTable = domainsTable.filter((domain) => domain.kind === 'custom')
  domainsTable = domainsTable.map(domain => Object.assign({}, domain, {type: type(domain)}))

  if (domainsTable.length === 0) {
    /* eslint-disable no-irregular-whitespace */
    cli.styledHeader(`${msg}  Add a custom domain to your app by running ${cli.color.app('heroku domains:add <yourdomain.com>')}`)
    /* eslint-enable no-irregular-whitespace */
  } else {
    cli.styledHeader(`${msg}  Update your application's DNS settings as follows`)

    let columns = [
        {label: 'Domain', key: 'hostname'},
        {label: 'Record Type', key: 'type'},
        {label: 'DNS Target', key: 'cname'}
    ]

    if (_.some(domainsTable, (domain) => domain.warning)) {
      columns.push({label: 'Warnings', key: 'warning'})
    }

    cli.table(domainsTable, {columns: columns})
  }
}

module.exports = {waitForDomains, printDomains}