liberama/server/core/Zip/ZipStreamer.js

const fs = require('fs-extra');
const path = require('path');

const zipStream = require('zip-stream');
const unzipStream = require('./node_stream_zip');

class ZipStreamer {
    constructor() {
    }

    //TODO: сделать рекурсивный обход директорий, пока только файлы
    //files = ['filename', 'dirname/']
    pack(zipFile, files, options, entryCallback) {
        return new Promise((resolve, reject) => { (async() => {
            entryCallback = (entryCallback ? entryCallback : () => {});
            const zip = new zipStream(options);

            const outputStream = fs.createWriteStream(zipFile);

            outputStream.on('error', reject);
            outputStream.on('finish', async() => {
                let file = {path: zipFile};
                try {
                    file.size = (await fs.stat(zipFile)).size;
                } catch (e) {
                    reject(e);
                }
                resolve(file);
            });            

            zip.on('error', reject);
            zip.pipe(outputStream);

            const zipAddEntry = (filename) => {
                return new Promise((resolve, reject) => {
                    const basename = path.basename(filename);
                    const source = fs.createReadStream(filename);

                    zip.entry(source, {name: basename}, (err, entry) => {
                        if (err) reject(err);
                        resolve(entry);
                    });
                });
            };

            for (const filename of files) {
                const entry = await zipAddEntry(filename);
                entryCallback({path: entry.name, size: entry.size, compressedSize: entry.csize});
            }

            zip.finish();
        })().catch(reject); });
    }

    unpack(zipFile, outputDir, options, entryCallback) {
        return new Promise((resolve, reject) => {
            entryCallback = (entryCallback ? entryCallback : () => {});
            const {
                limitFileSize = 0,
                limitFileCount = 0,
                decodeEntryNameCallback = false,
            } = options;

            const unzip = new unzipStream({file: zipFile});

            unzip.on('error', reject);

            let files = [];
            unzip.on('extract', (en) => {
                const entry = {path: en.name, size: en.size, compressedSize: en.compressedSize};
                entryCallback(entry);
                files.push(entry);
            });

            unzip.on('ready', () => {
                if (limitFileCount || limitFileSize || decodeEntryNameCallback) {
                    const entries = Object.values(unzip.entries());
                    if (limitFileCount && entries.length > limitFileCount) {
                        reject(new Error('Слишком много файлов'));
                        return;
                    }

                    for (const entry of entries) {
                        if (limitFileSize && !entry.isDirectory && entry.size > limitFileSize) {
                            reject(new Error('Файл слишком большой'));
                            return;
                        }

                        if (decodeEntryNameCallback) {
                            entry.name = (decodeEntryNameCallback(entry.nameRaw)).toString();
                        }
                    }
                }

                unzip.extract(null, outputDir, (err) => {
                    if (err) {
                        reject(err);
                        return;
                    }
                    try {
                        unzip.close();
                        resolve(files);
                    } catch (e) {
                        reject(e);
                    }
                });
            });            
        });
    }

}

module.exports = ZipStreamer;