63 lines
1.7 KiB
JavaScript
63 lines
1.7 KiB
JavaScript
|
module.exports = Extract;
|
||
|
|
|
||
|
|
const Parse = require('./parse');
|
||
|
|
const fs = require('fs-extra');
|
||
|
|
const path = require('path');
|
||
|
|
const stream = require('stream');
|
||
|
|
const duplexer2 = require('duplexer2');
|
||
|
|
|
||
|
|
function Extract (opts) {
|
||
|
|
// make sure path is normalized before using it
|
||
|
|
opts.path = path.resolve(path.normalize(opts.path));
|
||
|
|
|
||
|
|
const parser = new Parse(opts);
|
||
|
|
|
||
|
|
const outStream = new stream.Writable({objectMode: true});
|
||
|
|
outStream._write = async function(entry, encoding, cb) {
|
||
|
|
|
||
|
|
// to avoid zip slip (writing outside of the destination), we resolve
|
||
|
|
// the target path, and make sure it's nested in the intended
|
||
|
|
// destination, or not extract it otherwise.
|
||
|
|
// NOTE: Need to normalize to forward slashes for UNIX OS's to properly
|
||
|
|
// ignore the zip slipped file entirely
|
||
|
|
const extractPath = path.join(opts.path, entry.path.replace(/\\/g, '/'));
|
||
|
|
if (extractPath.indexOf(opts.path) != 0) {
|
||
|
|
return cb();
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
if (entry.type == 'Directory') {
|
||
|
|
await fs.ensureDir(extractPath);
|
||
|
|
return cb();
|
||
|
|
}
|
||
|
|
|
||
|
|
await fs.ensureDir(path.dirname(extractPath));
|
||
|
|
|
||
|
|
const writer = opts.getWriter ? opts.getWriter({path: extractPath}) : fs.createWriteStream(extractPath);
|
||
|
|
|
||
|
|
entry.pipe(writer)
|
||
|
|
.on('error', cb)
|
||
|
|
.on('close', cb);
|
||
|
|
};
|
||
|
|
|
||
|
|
const extract = duplexer2(parser, outStream);
|
||
|
|
parser.once('crx-header', function(crxHeader) {
|
||
|
|
extract.crxHeader = crxHeader;
|
||
|
|
});
|
||
|
|
|
||
|
|
parser
|
||
|
|
.pipe(outStream)
|
||
|
|
.on('finish', function() {
|
||
|
|
extract.emit('close');
|
||
|
|
});
|
||
|
|
|
||
|
|
extract.promise = function() {
|
||
|
|
return new Promise(function(resolve, reject) {
|
||
|
|
extract.on('close', resolve);
|
||
|
|
extract.on('error', reject);
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
return extract;
|
||
|
|
}
|