salix/modules/mdb/back/methods/mdbVersion/upload.js

const fs = require('fs-extra');
const path = require('path');
const UserError = require('vn-loopback/util/user-error');

module.exports = Self => {
    Self.remoteMethodCtx('upload', {
        description: 'Upload and attach a access file',
        accepts: [
            {
                arg: 'appName',
                type: 'string',
                required: true,
                description: 'The app name'
            }, {
                arg: 'toVersion',
                type: 'number',
                required: true,
                description: `The new version number`
            }, {
                arg: 'branch',
                type: 'string',
                required: true,
                description: `The branch name`
            }, {
                arg: 'fromVersion',
                type: 'string',
                required: true,
                description: `The old version number`
            }
        ],
        returns: {
            type: ['object'],
            root: true
        },
        http: {
            path: `/upload`,
            verb: 'POST'
        }
    });

    Self.upload = async(ctx, appName, toVersion, branch, fromVersion, options) => {
        const models = Self.app.models;
        const myOptions = {};
        const TempContainer = models.TempContainer;
        const AccessContainer = models.AccessContainer;
        const fileOptions = {};
        let tx;

        if (typeof options == 'object')
            Object.assign(myOptions, options);

        if (!myOptions.transaction) {
            tx = await Self.beginTransaction({});
            myOptions.transaction = tx;
        }

        let srcFile;
        try {
            const existBranch = await models.MdbBranch.findOne({
                where: {name: branch}
            }, myOptions);

            if (!existBranch)
                throw new UserError('Not exist this branch');

            let lastVersion = await models.MdbBranch.findOne({
                where: {appName}
            }, myOptions);

            if (lastVersion++ != toVersion)
                throw new UserError('Try again');

            const userId = ctx.req.accessToken.userId;
            models.MdbVersionTree.create({
                app: appName,
                version: toVersion,
                branchFk: branch,
                fromVersion,
                userFk: userId
            });

            const tempContainer = await TempContainer.container('access');
            const uploaded = await TempContainer.upload(tempContainer.name, ctx.req, ctx.result, fileOptions);
            const files = Object.values(uploaded.files).map(file => {
                return file[0];
            });
            const uploadedFile = files[0];

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

            const accessContainer = await AccessContainer.container('.archive');
            const destinationFile = path.join(
                accessContainer.client.root, accessContainer.name, appName, `${toVersion}.7z`);

            if (process.env.NODE_ENV == 'test')
                await fs.unlink(srcFile);
            else {
                await fs.move(srcFile, destinationFile, {
                    overwrite: true
                });
                await fs.chmod(destinationFile, 0o644);

                const branchPath = path.join(accessContainer.client.root, 'branches', branch);
                await fs.mkdir(branchPath, {recursive: true});

                const destinationBranch = path.join(branchPath, `${appName}.7z`);
                const destinationRelative = `../../.archive/${appName}/${toVersion}.7z`;
                try {
                    await fs.unlink(destinationBranch);
                } catch (e) {}
                await fs.symlink(destinationRelative, destinationBranch);

                if (branch == 'master') {
                    const destinationRoot = path.join(accessContainer.client.root, `${appName}.7z`);
                    const rootRelative = `./.archive/${appName}/${toVersion}.7z`;
                    try {
                        await fs.unlink(destinationRoot);
                    } catch (e) {}
                    await fs.symlink(rootRelative, destinationRoot);
                }
            }

            await models.MdbVersion.upsert({
                app: appName,
                branchFk: branch,
                version: toVersion
            }, myOptions);

            if (tx) await tx.commit();
        } catch (e) {
            if (tx) await tx.rollback();

            if (fs.existsSync(srcFile))
                fs.unlink(srcFile);

            throw e;
        }
    };
};