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: 'newVersion', type: 'number', required: true, description: `The new version number` }, { arg: 'branch', type: 'string', required: true, description: `The branch name` } ], returns: { type: ['object'], root: true }, http: { path: `/upload`, verb: 'POST' } }); Self.upload = async(ctx, appName, newVersion, branch, 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 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, `${newVersion}.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 existBranch = await models.MdbBranch.findOne({ where: {name: branch} }); if (!existBranch) throw new UserError('Not exist this branch'); const branchPath = path.join(accessContainer.client.root, 'branches', branch); await fs.mkdir(branchPath, {recursive: true}); const destinationBranch = path.join(branchPath, `${appName}.7z`); const destinationRoot = path.join(accessContainer.client.root, `${appName}.7z`); try { await fs.unlink(destinationBranch); } catch (e) {} await fs.symlink(destinationFile, destinationBranch); if (branch == 'master') { try { await fs.unlink(destinationRoot); } catch (e) {} await fs.symlink(destinationFile, destinationRoot); } } await models.MdbVersion.upsert({ app: appName, branchFk: branch, version: newVersion }); if (tx) await tx.commit(); } catch (e) { if (tx) await tx.rollback(); if (fs.existsSync(srcFile)) await fs.unlink(srcFile); throw e; } }; };