This commit is contained in:
parent
08de6ea171
commit
18b27ff320
|
@ -1,20 +1,36 @@
|
||||||
function createProvider(options) {
|
/**
|
||||||
|
* Create a client instance based on the options
|
||||||
|
* @param options
|
||||||
|
* @returns {*}
|
||||||
|
*/
|
||||||
|
function createClient(options) {
|
||||||
options = options || {};
|
options = options || {};
|
||||||
var provider = options.provider || 'filesystem';
|
var provider = options.provider || 'filesystem';
|
||||||
|
|
||||||
if ('function' !== typeof provider) {
|
try {
|
||||||
try {
|
// Try to load the provider from providers folder
|
||||||
// Try to load the provider from providers folder
|
provider = require('./providers/' + provider);
|
||||||
provider = require('./providers/' + provider);
|
return provider.createClient(options);
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
// Fall back to pkgcloud
|
// Fall back to pkgcloud
|
||||||
return require('pkgcloud').storage.createClient(options);
|
return require('pkgcloud').storage.createClient(options);
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return new provider(options);
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
module.exports = createProvider;
|
/**
|
||||||
module.exports.createProvider = createProvider;
|
* Look up a provider by name
|
||||||
|
* @param provider
|
||||||
|
* @returns {*}
|
||||||
|
*/
|
||||||
|
function getProvider(provider) {
|
||||||
|
try {
|
||||||
|
// Try to load the provider from providers folder
|
||||||
|
return require('./providers/' + provider);
|
||||||
|
} catch (err) {
|
||||||
|
// Fall back to pkgcloud
|
||||||
|
return require('pkgcloud').providers[provider];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports.createClient = createClient;
|
||||||
|
module.exports.getProvider = getProvider;
|
38
lib/index.js
38
lib/index.js
|
@ -1,10 +1,12 @@
|
||||||
var factory = require('./factory');
|
var factory = require('./factory');
|
||||||
var handler = require('./storage-handler');
|
var handler = require('./storage-handler');
|
||||||
|
|
||||||
|
var storage = require('pkgcloud').storage;
|
||||||
|
|
||||||
module.exports = StorageService;
|
module.exports = StorageService;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @param options The provider instance or options to create a provider
|
* @param options The options to create a provider
|
||||||
* @returns {StorageService}
|
* @returns {StorageService}
|
||||||
* @constructor
|
* @constructor
|
||||||
*/
|
*/
|
||||||
|
@ -12,27 +14,29 @@ function StorageService(options) {
|
||||||
if (!(this instanceof StorageService)) {
|
if (!(this instanceof StorageService)) {
|
||||||
return new StorageService(options);
|
return new StorageService(options);
|
||||||
}
|
}
|
||||||
if('function' === typeof options) {
|
this.provider = options.provider;
|
||||||
this.provider = options;
|
this.client = factory.createClient(options);
|
||||||
} else {
|
|
||||||
this.provider = factory.createProvider(options);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
StorageService.prototype.getContainers = function (cb) {
|
StorageService.prototype.getContainers = function (cb) {
|
||||||
return this.provider.getContainers(cb);
|
return this.client.getContainers(cb);
|
||||||
}
|
}
|
||||||
|
|
||||||
StorageService.prototype.createContainer = function (options, cb) {
|
StorageService.prototype.createContainer = function (options, cb) {
|
||||||
return this.provider.createContainer(options, cb);
|
options = options || {};
|
||||||
|
if('object' === typeof options && !(options instanceof storage.Container)) {
|
||||||
|
var Container = factory.getProvider(this.provider).Container;
|
||||||
|
options = new Container(this.client, options);
|
||||||
|
}
|
||||||
|
return this.client.createContainer(options, cb);
|
||||||
}
|
}
|
||||||
|
|
||||||
StorageService.prototype.destroyContainer = function (container, cb) {
|
StorageService.prototype.destroyContainer = function (container, cb) {
|
||||||
return this.provider.destroyContainer(container, cb);
|
return this.client.destroyContainer(container, cb);
|
||||||
}
|
}
|
||||||
|
|
||||||
StorageService.prototype.getContainer = function (container, cb) {
|
StorageService.prototype.getContainer = function (container, cb) {
|
||||||
return this.provider.getContainer(container, cb);
|
return this.client.getContainer(container, cb);
|
||||||
}
|
}
|
||||||
|
|
||||||
// File related functions
|
// File related functions
|
||||||
|
@ -45,7 +49,7 @@ StorageService.prototype.uploadStream = function (container, file, options, cb)
|
||||||
if(container) options.container = container;
|
if(container) options.container = container;
|
||||||
if(file) options.remote = file;
|
if(file) options.remote = file;
|
||||||
|
|
||||||
return this.provider.upload(options, cb);
|
return this.client.upload(options, cb);
|
||||||
}
|
}
|
||||||
|
|
||||||
StorageService.prototype.downloadStream = function (container, file, options, cb) {
|
StorageService.prototype.downloadStream = function (container, file, options, cb) {
|
||||||
|
@ -57,27 +61,27 @@ StorageService.prototype.downloadStream = function (container, file, options, cb
|
||||||
if(container) options.container = container;
|
if(container) options.container = container;
|
||||||
if(file) options.remote = file;
|
if(file) options.remote = file;
|
||||||
|
|
||||||
return this.provider.download(options, cb);
|
return this.client.download(options, cb);
|
||||||
}
|
}
|
||||||
|
|
||||||
StorageService.prototype.getFiles = function (container, download, cb) {
|
StorageService.prototype.getFiles = function (container, download, cb) {
|
||||||
return this.provider.getFiles(container, download, cb);
|
return this.client.getFiles(container, download, cb);
|
||||||
}
|
}
|
||||||
|
|
||||||
StorageService.prototype.getFile = function (container, file, cb) {
|
StorageService.prototype.getFile = function (container, file, cb) {
|
||||||
return this.provider.getFile(container, file, cb);
|
return this.client.getFile(container, file, cb);
|
||||||
}
|
}
|
||||||
|
|
||||||
StorageService.prototype.removeFile = function (container, file, cb) {
|
StorageService.prototype.removeFile = function (container, file, cb) {
|
||||||
return this.provider.removeFile(container, file, cb);
|
return this.client.removeFile(container, file, cb);
|
||||||
}
|
}
|
||||||
|
|
||||||
StorageService.prototype.upload = function (req, res, cb) {
|
StorageService.prototype.upload = function (req, res, cb) {
|
||||||
return handler.upload(this.provider, req, res, cb);
|
return handler.upload(this.client, req, res, cb);
|
||||||
}
|
}
|
||||||
|
|
||||||
StorageService.prototype.download = function (req, res, cb) {
|
StorageService.prototype.download = function (req, res, cb) {
|
||||||
return handler.download(this.provider, req, res, cb);
|
return handler.download(this.client, req, res, cb);
|
||||||
}
|
}
|
||||||
|
|
||||||
StorageService.modelName = 'storage';
|
StorageService.modelName = 'storage';
|
||||||
|
|
|
@ -8,7 +8,12 @@ var fs = require('fs'),
|
||||||
File = require('./file').File,
|
File = require('./file').File,
|
||||||
Container = require('./container').Container;
|
Container = require('./container').Container;
|
||||||
|
|
||||||
module.exports = FileSystemProvider;
|
module.exports.File = File;
|
||||||
|
module.exports.Container = Container;
|
||||||
|
module.exports.Client = FileSystemProvider;
|
||||||
|
module.exports.createClient = function (options) {
|
||||||
|
return new FileSystemProvider(options);
|
||||||
|
};
|
||||||
|
|
||||||
function FileSystemProvider(options) {
|
function FileSystemProvider(options) {
|
||||||
options = options || {};
|
options = options || {};
|
||||||
|
@ -24,6 +29,9 @@ var namePattern = new RegExp('[^' + path.sep + '/]+');
|
||||||
function validateName(name, cb) {
|
function validateName(name, cb) {
|
||||||
if (!name) {
|
if (!name) {
|
||||||
cb && process.nextTick(cb.bind(null, new Error('Invalid name: ' + name)));
|
cb && process.nextTick(cb.bind(null, new Error('Invalid name: ' + name)));
|
||||||
|
if(!cb) {
|
||||||
|
console.error('Invalid name: ', name);
|
||||||
|
}
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
var match = namePattern.exec(name);
|
var match = namePattern.exec(name);
|
||||||
|
@ -31,6 +39,9 @@ function validateName(name, cb) {
|
||||||
return true;
|
return true;
|
||||||
} else {
|
} else {
|
||||||
cb && process.nextTick(cb.bind(null, new Error('Invalid name: ' + name)));
|
cb && process.nextTick(cb.bind(null, new Error('Invalid name: ' + name)));
|
||||||
|
if(!cb) {
|
||||||
|
console.error('Invalid name: ', name);
|
||||||
|
}
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
var FileSystemProvider = require('../lib/providers/filesystem/index.js');
|
var FileSystemProvider = require('../lib/providers/filesystem/index.js').Client;
|
||||||
|
|
||||||
var assert = require('assert');
|
var assert = require('assert');
|
||||||
var path = require('path');
|
var path = require('path');
|
||||||
|
|
|
@ -0,0 +1,130 @@
|
||||||
|
var StorageService = require('../lib/index.js');
|
||||||
|
|
||||||
|
var assert = require('assert');
|
||||||
|
var path = require('path');
|
||||||
|
|
||||||
|
var storageService = new StorageService({root: path.join(__dirname, 'storage'), provider: 'filesystem'});
|
||||||
|
|
||||||
|
describe('Storage service', function () {
|
||||||
|
|
||||||
|
describe('container apis', function () {
|
||||||
|
|
||||||
|
it('should return an empty list of containers', function (done) {
|
||||||
|
storageService.getContainers(function (err, containers) {
|
||||||
|
assert(!err);
|
||||||
|
assert.equal(0, containers.length);
|
||||||
|
done(err, containers);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should create a new container', function (done) {
|
||||||
|
storageService.createContainer({name: 'c1'}, function (err, container) {
|
||||||
|
assert(!err);
|
||||||
|
done(err, container);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should get a container c1', function (done) {
|
||||||
|
storageService.getContainer('c1', function (err, container) {
|
||||||
|
assert(!err);
|
||||||
|
done(err, container);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should not get a container c2', function (done) {
|
||||||
|
storageService.getContainer('c2', function (err, container) {
|
||||||
|
assert(err);
|
||||||
|
done(null, container);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should return one container', function (done) {
|
||||||
|
storageService.getContainers(function (err, containers) {
|
||||||
|
assert(!err);
|
||||||
|
assert.equal(1, containers.length);
|
||||||
|
done(err, containers);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should destroy a container c1', function (done) {
|
||||||
|
storageService.destroyContainer('c1', function (err, container) {
|
||||||
|
assert(!err);
|
||||||
|
done(err, container);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should not get a container c1 after destroy', function (done) {
|
||||||
|
storageService.getContainer('c1', function (err, container) {
|
||||||
|
assert(err);
|
||||||
|
done(null, container);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('file apis', function () {
|
||||||
|
var fs = require('fs');
|
||||||
|
|
||||||
|
it('should create a new container', function (done) {
|
||||||
|
storageService.createContainer({name: 'c1'}, function (err, container) {
|
||||||
|
assert(!err);
|
||||||
|
done(err, container);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should upload a file', function (done) {
|
||||||
|
var writer = storageService.uploadStream('c1', 'f1.txt');
|
||||||
|
fs.createReadStream(path.join(__dirname, 'files/f1.txt')).pipe(writer);
|
||||||
|
writer.on('finish', done);
|
||||||
|
writer.on('error', done);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should download a file', function (done) {
|
||||||
|
var reader = storageService.downloadStream('c1','f1.txt');
|
||||||
|
reader.pipe(fs.createWriteStream(path.join(__dirname, 'files/f1_downloaded.txt')));
|
||||||
|
reader.on('end', done);
|
||||||
|
reader.on('error', done);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should get files for a container', function (done) {
|
||||||
|
storageService.getFiles('c1', function (err, files) {
|
||||||
|
assert(!err);
|
||||||
|
assert.equal(1, files.length);
|
||||||
|
done(err, files);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should get a file', function (done) {
|
||||||
|
storageService.getFile('c1', 'f1.txt', function (err, f) {
|
||||||
|
assert(!err);
|
||||||
|
assert.ok(f);
|
||||||
|
done(err, f);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should remove a file', function (done) {
|
||||||
|
storageService.removeFile('c1', 'f1.txt', function (err) {
|
||||||
|
assert(!err);
|
||||||
|
done(err);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should get no files from a container', function (done) {
|
||||||
|
storageService.getFiles('c1', function (err, files) {
|
||||||
|
assert(!err);
|
||||||
|
assert.equal(0, files.length);
|
||||||
|
done(err, files);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should destroy a container c1', function (done) {
|
||||||
|
storageService.destroyContainer('c1', function (err, container) {
|
||||||
|
// console.error(err);
|
||||||
|
assert(!err);
|
||||||
|
done(err, container);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue