index.js

"use strict";

const zlib = require("zlib");

class Base {
	constructor(type, options) {
		if(type.startsWith("Brotli")) {
			this._noFlushFlag = zlib.constants.BROTLI_OPERATION_PROCESS;
			if(!Number.isInteger(options.flush)) {
				options.flush = zlib.constants.BROTLI_OPERATION_FLUSH;
			}
		} else {
			this._noFlushFlag = zlib.constants.Z_NO_FLUSH;
			if(!Number.isInteger(options.flush)) {
				options.flush = zlib.constants.Z_SYNC_FLUSH;
			}
		}
		this.instance = new zlib[type](options);
		this._buffer = [];
		this._kError = Object.getOwnPropertySymbols(this.instance).find(x => x.toString().includes("kError"));
		this._backups = [this.instance.close, this.instance._handle, this.instance._handle.close];
	}
	process(_data, _flag) {
		const z = this.instance;
		const nff = this._noFlushFlag;
		const kError = this._kError;
		const [c, h, hc] = this._backups;
		const flag = !Number.isInteger(_flag) ? z._defaultFlushFlag : _flag;
		const data = !Buffer.isBuffer(_data) ? Buffer.from(_data) : _data;
		const buffer = this._buffer;
		let result;
		let error;
		z.close = () => void 0;
		z._handle.close = () => void 0;
		try {
			result = z._processChunk(data, flag);
		} catch(e) {
			error = e;
		}
		z.close = c;
		z._handle = h;
		z._handle.close = hc;
		z.removeAllListeners("error");
		if(error) {
			z.reset();
			z[kError] = null;
			throw error;
		}
		result = Buffer.from(result);
		if(flag === nff) {
			if(result.length) {
				buffer.push(result);
				return Buffer.allocUnsafe(0);
			}
		} else if(buffer.length) {
			buffer.push(result);
			result = Buffer.concat(buffer);
			this._buffer = [];
		}
		return result;
	}
	close() {
		if(!this.instance) { return; }
		this.instance._handle.close();
		this.instance._handle = null;
		this.instance.close();
		this.instance = null;
		this.process = () => void 0;
	}
}

exports.Inflate = class extends Base {
	constructor(options = {}) {
		super("Inflate", options);
	}
};

exports.Deflate = class extends Base {
	constructor(options = {}) {
		super("Deflate", options);
	}
};

exports.InflateRaw = class extends Base {
	constructor(options = {}) {
		super("InflateRaw", options);
	}
};

exports.DeflateRaw = class extends Base {
	constructor(options = {}) {
		super("DeflateRaw", options);
	}
};

exports.Gzip = class extends Base {
	constructor(options = {}) {
		super("Gzip", options);
	}
};

exports.Gunzip = class extends Base {
	constructor(options = {}) {
		super("Gunzip", options);
	}
};

exports.Unzip = class extends Base {
	constructor(options = {}) {
		super("Unzip", options);
	}
};

exports.BrotliCompress = class extends Base {
	constructor(options = {}) {
		super("BrotliCompress", options);
	}
};

exports.BrotliDecompress = class extends Base {
	constructor(options = {}) {
		super("BrotliDecompress", options);
	}
};

exports.constants = zlib.constants;
exports.default = exports;