142 lines
4.4 KiB
JavaScript
142 lines
4.4 KiB
JavaScript
const UserError = require('vn-loopback/util/user-error');
|
|
const fs = require('fs-extra');
|
|
|
|
module.exports = Self => {
|
|
Self.remoteMethodCtx('uploadFile', {
|
|
description: 'Uploads a file and inserts into dms model',
|
|
accessType: 'WRITE',
|
|
accepts: [
|
|
{
|
|
arg: 'warehouseId',
|
|
type: 'Number',
|
|
description: ''
|
|
}, {
|
|
arg: 'companyId',
|
|
type: 'Number',
|
|
description: ''
|
|
}, {
|
|
arg: 'dmsTypeId',
|
|
type: 'Number',
|
|
description: ''
|
|
}, {
|
|
arg: 'reference',
|
|
type: 'String',
|
|
description: ''
|
|
}, {
|
|
arg: 'description',
|
|
type: 'String',
|
|
description: ''
|
|
}, {
|
|
arg: 'hasFile',
|
|
type: 'Boolean',
|
|
description: ''
|
|
}],
|
|
returns: {
|
|
type: 'Object',
|
|
root: true
|
|
},
|
|
http: {
|
|
path: `/uploadFile`,
|
|
verb: 'POST'
|
|
}
|
|
});
|
|
|
|
Self.uploadFile = async(ctx, options) => {
|
|
const storageConnector = Self.app.dataSources.storage.connector;
|
|
const models = Self.app.models;
|
|
const fileOptions = {};
|
|
const args = ctx.args;
|
|
|
|
let tx;
|
|
let myOptions = {};
|
|
|
|
if (typeof options == 'object')
|
|
Object.assign(myOptions, options);
|
|
|
|
if (!myOptions.transaction) {
|
|
tx = await Self.beginTransaction({});
|
|
myOptions.transaction = tx;
|
|
}
|
|
|
|
try {
|
|
const hasWriteRole = await models.DmsType.hasWriteRole(ctx, args.dmsTypeId);
|
|
if (!hasWriteRole)
|
|
throw new UserError(`You don't have enough privileges`);
|
|
|
|
// Upload file to temporary path
|
|
const tempContainer = await getContainer('temp');
|
|
const uploaded = await models.Container.upload(tempContainer.name, ctx.req, ctx.result, fileOptions);
|
|
const files = Object.values(uploaded.files).map(file => {
|
|
return file[0];
|
|
});
|
|
|
|
const addedDms = [];
|
|
for (const file of files) {
|
|
const newDms = await createDms(ctx, file.name, myOptions);
|
|
const pathHash = storageConnector.getPathHash(newDms.id);
|
|
const container = await getContainer(pathHash);
|
|
|
|
const originPath = `${tempContainer.client.root}/${tempContainer.name}/${file.name}`;
|
|
const destinationPath = `${container.client.root}/${pathHash}/${newDms.file}`;
|
|
|
|
fs.rename(originPath, destinationPath);
|
|
|
|
addedDms.push(newDms);
|
|
}
|
|
|
|
if (tx) await tx.commit();
|
|
return addedDms;
|
|
} catch (e) {
|
|
if (tx) await tx.rollback();
|
|
throw e;
|
|
}
|
|
};
|
|
|
|
async function createDms(ctx, fileName, myOptions) {
|
|
const models = Self.app.models;
|
|
const storageConnector = Self.app.dataSources.storage.connector;
|
|
const myUserId = ctx.req.accessToken.userId;
|
|
const myWorker = await models.Worker.findOne({where: {userFk: myUserId}});
|
|
const args = ctx.args;
|
|
|
|
const newDms = await Self.create({
|
|
workerFk: myWorker.id,
|
|
dmsTypeFk: args.dmsTypeId,
|
|
companyFk: args.companyId,
|
|
warehouseFk: args.warehouseId,
|
|
reference: args.reference,
|
|
description: args.description,
|
|
hasFile: args.hasFile
|
|
}, myOptions);
|
|
|
|
const extension = storageConnector.getFileExtension(fileName);
|
|
fileName = `${newDms.id}.${extension}`;
|
|
|
|
return newDms.updateAttribute('file', fileName, myOptions);
|
|
}
|
|
|
|
|
|
/**
|
|
* Returns a container instance
|
|
* If doesn't exists creates a new one
|
|
*
|
|
* @param {String} name Container name
|
|
* @return {Object} Container instance
|
|
*/
|
|
async function getContainer(name) {
|
|
const models = Self.app.models;
|
|
let container;
|
|
try {
|
|
container = await models.Container.getContainer(name);
|
|
} catch (err) {
|
|
if (err.code === 'ENOENT') {
|
|
container = await models.Container.createContainer({
|
|
name: name
|
|
});
|
|
} else throw err;
|
|
}
|
|
|
|
return container;
|
|
}
|
|
};
|