joyent/node-triton#54 first pass at 'triton rbac key' and 'triton rbac keys' (with feeling, aka the new files actually added)
This commit is contained in:
Trent Mick
parent
dd0a70820b
commit
74b8f3e42e
323
lib/do_rbac/do_key.js
Normal file
323
lib/do_rbac/do_key.js
Normal file
@ -0,0 +1,323 @@
|
|||||||
|
/*
|
||||||
|
* 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 2015 Joyent, Inc.
|
||||||
|
*
|
||||||
|
* `triton rbac key ...`
|
||||||
|
*/
|
||||||
|
|
||||||
|
var assert = require('assert-plus');
|
||||||
|
var format = require('util').format;
|
||||||
|
var fs = require('fs');
|
||||||
|
var sshpk = require('sshpk');
|
||||||
|
var strsplit = require('strsplit');
|
||||||
|
var vasync = require('vasync');
|
||||||
|
|
||||||
|
var common = require('../common');
|
||||||
|
var errors = require('../errors');
|
||||||
|
|
||||||
|
|
||||||
|
function _showUserKey(opts, cb) {
|
||||||
|
assert.object(opts.cli, 'opts.cli');
|
||||||
|
assert.string(opts.userId, 'opts.userId');
|
||||||
|
assert.string(opts.id, 'opts.id');
|
||||||
|
assert.func(cb, 'cb');
|
||||||
|
var cli = opts.cli;
|
||||||
|
|
||||||
|
cli.tritonapi.cloudapi.getUserKey({
|
||||||
|
userId: opts.userId,
|
||||||
|
// Currently `cloudapi.getUserKey` isn't picky about the `name` being
|
||||||
|
// passed in as the `opts.fingerprint` arg.
|
||||||
|
fingerprint: opts.id
|
||||||
|
}, function onUserKey(err, userKey) {
|
||||||
|
if (err) {
|
||||||
|
cb(err);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (opts.json) {
|
||||||
|
console.log(JSON.stringify(userKey));
|
||||||
|
} else {
|
||||||
|
console.log(common.chomp(userKey.key));
|
||||||
|
}
|
||||||
|
cb();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function _deleteUserKeys(opts, cb) {
|
||||||
|
assert.object(opts.cli, 'opts.cli');
|
||||||
|
assert.string(opts.userId, 'opts.userId');
|
||||||
|
assert.arrayOfString(opts.ids, 'opts.ids');
|
||||||
|
assert.optionalBool(opts.yes, 'opts.yes');
|
||||||
|
assert.func(cb, 'cb');
|
||||||
|
var cli = opts.cli;
|
||||||
|
|
||||||
|
if (opts.ids.length === 0) {
|
||||||
|
cb();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
vasync.pipeline({funcs: [
|
||||||
|
function confirm(_, next) {
|
||||||
|
if (opts.yes) {
|
||||||
|
return next();
|
||||||
|
}
|
||||||
|
var msg;
|
||||||
|
if (opts.ids.length === 1) {
|
||||||
|
msg = 'Delete user key "' + opts.ids[0] + '"? [y/n] ';
|
||||||
|
} else {
|
||||||
|
msg = format('Delete %d user keys (%s)? [y/n] ',
|
||||||
|
opts.ids.length, opts.ids.join(', '));
|
||||||
|
}
|
||||||
|
common.promptYesNo({msg: msg}, function (answer) {
|
||||||
|
if (answer !== 'y') {
|
||||||
|
console.error('Aborting');
|
||||||
|
next(true); // early abort signal
|
||||||
|
} else {
|
||||||
|
next();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
},
|
||||||
|
function deleteThem(_, next) {
|
||||||
|
vasync.forEachPipeline({
|
||||||
|
inputs: opts.ids,
|
||||||
|
func: function deleteOne(id, nextId) {
|
||||||
|
var delOpts = {
|
||||||
|
userId: opts.userId,
|
||||||
|
fingerprint: id
|
||||||
|
};
|
||||||
|
cli.tritonapi.cloudapi.deleteUserKey(delOpts,
|
||||||
|
function (err) {
|
||||||
|
if (err) {
|
||||||
|
nextId(err);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
console.log('Deleted user %s key "%s"',
|
||||||
|
opts.userId, id);
|
||||||
|
nextId();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}, next);
|
||||||
|
}
|
||||||
|
]}, function (err) {
|
||||||
|
if (err === true) {
|
||||||
|
err = null;
|
||||||
|
}
|
||||||
|
cb(err);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
function _addUserKey(opts, cb) {
|
||||||
|
assert.object(opts.cli, 'opts.cli');
|
||||||
|
assert.string(opts.userId, 'opts.userId');
|
||||||
|
assert.string(opts.file, 'opts.file');
|
||||||
|
assert.optionalString(opts.name, 'opts.name');
|
||||||
|
assert.func(cb, 'cb');
|
||||||
|
var cli = opts.cli;
|
||||||
|
|
||||||
|
vasync.pipeline({arg: {}, funcs: [
|
||||||
|
function gatherDataStdin(ctx, next) {
|
||||||
|
if (opts.file !== '-') {
|
||||||
|
return next();
|
||||||
|
}
|
||||||
|
var stdin = '';
|
||||||
|
process.stdin.resume();
|
||||||
|
process.stdin.on('data', function (chunk) {
|
||||||
|
stdin += chunk;
|
||||||
|
});
|
||||||
|
process.stdin.on('end', function () {
|
||||||
|
ctx.data = stdin;
|
||||||
|
ctx.from = '<stdin>';
|
||||||
|
next();
|
||||||
|
});
|
||||||
|
},
|
||||||
|
function gatherDataFile(ctx, next) {
|
||||||
|
if (!opts.file || opts.file === '-') {
|
||||||
|
return next();
|
||||||
|
}
|
||||||
|
ctx.data = fs.readFileSync(opts.file);
|
||||||
|
ctx.from = opts.file;
|
||||||
|
next();
|
||||||
|
},
|
||||||
|
function validateData(ctx, next) {
|
||||||
|
try {
|
||||||
|
sshpk.parseKey(ctx.data, 'ssh', ctx.from);
|
||||||
|
} catch (keyErr) {
|
||||||
|
next(keyErr);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
next();
|
||||||
|
},
|
||||||
|
function createIt(ctx, next) {
|
||||||
|
var createOpts = {
|
||||||
|
userId: opts.userId,
|
||||||
|
key: ctx.data.toString('utf8')
|
||||||
|
};
|
||||||
|
if (opts.name) {
|
||||||
|
createOpts.name = opts.name;
|
||||||
|
}
|
||||||
|
cli.tritonapi.cloudapi.createUserKey(createOpts,
|
||||||
|
function (err, userKey) {
|
||||||
|
if (err) {
|
||||||
|
next(err);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (userKey.name) {
|
||||||
|
console.log('Added user %s key "%s" (%s)',
|
||||||
|
opts.userId, userKey.name, userKey.fingerprint);
|
||||||
|
} else {
|
||||||
|
console.log('Added user %s key %s',
|
||||||
|
opts.userId, userKey.fingerprint);
|
||||||
|
}
|
||||||
|
var extra = '';
|
||||||
|
if (userKey.name) {
|
||||||
|
extra = format(' (%s)', userKey.name);
|
||||||
|
}
|
||||||
|
console.log('Added user %s key "%s"%s',
|
||||||
|
opts.userId, userKey.fingerprint, extra);
|
||||||
|
next();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
]}, cb);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
function do_key(subcmd, opts, args, cb) {
|
||||||
|
if (opts.help) {
|
||||||
|
this.do_help('help', {}, [subcmd], cb);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Which action?
|
||||||
|
var actions = [];
|
||||||
|
if (opts.add) { actions.push('add'); }
|
||||||
|
if (opts['delete']) { actions.push('delete'); }
|
||||||
|
var action;
|
||||||
|
if (actions.length === 0) {
|
||||||
|
action = 'show';
|
||||||
|
} else if (actions.length > 1) {
|
||||||
|
return cb(new errors.UsageError(
|
||||||
|
'only one action option may be used at once'));
|
||||||
|
} else {
|
||||||
|
action = actions[0];
|
||||||
|
}
|
||||||
|
|
||||||
|
// Arg count validation.
|
||||||
|
if (action === 'show') {
|
||||||
|
if (args.length === 0) {
|
||||||
|
cb(new errors.UsageError('missing USER and KEY arguments'));
|
||||||
|
return;
|
||||||
|
} else if (args.length === 1) {
|
||||||
|
cb(new errors.UsageError('missing KEY argument'));
|
||||||
|
return;
|
||||||
|
} else if (args.length > 2) {
|
||||||
|
cb(new errors.UsageError('incorrect number of arguments'));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
} else if (action === 'delete') {
|
||||||
|
if (args.length === 0) {
|
||||||
|
cb(new errors.UsageError('missing USER argument'));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
} else if (action === 'add') {
|
||||||
|
if (args.length === 0) {
|
||||||
|
cb(new errors.UsageError('missing USER and FILE arguments'));
|
||||||
|
return;
|
||||||
|
} else if (args.length === 1) {
|
||||||
|
cb(new errors.UsageError('missing FILE argument'));
|
||||||
|
return;
|
||||||
|
} else if (args.length > 2) {
|
||||||
|
cb(new errors.UsageError('incorrect number of arguments'));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
switch (action) {
|
||||||
|
case 'show':
|
||||||
|
_showUserKey({
|
||||||
|
cli: this.top,
|
||||||
|
userId: args[0],
|
||||||
|
id: args[1],
|
||||||
|
json: opts.json
|
||||||
|
}, cb);
|
||||||
|
break;
|
||||||
|
case 'delete':
|
||||||
|
_deleteUserKeys({
|
||||||
|
cli: this.top,
|
||||||
|
userId: args[0],
|
||||||
|
ids: args.slice(1),
|
||||||
|
yes: opts.yes
|
||||||
|
}, cb);
|
||||||
|
break;
|
||||||
|
case 'add':
|
||||||
|
_addUserKey({
|
||||||
|
cli: this.top,
|
||||||
|
name: opts.name,
|
||||||
|
userId: args[0],
|
||||||
|
file: args[1]
|
||||||
|
}, cb);
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
return cb(new errors.InternalError('unknown action: ' + action));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
do_key.options = [
|
||||||
|
{
|
||||||
|
names: ['help', 'h'],
|
||||||
|
type: 'bool',
|
||||||
|
help: 'Show this help.'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
names: ['json', 'j'],
|
||||||
|
type: 'bool',
|
||||||
|
help: 'JSON stream output.'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
names: ['yes', 'y'],
|
||||||
|
type: 'bool',
|
||||||
|
help: 'Answer yes to confirmation to delete.'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
names: ['name', 'n'],
|
||||||
|
type: 'string',
|
||||||
|
helpArg: 'NAME',
|
||||||
|
help: 'An optional name for an added key.'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
group: 'Action Options'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
names: ['add', 'a'],
|
||||||
|
type: 'bool',
|
||||||
|
help: 'Add a new key.'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
names: ['delete', 'd'],
|
||||||
|
type: 'bool',
|
||||||
|
help: 'Delete the named key.'
|
||||||
|
}
|
||||||
|
];
|
||||||
|
do_key.help = [
|
||||||
|
/* BEGIN JSSTYLED */
|
||||||
|
'Show, upload, and delete RBAC user SSH keys.',
|
||||||
|
'',
|
||||||
|
'Usage:',
|
||||||
|
' {{name}} key USER KEY # show USER\'s KEY',
|
||||||
|
' {{name}} key -d|--delete USER [KEY...] # delete USER\'s KEY',
|
||||||
|
' {{name}} key -a|--add [-n NAME] USER FILE',
|
||||||
|
' # Add a new role. FILE must be a file path to an SSH public',
|
||||||
|
' # key or "-" to pass the public key in on stdin.',
|
||||||
|
'',
|
||||||
|
'{{options}}',
|
||||||
|
'Where "USER" is a full RBAC user "id", "login" name or a "shortid"; and',
|
||||||
|
'KEY is an SSH key "name" or "fingerprint".'
|
||||||
|
/* END JSSTYLED */
|
||||||
|
].join('\n');
|
||||||
|
|
||||||
|
module.exports = do_key;
|
||||||
79
lib/do_rbac/do_keys.js
Normal file
79
lib/do_rbac/do_keys.js
Normal file
@ -0,0 +1,79 @@
|
|||||||
|
/*
|
||||||
|
* 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 2015 Joyent, Inc.
|
||||||
|
*
|
||||||
|
* `triton rbac keys ...`
|
||||||
|
*/
|
||||||
|
|
||||||
|
var common = require('../common');
|
||||||
|
var errors = require('../errors');
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
function do_keys(subcmd, opts, args, cb) {
|
||||||
|
if (opts.help) {
|
||||||
|
this.do_help('help', {}, [subcmd], cb);
|
||||||
|
return;
|
||||||
|
} else if (args.length === 0) {
|
||||||
|
cb(new errors.UsageError('no USER argument given'));
|
||||||
|
return;
|
||||||
|
} else if (args.length !== 1) {
|
||||||
|
cb(new errors.UsageError('invalid args: ' + args));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
this.top.tritonapi.cloudapi.listUserKeys({userId: args[0]},
|
||||||
|
function (err, userKeys) {
|
||||||
|
if (err) {
|
||||||
|
cb(err);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (opts.json) {
|
||||||
|
common.jsonStream(userKeys);
|
||||||
|
} else {
|
||||||
|
userKeys.forEach(function (key) {
|
||||||
|
console.log(common.chomp(key.key));
|
||||||
|
});
|
||||||
|
}
|
||||||
|
cb();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
do_keys.options = [
|
||||||
|
{
|
||||||
|
names: ['help', 'h'],
|
||||||
|
type: 'bool',
|
||||||
|
help: 'Show this help.'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
names: ['json', 'j'],
|
||||||
|
type: 'bool',
|
||||||
|
help: 'JSON output.'
|
||||||
|
}
|
||||||
|
];
|
||||||
|
|
||||||
|
do_keys.help = (
|
||||||
|
/* BEGIN JSSTYLED */
|
||||||
|
'List RBAC user SSH keys.\n' +
|
||||||
|
'\n' +
|
||||||
|
'Usage:\n' +
|
||||||
|
' {{name}} keys [<options>] USER\n' +
|
||||||
|
'\n' +
|
||||||
|
'{{options}}' +
|
||||||
|
'\n' +
|
||||||
|
'Where "USER" is an RBAC user id, login or short id. By default this\n' +
|
||||||
|
'lists just the key content for each key -- in other words, content\n' +
|
||||||
|
'appropriate for a "~/.ssh/authorized_keys" file.\n' +
|
||||||
|
'Use `{{name}} keys -j USER` to see all fields.\n'
|
||||||
|
/* END JSSTYLED */
|
||||||
|
);
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
module.exports = do_keys;
|
||||||
Reference in New Issue
Block a user