node-spearhead/lib/do_snapshot/do_delete.js

/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */

/*
 * Copyright 2016 Joyent, Inc.
 *
 * `triton snapshot delete ...`
 */

var assert = require('assert-plus');
var format = require('util').format;
var vasync = require('vasync');

var common = require('../common');
var distractions = require('../distractions');
var errors = require('../errors');


function do_delete(subcmd, opts, args, cb) {
    assert.func(cb, 'cb');

    if (opts.help) {
        this.do_help('help', {}, [subcmd], cb);
        return;
    }

    if (args.length < 2) {
        cb(new errors.UsageError('missing INST and SNAPSHOT-NAME argument(s)'));
        return;
    }

    var cli = this.top;
    var inst = args[0];
    var names = args.slice(1, args.length);

    function wait(name, startTime, next) {
        //  1 'wait': no distraction.
        // >1 'wait': distraction, pass in the N.
        var distraction;
        if (process.stderr.isTTY && opts.wait.length > 1) {
            distraction = distractions.createDistraction(opts.wait.length);
        }

        var cloudapi = cli.tritonapi.cloudapi;
        var waiter = cloudapi.waitForSnapshotStates.bind(cloudapi);

        waiter({
            id: inst,
            name: name,
            states: ['deleted']
        }, function (err, snap) {
            if (distraction) {
                distraction.destroy();
            }
            if (err) {
                return next(err);
            }
            if (snap.state === 'deleted') {
                var duration = Date.now() - startTime;
                var durStr = common.humanDurationFromMs(duration);
                console.log('Deleted snapshot "%s" in %s', name, durStr);

                next();
            } else {
                // shouldn't get here, but...
                next(new Error(format('Failed to delete snapshot "%s"', name)));
            }
        });
    }

    vasync.pipeline({funcs: [
        function confirm(_, next) {
            if (opts.force) {
                return next();
            }

            var msg;
            if (names.length === 1) {
                msg = 'Delete snapshot "' + names[0] + '"? [y/n] ';
            } else {
                msg = format('Delete %d snapshots (%s)? [y/n] ',
                    names.length, names.join(', '));
            }

            common.promptYesNo({msg: msg}, function (answer) {
                if (answer !== 'y') {
                    console.error('Aborting');
                    next(true); // early abort signal
                } else {
                    next();
                }
            });
        },
        function deleteThem(_, next) {
            var startTime = Date.now();

            vasync.forEachParallel({
                inputs: names,
                func: function deleteOne(name, nextName) {
                    cli.tritonapi.cloudapi.deleteMachineSnapshot({
                        id: inst,
                        name: name
                    }, function (err) {
                        if (err) {
                            nextName(err);
                            return;
                        }

                        console.log('Deleting snapshot "%s"', name);

                        if (opts.wait) {
                            wait(name, startTime, nextName);
                        } else {
                            nextName();
                        }
                    });
                }
            }, next);
        }
    ]}, function (err) {
        if (err === true) {
            err = null;
        }
        cb(err);
    });
}


do_delete.options = [
    {
        names: ['help', 'h'],
        type: 'bool',
        help: 'Show this help.'
    },
    {
        names: ['force', 'f'],
        type: 'bool',
        help: 'Skip confirmation of delete.'
    },
    {
        names: ['wait', 'w'],
        type: 'arrayOfBool',
        help: 'Wait for the deletion to complete. Use multiple times for a ' +
            'spinner.'
    }
];
do_delete.help = [
    'Remove a snapshot from a machine.',
    '',
    'Usage:',
    '     {{name}} delete [<options>] SNAPSHOT-NAME [SNAPSHOT-NAME...]',
    '',
    '{{options}}'
].join('\n');

do_delete.aliases = ['rm'];

module.exports = do_delete;