148 lines
4.0 KiB
JavaScript
148 lines
4.0 KiB
JavaScript
/*
|
|
* 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 instance enable-firewall ...`
|
|
*/
|
|
|
|
var assert = require('assert-plus');
|
|
var format = require('util').format;
|
|
var vasync = require('vasync');
|
|
|
|
var common = require('../common');
|
|
var errors = require('../errors');
|
|
|
|
|
|
function do_enable_firewall(subcmd, opts, args, cb) {
|
|
assert.func(cb, 'cb');
|
|
|
|
if (opts.help) {
|
|
this.do_help('help', {}, [subcmd], cb);
|
|
return;
|
|
}
|
|
|
|
if (args.length === 0) {
|
|
cb(new errors.UsageError('Missing <inst> argument(s)'));
|
|
return;
|
|
}
|
|
|
|
var cli = this.top;
|
|
var insts = args;
|
|
|
|
function wait(instId, startTime, next) {
|
|
var cloudapi = cli.tritonapi.cloudapi;
|
|
var waiter = cloudapi.waitForMachineFirewallState.bind(cloudapi);
|
|
|
|
waiter({
|
|
id: instId,
|
|
state: true
|
|
}, function (err, inst) {
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
|
|
if (inst.firewall_enabled === true) {
|
|
var duration = Date.now() - startTime;
|
|
var durStr = common.humanDurationFromMs(duration);
|
|
console.log('Enabled firewall for instance "%s" in %s', instId,
|
|
durStr);
|
|
next();
|
|
} else {
|
|
// shouldn't get here, but...
|
|
var msg = 'Failed to enable firewall for instance "%s"';
|
|
next(new Error(format(msg, instId)));
|
|
}
|
|
});
|
|
}
|
|
|
|
vasync.pipeline({funcs: [
|
|
function confirm(_, next) {
|
|
if (opts.force) {
|
|
return next();
|
|
}
|
|
|
|
var msg;
|
|
if (insts.length === 1) {
|
|
msg = 'Enable firewall for instance "' + insts[0] + '"? [y/n] ';
|
|
} else {
|
|
msg = format('Enable firewalls for %d instances (%s)? [y/n] ',
|
|
insts.length, insts.join(', '));
|
|
}
|
|
|
|
common.promptYesNo({msg: msg}, function (answer) {
|
|
if (answer !== 'y') {
|
|
console.error('Aborting');
|
|
next(true); // early abort signal
|
|
} else {
|
|
next();
|
|
}
|
|
});
|
|
},
|
|
function enableThem(_, next) {
|
|
var startTime = Date.now();
|
|
|
|
vasync.forEachParallel({
|
|
inputs: insts,
|
|
func: function enableOne(instId, nextId) {
|
|
cli.tritonapi.enableInstanceFirewall({
|
|
id: instId
|
|
}, function (err, __, res) {
|
|
if (err) {
|
|
nextId(err);
|
|
return;
|
|
}
|
|
|
|
var msg = 'Enabling firewall for instance "%s"';
|
|
console.log(msg, res.instId);
|
|
|
|
if (opts.wait) {
|
|
wait(res.instId, startTime, nextId);
|
|
} else {
|
|
nextId();
|
|
}
|
|
});
|
|
}
|
|
}, next);
|
|
}
|
|
]}, function (err) {
|
|
if (err === true) {
|
|
err = null;
|
|
}
|
|
cb(err);
|
|
});
|
|
}
|
|
|
|
|
|
do_enable_firewall.options = [
|
|
{
|
|
names: ['help', 'h'],
|
|
type: 'bool',
|
|
help: 'Show this help.'
|
|
},
|
|
{
|
|
names: ['force', 'f'],
|
|
type: 'bool',
|
|
help: 'Skip confirmation to enable.'
|
|
},
|
|
{
|
|
names: ['wait', 'w'],
|
|
type: 'bool',
|
|
help: 'Wait for the firewall to be enabled.'
|
|
}
|
|
];
|
|
do_enable_firewall.help = [
|
|
'Enable the firewall of one or more instances.',
|
|
'',
|
|
'Usage:',
|
|
' {{name}} enable-firewall [<options>] <inst> [<inst>...]',
|
|
'',
|
|
'{{options}}'
|
|
].join('\n');
|
|
|
|
module.exports = do_enable_firewall;
|