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/joyentcloud.js

205 lines
5.5 KiB
JavaScript
Raw Normal View History

2014-02-07 23:21:24 +02:00
/*
* Copyright (c) 2014, Joyent, Inc. All rights reserved.
*
2014-02-08 02:49:07 +02:00
* Core JoyentCloud driver class.
2014-02-07 23:21:24 +02:00
*/
var p = console.log;
var assert = require('assert-plus');
var async = require('async');
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 smartdc = require('smartdc');
var common = require('./common');
var loadConfigSync = require('./config').loadConfigSync;
2014-02-08 02:49:07 +02:00
//---- JoyentCloud class
2014-02-07 23:21:24 +02:00
/**
2014-02-08 02:49:07 +02:00
* Create a JoyentCloud 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-08 02:49:07 +02:00
function JoyentCloud(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');
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-08 02:49:07 +02:00
JoyentCloud.prototype.setDefaultProfile =
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-08 02:49:07 +02:00
JoyentCloud.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-08 02:49:07 +02:00
JoyentCloud.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-08 02:49:07 +02:00
JoyentCloud.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-08 02:49:07 +02:00
JoyentCloud.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 = smartdc.privateKeySigner({
user: prof.account,
keyId: prof.keyId,
key: prof.privKey
});
} else {
sign = smartdc.cliSigner({keyId: prof.keyId, user: prof.account});
}
var client = smartdc.createClient({
url: this.config.dcs[dc],
account: prof.account,
version: '*',
noCache: true, //XXX
rejectUnauthorized: Boolean(prof.rejectUnauthorized),
sign: sign,
// XXX cloudapi.js stupidly uses its own logger, but takes logLevel
logLevel: this.log && this.log.level(),
// Pass our logger to underlying restify client.
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-08 10:15:26 +02:00
JoyentCloud.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-08 02:49:07 +02:00
module.exports = JoyentCloud;