now-cli/bin/now-upgrade.js

#!/usr/bin/env node

// Packages
const chalk = require('chalk')
const minimist = require('minimist')
const ms = require('ms')
const stripAnsi = require('strip-ansi')

// Ours
const login = require('../lib/login')
const cfg = require('../lib/cfg')
const NowPlans = require('../lib/plans')
const indent = require('../lib/indent')
const listInput = require('../lib/utils/input/list')
const code = require('../lib/utils/output/code')
const error = require('../lib/utils/output/error')
const success = require('../lib/utils/output/success')
const cmd = require('../lib/utils/output/cmd')

const argv = minimist(process.argv.slice(2), {
  string: ['config', 'token'],
  boolean: ['help', 'debug'],
  alias: {
    help: 'h',
    config: 'c',
    debug: 'd',
    token: 't'
  }
})

const help = () => {
  console.log(`
  ${chalk.bold('𝚫 now upgrade')} [plan]

  ${chalk.dim('Options:')}

    -h, --help              Output usage information
    -c ${chalk.bold.underline('FILE')}, --config=${chalk.bold.underline('FILE')}  Config file
    -d, --debug             Debug mode [off]
    -t ${chalk.bold.underline('TOKEN')}, --token=${chalk.bold.underline('TOKEN')} Login token

  ${chalk.dim('Examples:')}

  ${chalk.gray('–')} List available plans and pick one interactively

      ${chalk.cyan('$ now upgrade')}

      ${chalk.yellow('NOTE:')} ${chalk.gray('Make sure you have a payment method, or add one:')}

      ${chalk.cyan(`$ now billing add`)}

  ${chalk.gray('–')} Pick a specific plan (premium):

      ${chalk.cyan(`$ now upgrade premium`)}
  `)
}

// options
const debug = argv.debug
const apiUrl = argv.url || 'https://api.zeit.co'

if (argv.config) {
  cfg.setConfigFile(argv.config)
}

const exit = code => {
  // we give stdout some time to flush out
  // because there's a node bug where
  // stdout writes are asynchronous
  // https://github.com/nodejs/node/issues/6456
  setTimeout(() => process.exit(code || 0), 100)
}

if (argv.help) {
  help()
  exit(0)
} else {
  const config = cfg.read()

  Promise.resolve(argv.token || config.token || login(apiUrl))
  .then(async token => {
    try {
      await run(token)
    } catch (err) {
      if (err.userError) {
        error(err.message)
      } else {
        error(`Unknown error: ${err.stack}`)
      }
      exit(1)
    }
  })
  .catch(e => {
    error(`Authentication error – ${e.message}`)
    exit(1)
  })
}

function buildInquirerChoices(current, until) {
  if (until) {
    until = until.split(' ')
    until = ' for ' + chalk.bold(until[0]) + ' more ' + until[1]
  } else {
    until = ''
  }
  const ossTitle = current === 'oss' ?
    `oss FREE ${' '.repeat(28)} (current)` :
    'oss FREE'
  const premiumTitle = current === 'premium' ?
    `premium $15/mo ${' '.repeat(24 - stripAnsi(until).length)} (current${until})` :
    'premium $15/mo'
  return [
    {
      name: [
        ossTitle,
        indent('✓ All code is public and open-source', 2),
        indent('✓ 20 deploys per month | 1GB monthly bandwidth', 2),
        indent('✓ 1GB FREE storage | 1MB size limit per file', 2)
      ].join('\n'),
      value: 'oss',
      short: 'oss FREE'
    },
    {
      name: [
        premiumTitle,
        indent('✓ All code is private and secure', 2),
        indent('✓ 1000 deploys per month | 50GB monthly bandwidth', 2),
        indent('✓ 100GB storage | No filesize limit', 2)
      ].join('\n'),
      value: 'premium',
      short: 'premium $15/mo'
    }
  ]
}

async function run(token) {
  const args = argv._
  if (args.length > 1) {
    error('Invalid number of arguments')
    return exit(1)
  }

  const start = new Date()
  const plans = new NowPlans(apiUrl, token, {debug})

  let planId = args[0]

  if (![undefined, 'oss', 'premium'].includes(planId)) {
    error(`Invalid plan name – should be ${code('oss')} or ${code('premium')}`)
    return exit(1)
  }

  const currentPlan = await plans.getCurrent()

  if (planId === undefined) {
    const elapsed = ms(new Date() - start)

    let message = `To manage this from the web UI, head to https://zeit.co/account\n`
    message += `> Selecting a plan for your account ${chalk.gray(`[${elapsed}]`)}`
    const choices = buildInquirerChoices(currentPlan.id, currentPlan.until)

    planId = await listInput({
      message,
      choices,
      separator: true,
      abort: 'end'
    })
  }

  if (planId === undefined || (planId === currentPlan.id && currentPlan.until === undefined)) {
    return console.log('No changes made')
  }

  let newPlan

  try {
    newPlan = await plans.set(planId)
  } catch (err) {
    let errorBody
    if (err.res && err.res.status === 400) {
      errorBody = err.res.json()
    } else {
      const message = 'A network error has occurred. Please retry.'
      errorBody = {message}
    }

    const _err = (await errorBody).error
    const {code, message} = _err

    if (code === 'customer_not_found' || code === 'source_not_found') {
      error(`You have no payment methods available. Run ${cmd('now billing add')} to add one`)
    } else {
      error(`An unknow error occured. Please try again later ${message}`)
    }
    plans.close()
    return
  }

  if (currentPlan.until && newPlan.id === 'premium') {
    success(`The cancelation has been undone. You're back on the ${chalk.bold('Premium plan')}`)
  } else if (newPlan.until) {
    success(`Your plan will be switched to OSS in ${chalk.bold(newPlan.until)}. Your card will not be charged again`)
  } else {
    success(`You're now on the ${chalk.bold('Premium plan')}`)
  }

  plans.close()
}