heroku/heroku-pg

'use strict'

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

function * run (context, heroku) {
  const fetcher = require('../../lib/fetcher')(heroku)
  const host = require('../../lib/host')
  const util = require('../../lib/util')
  const {app, args, flags} = context
  const db = yield fetcher.addon(app, args.database)

  if (util.starterPlan(db)) throw new Error('pg:maintenance is only available for production databases')
  yield cli.action(`Starting maintenance for ${cli.color.addon(db.name)}`, co(function * () {
    if (!flags.force) {
      let appInfo = yield heroku.get(`/apps/${app}`)
      if (!appInfo.maintenance) throw new Error('Application must be in maintenance mode or run with --force')
    }
    let response = yield heroku.post(`/client/v11/databases/${db.id}/maintenance`, {host: host(db)})
    cli.action.done(response.message || 'done')
  }))
}

module.exports = {
  topic: 'pg',
  command: 'maintenance:run',
  description: 'start maintenance',
  needsApp: true,
  needsAuth: true,
  args: [{name: 'database', optional: true}],
  flags: [{name: 'force', char: 'f'}],
  run: cli.command({preauth: true}, co.wrap(run))
}