2018-02-23 14:57:58 +01:00

120 lines
3.6 KiB
JavaScript

'use strict';
/**
* Upload.js service
*
* @description: A set of functions similar to controller's actions to avoid code duplication.
*/
const _ = require('lodash');
const path = require('path');
const fs = require('fs');
const toArray = require('stream-to-array');
const uuid = require('uuid/v4');
module.exports = {
bufferize: async values => {
if (_.size(values.files) === 0) {
throw 'Missing files.';
}
// files is always an array to map on
const files = _.isArray(values.files) ? values.files : [values.files];
// transform all files in buffer
return Promise.all(
files.map(async stream => {
const parts = await toArray(fs.createReadStream(stream.path));
const buffers = parts.map(
part => _.isBuffer(part) ? part : Buffer.from(part)
);
return {
name: stream.name,
hash: uuid().replace(/-/g, ''),
ext: stream.name.split('.').length > 1 ? `.${_.last(stream.name.split('.'))}` : '',
buffer: Buffer.concat(buffers),
mime: stream.type,
size: (stream.size / 1000).toFixed(2)
};
})
);
},
upload: async (files, config) => {
// get upload provider settings to configure the provider to use
const provider = _.cloneDeep(_.find(strapi.plugins.upload.config.providers, {provider: config.provider}));
_.assign(provider, config);
const actions = provider.init(strapi, config);
// execute upload function of the provider for all files
return Promise.all(
files.map(async file => {
await actions.upload(file);
// remove buffer to don't save it
delete file.buffer;
await strapi.plugins['upload'].services.upload.add(file);
})
);
},
add: async (values) => {
// Use Content Manager business logic to handle relation.
if (strapi.plugins['content-manager']) {
return await strapi.plugins['content-manager'].services['contentmanager'].add({
model: 'file'
}, values, 'upload');
}
return strapi.query('file', 'upload').create(values);
},
edit: async (params, values) => {
// Use Content Manager business logic to handle relation.
if (strapi.plugins['content-manager']) {
params.model = 'file';
params.id = (params._id || params.id);
return await strapi.plugins['content-manager'].services['contentmanager'].edit(params, values, 'upload');
}
return strapi.query('file', 'upload').update(_.assign(params, values));
},
fetch: (params) => {
return strapi.query('file', 'upload').findOne(_.pick(params, ['_id', 'id']));
},
fetchAll: (params) => {
return strapi.query('file', 'upload').find(strapi.utils.models.convertParams('file', params));
},
count: async (params, source) => {
return await strapi.query('file', 'upload').count();
},
remove: async (params, config) => {
const file = await strapi.plugins['upload'].services.upload.fetch(params);
// get upload provider settings to configure the provider to use
const provider = _.cloneDeep(_.find(strapi.plugins.upload.config.providers, {provider: config.provider}));
_.assign(provider, config);
const actions = provider.init(strapi, config);
// execute delete function of the provider
await actions.delete(file);
// Use Content Manager business logic to handle relation.
if (strapi.plugins['content-manager']) {
params.model = 'file';
params.id = (params._id || params.id);
await strapi.plugins['content-manager'].services['contentmanager'].delete(params, {source: 'upload'});
}
return strapi.query('file', 'upload').delete(params);
}
};