1 | 'use strict'
|
2 |
|
3 | const co = require('co')
|
4 | const cli = require('heroku-cli-util')
|
5 |
|
6 | function defaultFilename () {
|
7 | const fs = require('fs')
|
8 | let f = 'latest.dump'
|
9 | if (!fs.existsSync(f)) return f
|
10 | let i = 1
|
11 | do f = `latest.dump.${i++}`
|
12 | while (fs.existsSync(f))
|
13 | return f
|
14 | }
|
15 |
|
16 | function * run (context, heroku) {
|
17 | const host = require('../../lib/host')()
|
18 | const pgbackups = require('../../lib/pgbackups')(context, heroku)
|
19 | const download = require('../../lib/download')
|
20 | const sortBy = require('lodash.sortby')
|
21 |
|
22 | const {app, args, flags} = context
|
23 | const output = flags.output || defaultFilename()
|
24 |
|
25 | let num, info
|
26 |
|
27 | yield cli.action(`Getting backup from ${cli.color.app(app)}`, co(function * () {
|
28 | if (args.backup_id) {
|
29 | num = yield pgbackups.transfer.num(args.backup_id)
|
30 | if (!num) throw new Error(`Invalid Backup: ${args.backup_id}`)
|
31 | } else {
|
32 | let transfers = yield heroku.get(`/client/v11/apps/${app}/transfers`, {host})
|
33 | let lastBackup = sortBy(transfers.filter(t => t.succeeded && t.to_type === 'gof3r'), 'created_at').pop()
|
34 | if (!lastBackup) throw new Error(`No backups on ${cli.color.app(app)}. Capture one with ${cli.color.cmd('heroku pg:backups:capture')}`)
|
35 | num = lastBackup.num
|
36 | }
|
37 | cli.action.status(`fetching url of #${num}`)
|
38 |
|
39 | info = yield heroku.post(`/client/v11/apps/${app}/transfers/${num}/actions/public-url`, {host})
|
40 | cli.action.done(`done, #${num}`)
|
41 | }))
|
42 |
|
43 | yield download(info.url, output, {progress: true})
|
44 | }
|
45 |
|
46 | module.exports = {
|
47 | topic: 'pg',
|
48 | command: 'backups:download',
|
49 | description: 'downloads database backup',
|
50 | needsApp: true,
|
51 | needsAuth: true,
|
52 | args: [{name: 'backup_id', optional: true}],
|
53 | flags: [
|
54 | {name: 'output', char: 'o', description: 'location to download to. Defaults to latest.dump', hasValue: true}
|
55 | ],
|
56 | run: cli.command({preauth: true}, co.wrap(run))
|
57 | }
|