127 lines
4.5 KiB
JavaScript
127 lines
4.5 KiB
JavaScript
'use strict';
|
|
|
|
exports = module.exports = {
|
|
getSignedUploadUrl: getSignedUploadUrl,
|
|
getSignedDownloadUrl: getSignedDownloadUrl,
|
|
|
|
copyObject: copyObject,
|
|
|
|
getAllPaged: getAllPaged
|
|
};
|
|
|
|
var assert = require('assert'),
|
|
AWS = require('aws-sdk'),
|
|
config = require('../config.js'),
|
|
superagent = require('superagent'),
|
|
util = require('util');
|
|
|
|
function getBackupCredentials(backupConfig, callback) {
|
|
assert.strictEqual(typeof backupConfig, 'object');
|
|
assert.strictEqual(typeof callback, 'function');
|
|
assert(backupConfig.token);
|
|
|
|
var url = config.apiServerOrigin() + '/api/v1/boxes/' + config.fqdn() + '/awscredentials';
|
|
superagent.post(url).query({ token: backupConfig.token }).end(function (error, result) {
|
|
if (error && !error.response) return callback(error);
|
|
if (result.statusCode !== 201) return callback(new Error(result.text));
|
|
if (!result.body || !result.body.credentials) return callback(new Error('Unexpected response'));
|
|
|
|
var credentials = {
|
|
accessKeyId: result.body.credentials.AccessKeyId,
|
|
secretAccessKey: result.body.credentials.SecretAccessKey,
|
|
sessionToken: result.body.credentials.SessionToken,
|
|
region: 'us-east-1'
|
|
};
|
|
|
|
if (backupConfig.endpoint) credentials.endpoint = new AWS.Endpoint(backupConfig.endpoint);
|
|
|
|
callback(null, credentials);
|
|
});
|
|
}
|
|
|
|
function getAllPaged(backupConfig, page, perPage, callback) {
|
|
assert.strictEqual(typeof backupConfig, 'object');
|
|
assert.strictEqual(typeof page, 'number');
|
|
assert.strictEqual(typeof perPage, 'number');
|
|
assert.strictEqual(typeof callback, 'function');
|
|
|
|
var url = config.apiServerOrigin() + '/api/v1/boxes/' + config.fqdn() + '/backups';
|
|
superagent.get(url).query({ token: backupConfig.token }).end(function (error, result) {
|
|
if (error && !error.response) return callback(error);
|
|
if (result.statusCode !== 200) return callback(new Error(result.text));
|
|
if (!result.body || !util.isArray(result.body.backups)) return callback(new Error('Unexpected response'));
|
|
|
|
return callback(null, result.body.backups);
|
|
});
|
|
}
|
|
|
|
function getSignedUploadUrl(backupConfig, filename, callback) {
|
|
assert.strictEqual(typeof backupConfig, 'object');
|
|
assert.strictEqual(typeof filename, 'string');
|
|
assert.strictEqual(typeof callback, 'function');
|
|
|
|
if (!backupConfig.bucket || !backupConfig.prefix) return new Error('Invalid configuration'); // prevent error in s3
|
|
|
|
getBackupCredentials(backupConfig, function (error, credentials) {
|
|
if (error) return callback(error);
|
|
|
|
var s3 = new AWS.S3(credentials);
|
|
|
|
var params = {
|
|
Bucket: backupConfig.bucket,
|
|
Key: backupConfig.prefix + '/' + filename,
|
|
Expires: 60 * 30 /* 30 minutes */
|
|
};
|
|
|
|
var url = s3.getSignedUrl('putObject', params);
|
|
|
|
callback(null, { url: url, sessionToken: credentials.sessionToken });
|
|
});
|
|
}
|
|
|
|
function getSignedDownloadUrl(backupConfig, info, callback) {
|
|
assert.strictEqual(typeof backupConfig, 'object');
|
|
assert.strictEqual(typeof info, 'object');
|
|
assert.strictEqual(typeof callback, 'function');
|
|
|
|
if (!info.bucket || !info.prefix) return new Error('Invalid configuration'); // prevent error in s3
|
|
|
|
getBackupCredentials(backupConfig, function (error, credentials) {
|
|
if (error) return callback(error);
|
|
|
|
var s3 = new AWS.S3(credentials);
|
|
|
|
var params = {
|
|
Bucket: info.bucket,
|
|
Key: info.prefix + '/' + info.filename,
|
|
Expires: 60 * 30 /* 30 minutes */
|
|
};
|
|
|
|
var url = s3.getSignedUrl('getObject', params);
|
|
|
|
callback(null, { url: url, sessionToken: credentials.sessionToken });
|
|
});
|
|
}
|
|
|
|
function copyObject(backupConfig, from, to, callback) {
|
|
assert.strictEqual(typeof backupConfig, 'object');
|
|
assert.strictEqual(typeof from, 'string');
|
|
assert.strictEqual(typeof to, 'string');
|
|
assert.strictEqual(typeof callback, 'function');
|
|
|
|
if (!backupConfig.bucket || !backupConfig.prefix) return new Error('Invalid configuration'); // prevent error in s3
|
|
|
|
getBackupCredentials(backupConfig, function (error, credentials) {
|
|
if (error) return callback(error);
|
|
|
|
var params = {
|
|
Bucket: backupConfig.bucket, // target bucket
|
|
Key: backupConfig.prefix + '/' + to, // target file
|
|
CopySource: backupConfig.bucket + '/' + backupConfig.prefix + '/' + from, // source
|
|
};
|
|
|
|
var s3 = new AWS.S3(credentials);
|
|
s3.copyObject(params, callback);
|
|
});
|
|
}
|