Skip to content

Instantly share code, notes, and snippets.

@Dafrok
Last active Nov 13, 2018
Embed
What would you like to do?
/**
* @file 在内存中打包 zip 文件的类,需要设置 `/var/run/${namespace}` 的权限为777
* @author o.o@mug.dog
*/
import * as fs from 'fs';
import * as path from 'path';
import * as mkdirp from 'mkdirp';
import * as archiver from 'archiver';
import * as copydir from 'copy-dir';
import * as rmdir from 'rimraf';
export default class Pack {
constructor(dir, namespace = 'pack') {
this.runtimeDir = `/var/run/${namespace}`;
this.baseDir = path.join(this.runtimeDir, dir);
this.sourceDir = path.join(this.baseDir, './source');
this.outputDir = path.join(this.baseDir, './output');
mkdirp.sync(this.sourceDir);
mkdirp.sync(this.outputDir);
}
// 从文件夹导入
import(sourcePath) {
copydir.sync(sourcePath, this.sourceDir);
}
// 向 pack 中单个或多个添加文件
addFile(filePath, content) {
const paths = Array.isArray(filePath) ? filePath : [filePath];
paths.forEach(filePath => {
const targetPath = path.join(this.sourceDir, filePath);
mkdirp.sync(targetPath.split('/').slice(0, -1).join('/'));
fs.writeFileSync(
targetPath,
typeof content === 'object' ? JSON.stringify(content) : content
);
});
}
// 从 pack 中删除单个或多个文件
dropFile(filePath) {
const paths = Array.isArray(filePath) ? filePath : [filePath];
paths.forEach(filePath => {
const targetPath = path.join(this.sourceDir, filePath);
try {
const stat = fs.statSync(targetPath);
if (stat) {
fs.unlinkSync(targetPath);
}
}
catch (e) {
console.log(e);
}
});
}
// 输出一个 zip 文件并返回输出地址
output(filename = 'pack') {
const zipDir = path.join(this.outputDir, `./${filename}.zip`);
const output = fs.createWriteStream(zipDir);
const archive = archiver('zip', {zlib: {level: 9}}); // 压缩等级
return new Promise((resolve, reject) => {
output.on('close', () => {
resolve(zipDir);
});
archive.on('error', err => {
reject(err);
});
archive.pipe(output);
archive.directory(this.sourceDir, filename);
archive.finalize();
});
}
// 删除源文件和输出的文件
dispose() {
rmdir(this.baseDir, err => {
if (err) {
throw err;
}
});
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment