This repository has been archived on 2020-01-20. You can view files and clone it, but cannot push or open issues or pull requests.
node-spearhead/lib/sdc.js

217 lines
5.7 KiB
JavaScript
Raw Normal View History

2014-02-07 23:21:24 +02:00
/*
* Copyright (c) 2014, Joyent, Inc. All rights reserved.
*
2014-02-12 22:22:08 +02:00
* Core SDC driver class.
2014-02-07 23:21:24 +02:00
*/
var p = console.log;
var assert = require('assert-plus');
var async = require('async');
var auth = require('smartdc-auth');
2014-02-08 10:15:26 +02:00
var EventEmitter = require('events').EventEmitter;
2014-02-07 23:21:24 +02:00
var format = require('util').format;
var fs = require('fs');
var path = require('path');
var restify = require('restify');
2014-02-07 23:21:24 +02:00
var cloudapi = require('./cloudapi2');
2014-02-07 23:21:24 +02:00
var common = require('./common');
var loadConfigSync = require('./config').loadConfigSync;
2014-02-12 22:22:08 +02:00
//---- SDC class
2014-02-07 23:21:24 +02:00
/**
2014-02-12 22:22:08 +02:00
* Create a SDC client.
2014-02-07 23:21:24 +02:00
*
* @param options {Object}
* - log {Bunyan Logger}
* - profile {String} Optional. Name of profile to use. Defaults to
* 'defaultProfile' in the config.
*/
2014-02-12 22:22:08 +02:00
function SDC(options) {
2014-02-07 23:21:24 +02:00
assert.object(options, 'options');
assert.object(options.log, 'options.log');
assert.optionalString(options.profile, 'options.profile');
// Make sure a given bunyan logger has reasonable client_re[qs] serializers.
// Note: This was fixed in restify, then broken again in
// https://github.com/mcavage/node-restify/pull/501
if (options.log.serializers &&
(!options.log.serializers.client_req ||
!options.log.serializers.client_req)) {
this.log = options.log.child({
serializers: restify.bunyan.serializers
});
} else {
this.log = options.log;
}
2014-02-07 23:21:24 +02:00
this.config = loadConfigSync();
this.profiles = this.config.profiles;
this.profile = this.getProfile(
options.profile || this.config.defaultProfile);
this.log.trace({profile: this.profile}, 'profile data');
}
2014-02-12 22:22:08 +02:00
SDC.prototype.setDefaultProfile =
2014-02-08 02:49:07 +02:00
function setDefaultProfile(name, callback) {
2014-02-07 23:21:24 +02:00
if (!this.getProfile(name)) {
return callback(new Error('no such profile: ' + name));
}
this.defaultProfileName = this.config.defaultProfile = name;
common.saveConfigSync(this.config);
callback();
};
2014-02-12 22:22:08 +02:00
SDC.prototype.getProfile = function getProfile(name) {
2014-02-07 23:21:24 +02:00
for (var i = 0; i < this.profiles.length; i++) {
if (this.profiles[i].name === name) {
return this.profiles[i];
}
}
};
/**
* Create or update a profile.
*
* @param profile {Object}
* @param options {Object}
* - setDefault {Boolean}
* @param callback {Function} `function (err)`
*/
2014-02-12 22:22:08 +02:00
SDC.prototype.createOrUpdateProfile = function createOrUpdateProfile(
2014-02-07 23:21:24 +02:00
profile, options, callback) {
assert.object(profile, 'profile');
2014-02-08 02:49:07 +02:00
if (typeof (options) === 'function') {
2014-02-07 23:21:24 +02:00
callback = options;
options = {};
}
2014-02-08 02:49:07 +02:00
assert.object(options, 'options');
assert.optionalBool(options.setDefault, 'options.setDefault');
assert.func(callback, 'callback');
2014-02-07 23:21:24 +02:00
var found = false;
for (var i = 0; i < this.profiles.length; i++) {
if (this.profiles[i].name === profile.name) {
this.profiles[i] = profile;
found = true;
}
}
if (!found) {
this.profiles.push(profile);
}
if (options.setDefault) {
this.defaultProfileName = this.config.defaultProfile = profile.name;
}
common.saveConfigSync(this.config);
callback();
};
2014-02-12 22:22:08 +02:00
SDC.prototype.deleteProfile = function deleteProfile(name, callback) {
2014-02-07 23:21:24 +02:00
var found = false;
for (var i = 0; i < this.profiles.length; i++) {
if (this.profiles[i].name === name) {
found = true;
this.profiles.splice(i, 1);
}
}
if (!found) {
return callback(new Error('no such profile: ' + name));
}
if (this.defaultProfileName === name) {
this.defaultProfileName = this.config.defaultProfile = null;
}
common.saveConfigSync(this.config);
callback();
};
2014-02-12 22:22:08 +02:00
SDC.prototype._clientFromDc = function _clientFromDc(dc) {
2014-02-07 23:21:24 +02:00
assert.string(dc, 'dc');
if (!this._clientFromDcCache) {
this._clientFromDcCache = {};
}
if (!this._clientFromDcCache[dc]) {
var prof = this.profile;
var sign;
if (prof.privKey) {
sign = auth.privateKeySigner({
user: prof.user,
2014-02-07 23:21:24 +02:00
keyId: prof.keyId,
key: prof.privKey
});
} else {
sign = auth.cliSigner({
keyId: prof.keyId,
user: prof.user
});
2014-02-07 23:21:24 +02:00
}
var client = cloudapi.createClient({
2014-02-07 23:21:24 +02:00
url: this.config.dcs[dc],
user: prof.user,
2014-02-07 23:21:24 +02:00
version: '*',
rejectUnauthorized: Boolean(prof.rejectUnauthorized),
sign: sign,
log: this.log
});
this._clientFromDcCache[dc] = client;
}
return this._clientFromDcCache[dc];
};
/**
* List machines for the current profile.
*
2014-02-08 10:15:26 +02:00
* var res = this.jc.listMachines();
* res.on('data', function (dc, dcMachines) {
* //...
* });
* res.on('dcError', function (dc, dcErr) {
* //...
* });
* res.on('end', function () {
* //...
* });
*
2014-02-07 23:21:24 +02:00
* @param {Object} options Optional
*/
2014-02-12 22:22:08 +02:00
SDC.prototype.listMachines = function listMachines(options) {
2014-02-07 23:21:24 +02:00
var self = this;
2014-02-08 10:15:26 +02:00
if (options === undefined) {
2014-02-08 02:49:07 +02:00
options = {};
2014-02-07 23:21:24 +02:00
}
assert.object(options, 'options');
2014-02-08 10:15:26 +02:00
var emitter = new EventEmitter();
2014-02-07 23:21:24 +02:00
async.each(
self.profile.dcs || Object.keys(self.config.dcs),
function oneDc(dc, next) {
var client = self._clientFromDc(dc);
client.listMachines(function (err, machines) {
if (err) {
2014-02-08 10:15:26 +02:00
emitter.emit('dcError', dc, err);
2014-02-07 23:21:24 +02:00
} else {
2014-02-08 10:15:26 +02:00
emitter.emit('data', dc, machines);
2014-02-07 23:21:24 +02:00
}
next();
});
},
function done(err) {
2014-02-08 10:15:26 +02:00
emitter.emit('end');
2014-02-07 23:21:24 +02:00
}
);
2014-02-08 10:15:26 +02:00
return emitter;
2014-02-07 23:21:24 +02:00
};
//---- exports
2014-02-12 22:22:08 +02:00
module.exports = SDC;