heroku-cli-addons/lib/util.js

'use strict'

const cli = require('heroku-cli-util')

let styles = {
  app: 'cyan',
  attachment: 'green',
  addon: 'magenta'
}

module.exports = {
  // style given text or return a function that styles text according to provided style
  style: function style (s, t) {
    if (!t) return (text) => style(s, text)
    return cli.color[styles[s] || s](t)
  },

  table: function (data, options) {
    const merge = require('lodash.merge')

    return cli.table(data, merge(options, {
      printLine: cli.log
    }))
  },

  formatPrice: function (price) {
    const printf = require('printf')

    if (!price) return
    if (price.cents === 0) return 'free'

    let fmt = price.cents % 100 === 0 ? '$%.0f/%s' : '$%.02f/%s'
    return printf(fmt, price.cents / 100, price.unit)
  },

  trapConfirmationRequired: function * (context, fn) {
    return yield fn(context.flags.confirm)
    .catch((err) => {
      if (!err.body || err.body.id !== 'confirmation_required') throw err
      return cli.confirmApp(context.app, context.flags.confirm, err.body.message)
        .then(() => fn(context.app))
    })
  },

  formatState: function (state) {
    switch (state) {
      case 'provisioned':
        state = 'created'
        break
      case 'provisioning':
        state = 'creating'
        break
      case 'deprovisioned':
        state = 'errored'
        break
      default:
        state = ''
    }
    return state
  }
}