@heroku-cli/plugin-pg-v5/commands/backups/download.js

Heroku CLI plugin to manage Postgres.

'use strict'

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

function defaultFilename() {
  const fs = require('fs')
  let f = 'latest.dump'
  if (!fs.existsSync(f)) return f
  let i = 1
  do f = `latest.dump.${i++}`
  while (fs.existsSync(f))
  return f
}

async function run(context, heroku) {
  const host = require('../../lib/host')()
  const pgbackups = require('../../lib/pgbackups')(context, heroku)
  const download = require('../../lib/download')
  const {sortBy} = require('lodash')

  const {app, args, flags} = context
  const output = flags.output || defaultFilename()

  let num
  let info

  await cli.action(`Getting backup from ${cli.color.app(app)}`, (async function () {
    if (args.backup_id) {
      num = await pgbackups.transfer.num(args.backup_id)
      if (!num) throw new Error(`Invalid Backup: ${args.backup_id}`)
    } else {
      let transfers = await heroku.get(`/client/v11/apps/${app}/transfers`, {host})
      let lastBackup = sortBy(transfers.filter(t => t.succeeded && t.to_type === 'gof3r'), 'created_at').pop()
      if (!lastBackup) throw new Error(`No backups on ${cli.color.app(app)}. Capture one with ${cli.color.cmd('heroku pg:backups:capture')}`)
      num = lastBackup.num
    }

    cli.action.status(`fetching url of #${num}`)

    info = await heroku.post(`/client/v11/apps/${app}/transfers/${num}/actions/public-url`, {host})
    cli.action.done(`done, #${num}`)
  })())

  await download(info.url, output, {progress: true})
}

module.exports = {
  topic: 'pg',
  command: 'backups:download',
  description: 'downloads database backup',
  needsApp: true,
  needsAuth: true,
  args: [{name: 'backup_id', optional: true}],
  flags: [
    {name: 'output', char: 'o', description: 'location to download to. Defaults to latest.dump', hasValue: true},
  ],
  run: cli.command({preauth: true}, run),
}