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 lastMethod = await Self.last(ctx, appName, myOptions); lastMethod.version++; if (lastMethod.version != toVersion) throw new UserError('Try again'); const userId = ctx.req.accessToken.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.MdbVersionTree.create({ app: appName, version: toVersion, branchFk: branch, fromVersion, userFk: userId }, myOptions); 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; } }; };