salix/back/methods/image/upload.js

const UserError = require('vn-loopback/util/user-error');
const fs = require('fs/promises');
const path = require('path');
const isProduction = require('vn-loopback/server/boot/isProduction');

module.exports = Self => {
    Self.remoteMethodCtx('upload', {
        description: 'Uploads a file and inserts into dms model',
        accessType: 'WRITE',
        accepts: [
            {
                arg: 'id',
                type: 'Number',
                description: 'The entity id',
                required: true
            }, {
                arg: 'collection',
                type: 'string',
                description: 'The collection name',
                required: true
            }
        ],
        returns: {
            type: 'Object',
            root: true
        },
        http: {
            path: `/upload`,
            verb: 'POST'
        }
    });

    Self.upload = async ctx => {
        const models = Self.app.models;
        const TempContainer = models.TempContainer;
        const fileOptions = {};
        const args = ctx.args;

        let tempFilePath;
        try {
            const hasWriteRole = await models.ImageCollection.hasWriteRole(ctx, args.collection);
            if (!hasWriteRole)
                throw new UserError(`You don't have enough privileges`);

            if (!isProduction())
                throw new UserError(`Action not allowed on the test environment`);

            // Upload file to temporary path
            const tempContainer = await TempContainer.container(args.collection);
            const uploaded = await TempContainer.upload(tempContainer.name, ctx.req, ctx.result, fileOptions);
            const [uploadedFile] = Object.values(uploaded.files).map(file => {
                return file[0];
            });

            const file = await TempContainer.getFile(tempContainer.name, uploadedFile.name);
            tempFilePath = path.join(file.client.root, file.container, file.name);

            const fileName = `${args.id}.png`;

            await models.Image.resize({
                collectionName: args.collection,
                srcFile: tempFilePath,
                fileName: fileName,
                entityId: args.id
            });
        } finally {
            try {
                await fs.unlink(tempFilePath);
            } catch (error) { }
        }
    };
};