BREAKING CHANGE(SmartArchive): Refactor public API: rename factory/extraction methods, introduce typed interfaces and improved compression tools
This commit is contained in:
@@ -3,6 +3,6 @@
|
||||
*/
|
||||
export const commitinfo = {
|
||||
name: '@push.rocks/smartarchive',
|
||||
version: '4.2.4',
|
||||
version: '5.0.0',
|
||||
description: 'A library for working with archive files, providing utilities for compressing and decompressing data.'
|
||||
}
|
||||
|
||||
@@ -1,44 +1,60 @@
|
||||
var BITMASK = [0, 0x01, 0x03, 0x07, 0x0f, 0x1f, 0x3f, 0x7f, 0xff];
|
||||
import type { IBitReader } from '../interfaces.js';
|
||||
|
||||
// returns a function that reads bits.
|
||||
// takes a buffer iterator as input
|
||||
export function bitIterator(nextBuffer: () => Buffer) {
|
||||
var bit = 0,
|
||||
byte = 0;
|
||||
var bytes = nextBuffer();
|
||||
var f = function (n) {
|
||||
if (n === null && bit != 0) {
|
||||
const BITMASK = [0, 0x01, 0x03, 0x07, 0x0f, 0x1f, 0x3f, 0x7f, 0xff] as const;
|
||||
|
||||
/**
|
||||
* Creates a bit reader function for BZIP2 decompression.
|
||||
* Takes a buffer iterator as input and returns a function that reads bits.
|
||||
*/
|
||||
export function bitIterator(nextBuffer: () => Buffer): IBitReader {
|
||||
let bit = 0;
|
||||
let byte = 0;
|
||||
let bytes = nextBuffer();
|
||||
let _bytesRead = 0;
|
||||
|
||||
const reader = function (n: number | null): number | void {
|
||||
if (n === null && bit !== 0) {
|
||||
// align to byte boundary
|
||||
bit = 0;
|
||||
byte++;
|
||||
return;
|
||||
}
|
||||
var result = 0;
|
||||
while (n > 0) {
|
||||
|
||||
let result = 0;
|
||||
let remaining = n as number;
|
||||
|
||||
while (remaining > 0) {
|
||||
if (byte >= bytes.length) {
|
||||
byte = 0;
|
||||
bytes = nextBuffer();
|
||||
}
|
||||
var left = 8 - bit;
|
||||
if (bit === 0 && n > 0)
|
||||
// @ts-ignore
|
||||
f.bytesRead++;
|
||||
if (n >= left) {
|
||||
|
||||
const left = 8 - bit;
|
||||
|
||||
if (bit === 0 && remaining > 0) {
|
||||
_bytesRead++;
|
||||
}
|
||||
|
||||
if (remaining >= left) {
|
||||
result <<= left;
|
||||
result |= BITMASK[left] & bytes[byte++];
|
||||
bit = 0;
|
||||
n -= left;
|
||||
remaining -= left;
|
||||
} else {
|
||||
result <<= n;
|
||||
result |=
|
||||
(bytes[byte] & (BITMASK[n] << (8 - n - bit))) >> (8 - n - bit);
|
||||
bit += n;
|
||||
n = 0;
|
||||
result <<= remaining;
|
||||
result |= (bytes[byte] & (BITMASK[remaining] << (8 - remaining - bit))) >> (8 - remaining - bit);
|
||||
bit += remaining;
|
||||
remaining = 0;
|
||||
}
|
||||
}
|
||||
|
||||
return result;
|
||||
};
|
||||
// @ts-ignore
|
||||
f.bytesRead = 0;
|
||||
return f;
|
||||
} as IBitReader;
|
||||
|
||||
Object.defineProperty(reader, 'bytesRead', {
|
||||
get: () => _bytesRead,
|
||||
enumerable: true,
|
||||
});
|
||||
|
||||
return reader;
|
||||
}
|
||||
|
||||
@@ -1,23 +1,22 @@
|
||||
export class Bzip2Error extends Error {
|
||||
public name: string = 'Bzip2Error';
|
||||
public message: string;
|
||||
public stack = new Error().stack;
|
||||
import { Bzip2Error, BZIP2_ERROR_CODES } from '../errors.js';
|
||||
import type { IBitReader, IHuffmanGroup } from '../interfaces.js';
|
||||
|
||||
constructor(messageArg: string) {
|
||||
super();
|
||||
this.message = messageArg;
|
||||
}
|
||||
// Re-export Bzip2Error for backward compatibility
|
||||
export { Bzip2Error };
|
||||
|
||||
/**
|
||||
* Throw a BZIP2 error with proper error code
|
||||
*/
|
||||
function throwError(message: string, code: string = BZIP2_ERROR_CODES.INVALID_BLOCK_DATA): never {
|
||||
throw new Bzip2Error(message, code);
|
||||
}
|
||||
|
||||
var messageArg = {
|
||||
Error: function (message) {
|
||||
throw new Bzip2Error(message);
|
||||
},
|
||||
};
|
||||
|
||||
/**
|
||||
* BZIP2 decompression implementation
|
||||
*/
|
||||
export class Bzip2 {
|
||||
public Bzip2Error = Bzip2Error;
|
||||
public crcTable = [
|
||||
// CRC32 lookup table for BZIP2
|
||||
public readonly crcTable: readonly number[] = [
|
||||
0x00000000, 0x04c11db7, 0x09823b6e, 0x0d4326d9, 0x130476dc, 0x17c56b6b,
|
||||
0x1a864db2, 0x1e475005, 0x2608edb8, 0x22c9f00f, 0x2f8ad6d6, 0x2b4bcb61,
|
||||
0x350c9b64, 0x31cd86d3, 0x3c8ea00a, 0x384fbdbd, 0x4c11db70, 0x48d0c6c7,
|
||||
@@ -63,14 +62,24 @@ export class Bzip2 {
|
||||
0xbcb4666d, 0xb8757bda, 0xb5365d03, 0xb1f740b4,
|
||||
];
|
||||
|
||||
array = function (bytes) {
|
||||
var bit = 0,
|
||||
byte = 0;
|
||||
var BITMASK = [0, 0x01, 0x03, 0x07, 0x0f, 0x1f, 0x3f, 0x7f, 0xff];
|
||||
return function (n) {
|
||||
var result = 0;
|
||||
// State arrays initialized in header()
|
||||
private byteCount!: Int32Array;
|
||||
private symToByte!: Uint8Array;
|
||||
private mtfSymbol!: Int32Array;
|
||||
private selectors!: Uint8Array;
|
||||
|
||||
/**
|
||||
* Create a bit reader from a byte array
|
||||
*/
|
||||
array(bytes: Uint8Array | Buffer): (n: number) => number {
|
||||
let bit = 0;
|
||||
let byte = 0;
|
||||
const BITMASK = [0, 0x01, 0x03, 0x07, 0x0f, 0x1f, 0x3f, 0x7f, 0xff];
|
||||
|
||||
return function (n: number): number {
|
||||
let result = 0;
|
||||
while (n > 0) {
|
||||
var left = 8 - bit;
|
||||
const left = 8 - bit;
|
||||
if (n >= left) {
|
||||
result <<= left;
|
||||
result |= BITMASK[left] & bytes[byte++];
|
||||
@@ -78,234 +87,341 @@ export class Bzip2 {
|
||||
n -= left;
|
||||
} else {
|
||||
result <<= n;
|
||||
result |=
|
||||
(bytes[byte] & (BITMASK[n] << (8 - n - bit))) >> (8 - n - bit);
|
||||
result |= (bytes[byte] & (BITMASK[n] << (8 - n - bit))) >> (8 - n - bit);
|
||||
bit += n;
|
||||
n = 0;
|
||||
}
|
||||
}
|
||||
return result;
|
||||
};
|
||||
};
|
||||
}
|
||||
|
||||
simple = function (srcbuffer, stream) {
|
||||
var bits = this.array(srcbuffer);
|
||||
var size = this.header(bits);
|
||||
var ret = false;
|
||||
var bufsize = 100000 * size;
|
||||
var buf = new Int32Array(bufsize);
|
||||
/**
|
||||
* Simple decompression from a buffer
|
||||
*/
|
||||
simple(srcbuffer: Uint8Array | Buffer, stream: (byte: number) => void): void {
|
||||
const bits = this.array(srcbuffer);
|
||||
const size = this.header(bits as IBitReader);
|
||||
let ret: number | null = 0;
|
||||
const bufsize = 100000 * size;
|
||||
const buf = new Int32Array(bufsize);
|
||||
|
||||
do {
|
||||
ret = this.decompress(bits, stream, buf, bufsize);
|
||||
} while (!ret);
|
||||
};
|
||||
ret = this.decompress(bits as IBitReader, stream, buf, bufsize, ret);
|
||||
} while (ret !== null);
|
||||
}
|
||||
|
||||
header = function (bits) {
|
||||
/**
|
||||
* Parse BZIP2 header and return block size
|
||||
*/
|
||||
header(bits: IBitReader): number {
|
||||
this.byteCount = new Int32Array(256);
|
||||
this.symToByte = new Uint8Array(256);
|
||||
this.mtfSymbol = new Int32Array(256);
|
||||
this.selectors = new Uint8Array(0x8000);
|
||||
|
||||
if (bits(8 * 3) != 4348520) messageArg.Error('No magic number found');
|
||||
|
||||
var i = bits(8) - 48;
|
||||
if (i < 1 || i > 9) messageArg.Error('Not a BZIP archive');
|
||||
return i;
|
||||
};
|
||||
|
||||
decompress = function (bits, stream, buf, bufsize, streamCRC) {
|
||||
var MAX_HUFCODE_BITS = 20;
|
||||
var MAX_SYMBOLS = 258;
|
||||
var SYMBOL_RUNA = 0;
|
||||
var SYMBOL_RUNB = 1;
|
||||
var GROUP_SIZE = 50;
|
||||
var crc = 0 ^ -1;
|
||||
|
||||
for (var h = '', i = 0; i < 6; i++) h += bits(8).toString(16);
|
||||
if (h == '177245385090') {
|
||||
var finalCRC = bits(32) | 0;
|
||||
if (finalCRC !== streamCRC)
|
||||
messageArg.Error('Error in bzip2: crc32 do not match');
|
||||
// align stream to byte
|
||||
bits(null);
|
||||
return null; // reset streamCRC for next call
|
||||
if (bits(8 * 3) !== 4348520) {
|
||||
throwError('No BZIP2 magic number found at start of stream', BZIP2_ERROR_CODES.NO_MAGIC_NUMBER);
|
||||
}
|
||||
if (h != '314159265359') messageArg.Error('eek not valid bzip data');
|
||||
var crcblock = bits(32) | 0; // CRC code
|
||||
if (bits(1)) messageArg.Error('unsupported obsolete version');
|
||||
var origPtr = bits(24);
|
||||
if (origPtr > bufsize)
|
||||
messageArg.Error('Initial position larger than buffer size');
|
||||
var t = bits(16);
|
||||
var symTotal = 0;
|
||||
for (i = 0; i < 16; i++) {
|
||||
if (t & (1 << (15 - i))) {
|
||||
var k = bits(16);
|
||||
for (j = 0; j < 16; j++) {
|
||||
if (k & (1 << (15 - j))) {
|
||||
|
||||
const blockSize = (bits(8) as number) - 48;
|
||||
if (blockSize < 1 || blockSize > 9) {
|
||||
throwError('Invalid BZIP2 archive: block size must be 1-9', BZIP2_ERROR_CODES.INVALID_ARCHIVE);
|
||||
}
|
||||
return blockSize;
|
||||
}
|
||||
|
||||
/**
|
||||
* Decompress a BZIP2 block
|
||||
*/
|
||||
decompress(
|
||||
bits: IBitReader,
|
||||
stream: (byte: number) => void,
|
||||
buf: Int32Array,
|
||||
bufsize: number,
|
||||
streamCRC?: number | null
|
||||
): number | null {
|
||||
const MAX_HUFCODE_BITS = 20;
|
||||
const MAX_SYMBOLS = 258;
|
||||
const SYMBOL_RUNA = 0;
|
||||
const SYMBOL_RUNB = 1;
|
||||
const GROUP_SIZE = 50;
|
||||
let crc = 0 ^ -1;
|
||||
|
||||
// Read block header
|
||||
let headerHex = '';
|
||||
for (let i = 0; i < 6; i++) {
|
||||
headerHex += (bits(8) as number).toString(16);
|
||||
}
|
||||
|
||||
// Check for end-of-stream marker
|
||||
if (headerHex === '177245385090') {
|
||||
const finalCRC = bits(32) as number | 0;
|
||||
if (finalCRC !== streamCRC) {
|
||||
throwError('CRC32 mismatch: stream checksum verification failed', BZIP2_ERROR_CODES.CRC_MISMATCH);
|
||||
}
|
||||
// Align stream to byte boundary
|
||||
bits(null);
|
||||
return null;
|
||||
}
|
||||
|
||||
// Verify block signature (pi digits)
|
||||
if (headerHex !== '314159265359') {
|
||||
throwError('Invalid block header: expected pi signature (0x314159265359)', BZIP2_ERROR_CODES.INVALID_BLOCK_DATA);
|
||||
}
|
||||
|
||||
const crcblock = bits(32) as number | 0;
|
||||
|
||||
if (bits(1)) {
|
||||
throwError('Unsupported obsolete BZIP2 format version', BZIP2_ERROR_CODES.INVALID_ARCHIVE);
|
||||
}
|
||||
|
||||
const origPtr = bits(24) as number;
|
||||
if (origPtr > bufsize) {
|
||||
throwError('Initial position larger than buffer size', BZIP2_ERROR_CODES.BUFFER_OVERFLOW);
|
||||
}
|
||||
|
||||
// Read symbol map
|
||||
let symbolMapBits = bits(16) as number;
|
||||
let symTotal = 0;
|
||||
for (let i = 0; i < 16; i++) {
|
||||
if (symbolMapBits & (1 << (15 - i))) {
|
||||
const subMap = bits(16) as number;
|
||||
for (let j = 0; j < 16; j++) {
|
||||
if (subMap & (1 << (15 - j))) {
|
||||
this.symToByte[symTotal++] = 16 * i + j;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
var groupCount = bits(3);
|
||||
if (groupCount < 2 || groupCount > 6) messageArg.Error('another error');
|
||||
var nSelectors = bits(15);
|
||||
if (nSelectors == 0) messageArg.Error('meh');
|
||||
for (var i = 0; i < groupCount; i++) this.mtfSymbol[i] = i;
|
||||
// Read Huffman groups
|
||||
const groupCount = bits(3) as number;
|
||||
if (groupCount < 2 || groupCount > 6) {
|
||||
throwError('Invalid group count: must be between 2 and 6', BZIP2_ERROR_CODES.INVALID_HUFFMAN);
|
||||
}
|
||||
|
||||
for (var i = 0; i < nSelectors; i++) {
|
||||
for (var j = 0; bits(1); j++)
|
||||
if (j >= groupCount) messageArg.Error('whoops another error');
|
||||
var uc = this.mtfSymbol[j];
|
||||
for (var k: any = j - 1; k >= 0; k--) {
|
||||
const nSelectors = bits(15) as number;
|
||||
if (nSelectors === 0) {
|
||||
throwError('Invalid selector count: cannot be zero', BZIP2_ERROR_CODES.INVALID_SELECTOR);
|
||||
}
|
||||
|
||||
// Initialize MTF symbol array
|
||||
for (let i = 0; i < groupCount; i++) {
|
||||
this.mtfSymbol[i] = i;
|
||||
}
|
||||
|
||||
// Read selectors using MTF decoding
|
||||
for (let i = 0; i < nSelectors; i++) {
|
||||
let j = 0;
|
||||
while (bits(1)) {
|
||||
j++;
|
||||
if (j >= groupCount) {
|
||||
throwError('Invalid MTF index: exceeds group count', BZIP2_ERROR_CODES.INVALID_HUFFMAN);
|
||||
}
|
||||
}
|
||||
const uc = this.mtfSymbol[j];
|
||||
for (let k = j - 1; k >= 0; k--) {
|
||||
this.mtfSymbol[k + 1] = this.mtfSymbol[k];
|
||||
}
|
||||
this.mtfSymbol[0] = uc;
|
||||
this.selectors[i] = uc;
|
||||
}
|
||||
|
||||
var symCount = symTotal + 2;
|
||||
var groups = [];
|
||||
var length = new Uint8Array(MAX_SYMBOLS),
|
||||
temp = new Uint16Array(MAX_HUFCODE_BITS + 1);
|
||||
// Build Huffman tables
|
||||
const symCount = symTotal + 2;
|
||||
const groups: IHuffmanGroup[] = [];
|
||||
const length = new Uint8Array(MAX_SYMBOLS);
|
||||
const temp = new Uint16Array(MAX_HUFCODE_BITS + 1);
|
||||
|
||||
var hufGroup;
|
||||
|
||||
for (var j = 0; j < groupCount; j++) {
|
||||
t = bits(5); //lengths
|
||||
for (var i = 0; i < symCount; i++) {
|
||||
for (let j = 0; j < groupCount; j++) {
|
||||
let t = bits(5) as number;
|
||||
for (let i = 0; i < symCount; i++) {
|
||||
while (true) {
|
||||
if (t < 1 || t > MAX_HUFCODE_BITS)
|
||||
messageArg.Error('I gave up a while ago on writing error messages');
|
||||
if (t < 1 || t > MAX_HUFCODE_BITS) {
|
||||
throwError('Invalid Huffman code length: must be between 1 and 20', BZIP2_ERROR_CODES.INVALID_HUFFMAN);
|
||||
}
|
||||
if (!bits(1)) break;
|
||||
if (!bits(1)) t++;
|
||||
else t--;
|
||||
}
|
||||
length[i] = t;
|
||||
}
|
||||
var minLen, maxLen;
|
||||
minLen = maxLen = length[0];
|
||||
for (var i = 1; i < symCount; i++) {
|
||||
|
||||
let minLen = length[0];
|
||||
let maxLen = length[0];
|
||||
for (let i = 1; i < symCount; i++) {
|
||||
if (length[i] > maxLen) maxLen = length[i];
|
||||
else if (length[i] < minLen) minLen = length[i];
|
||||
}
|
||||
hufGroup = groups[j] = {};
|
||||
hufGroup.permute = new Int32Array(MAX_SYMBOLS);
|
||||
hufGroup.limit = new Int32Array(MAX_HUFCODE_BITS + 1);
|
||||
hufGroup.base = new Int32Array(MAX_HUFCODE_BITS + 1);
|
||||
|
||||
hufGroup.minLen = minLen;
|
||||
hufGroup.maxLen = maxLen;
|
||||
var base = hufGroup.base;
|
||||
var limit = hufGroup.limit;
|
||||
var pp = 0;
|
||||
for (var i: number = minLen; i <= maxLen; i++)
|
||||
for (var t: any = 0; t < symCount; t++)
|
||||
if (length[t] == i) hufGroup.permute[pp++] = t;
|
||||
for (i = minLen; i <= maxLen; i++) temp[i] = limit[i] = 0;
|
||||
for (i = 0; i < symCount; i++) temp[length[i]]++;
|
||||
pp = t = 0;
|
||||
for (i = minLen; i < maxLen; i++) {
|
||||
const hufGroup: IHuffmanGroup = {
|
||||
permute: new Int32Array(MAX_SYMBOLS),
|
||||
limit: new Int32Array(MAX_HUFCODE_BITS + 1),
|
||||
base: new Int32Array(MAX_HUFCODE_BITS + 1),
|
||||
minLen,
|
||||
maxLen,
|
||||
};
|
||||
groups[j] = hufGroup;
|
||||
|
||||
const base = hufGroup.base;
|
||||
const limit = hufGroup.limit;
|
||||
|
||||
let pp = 0;
|
||||
for (let i = minLen; i <= maxLen; i++) {
|
||||
for (let t = 0; t < symCount; t++) {
|
||||
if (length[t] === i) hufGroup.permute[pp++] = t;
|
||||
}
|
||||
}
|
||||
|
||||
for (let i = minLen; i <= maxLen; i++) {
|
||||
temp[i] = 0;
|
||||
limit[i] = 0;
|
||||
}
|
||||
for (let i = 0; i < symCount; i++) {
|
||||
temp[length[i]]++;
|
||||
}
|
||||
|
||||
pp = 0;
|
||||
let tt = 0;
|
||||
for (let i = minLen; i < maxLen; i++) {
|
||||
pp += temp[i];
|
||||
limit[i] = pp - 1;
|
||||
pp <<= 1;
|
||||
base[i + 1] = pp - (t += temp[i]);
|
||||
base[i + 1] = pp - (tt += temp[i]);
|
||||
}
|
||||
limit[maxLen] = pp + temp[maxLen] - 1;
|
||||
base[minLen] = 0;
|
||||
}
|
||||
|
||||
for (var i = 0; i < 256; i++) {
|
||||
// Initialize for decoding
|
||||
for (let i = 0; i < 256; i++) {
|
||||
this.mtfSymbol[i] = i;
|
||||
this.byteCount[i] = 0;
|
||||
}
|
||||
var runPos, count, symCount: number, selector;
|
||||
runPos = count = symCount = selector = 0;
|
||||
|
||||
let runPos = 0;
|
||||
let count = 0;
|
||||
let symCountRemaining = 0;
|
||||
let selector = 0;
|
||||
let hufGroup = groups[0];
|
||||
let base = hufGroup.base;
|
||||
let limit = hufGroup.limit;
|
||||
|
||||
// Main decoding loop
|
||||
while (true) {
|
||||
if (!symCount--) {
|
||||
symCount = GROUP_SIZE - 1;
|
||||
if (selector >= nSelectors)
|
||||
messageArg.Error("meow i'm a kitty, that's an error");
|
||||
if (!symCountRemaining--) {
|
||||
symCountRemaining = GROUP_SIZE - 1;
|
||||
if (selector >= nSelectors) {
|
||||
throwError('Invalid selector index: exceeds available groups', BZIP2_ERROR_CODES.INVALID_SELECTOR);
|
||||
}
|
||||
hufGroup = groups[this.selectors[selector++]];
|
||||
base = hufGroup.base;
|
||||
limit = hufGroup.limit;
|
||||
}
|
||||
i = hufGroup.minLen;
|
||||
j = bits(i);
|
||||
|
||||
let i = hufGroup.minLen;
|
||||
let j = bits(i) as number;
|
||||
|
||||
while (true) {
|
||||
if (i > hufGroup.maxLen) messageArg.Error("rawr i'm a dinosaur");
|
||||
if (i > hufGroup.maxLen) {
|
||||
throwError('Huffman decoding error: bit length exceeds maximum allowed', BZIP2_ERROR_CODES.INVALID_HUFFMAN);
|
||||
}
|
||||
if (j <= limit[i]) break;
|
||||
i++;
|
||||
j = (j << 1) | bits(1);
|
||||
j = (j << 1) | (bits(1) as number);
|
||||
}
|
||||
|
||||
j -= base[i];
|
||||
if (j < 0 || j >= MAX_SYMBOLS) messageArg.Error("moo i'm a cow");
|
||||
var nextSym = hufGroup.permute[j];
|
||||
if (nextSym == SYMBOL_RUNA || nextSym == SYMBOL_RUNB) {
|
||||
if (j < 0 || j >= MAX_SYMBOLS) {
|
||||
throwError('Symbol index out of bounds during Huffman decoding', BZIP2_ERROR_CODES.INVALID_HUFFMAN);
|
||||
}
|
||||
|
||||
const nextSym = hufGroup.permute[j];
|
||||
|
||||
if (nextSym === SYMBOL_RUNA || nextSym === SYMBOL_RUNB) {
|
||||
if (!runPos) {
|
||||
runPos = 1;
|
||||
t = 0;
|
||||
j = 0;
|
||||
}
|
||||
if (nextSym == SYMBOL_RUNA) t += runPos;
|
||||
else t += 2 * runPos;
|
||||
if (nextSym === SYMBOL_RUNA) j += runPos;
|
||||
else j += 2 * runPos;
|
||||
runPos <<= 1;
|
||||
continue;
|
||||
}
|
||||
|
||||
if (runPos) {
|
||||
runPos = 0;
|
||||
if (count + t > bufsize) messageArg.Error('Boom.');
|
||||
uc = this.symToByte[this.mtfSymbol[0]];
|
||||
this.byteCount[uc] += t;
|
||||
while (t--) buf[count++] = uc;
|
||||
const runLength = j;
|
||||
if (count + runLength > bufsize) {
|
||||
throwError('Run-length overflow: decoded run exceeds buffer capacity', BZIP2_ERROR_CODES.BUFFER_OVERFLOW);
|
||||
}
|
||||
const uc = this.symToByte[this.mtfSymbol[0]];
|
||||
this.byteCount[uc] += runLength;
|
||||
for (let t = 0; t < runLength; t++) {
|
||||
buf[count++] = uc;
|
||||
}
|
||||
}
|
||||
|
||||
if (nextSym > symTotal) break;
|
||||
if (count >= bufsize)
|
||||
messageArg.Error("I can't think of anything. Error");
|
||||
i = nextSym - 1;
|
||||
uc = this.mtfSymbol[i];
|
||||
for (var k: any = i - 1; k >= 0; k--) {
|
||||
|
||||
if (count >= bufsize) {
|
||||
throwError('Buffer overflow: decoded data exceeds buffer capacity', BZIP2_ERROR_CODES.BUFFER_OVERFLOW);
|
||||
}
|
||||
|
||||
const mtfIndex = nextSym - 1;
|
||||
const uc = this.mtfSymbol[mtfIndex];
|
||||
for (let k = mtfIndex - 1; k >= 0; k--) {
|
||||
this.mtfSymbol[k + 1] = this.mtfSymbol[k];
|
||||
}
|
||||
this.mtfSymbol[0] = uc;
|
||||
uc = this.symToByte[uc];
|
||||
this.byteCount[uc]++;
|
||||
buf[count++] = uc;
|
||||
const decodedByte = this.symToByte[uc];
|
||||
this.byteCount[decodedByte]++;
|
||||
buf[count++] = decodedByte;
|
||||
}
|
||||
if (origPtr < 0 || origPtr >= count)
|
||||
messageArg.Error(
|
||||
"I'm a monkey and I'm throwing something at someone, namely you",
|
||||
);
|
||||
var j = 0;
|
||||
for (var i = 0; i < 256; i++) {
|
||||
k = j + this.byteCount[i];
|
||||
|
||||
if (origPtr < 0 || origPtr >= count) {
|
||||
throwError('Invalid original pointer: position outside decoded block', BZIP2_ERROR_CODES.INVALID_POSITION);
|
||||
}
|
||||
|
||||
// Inverse BWT transform
|
||||
let j = 0;
|
||||
for (let i = 0; i < 256; i++) {
|
||||
const k = j + this.byteCount[i];
|
||||
this.byteCount[i] = j;
|
||||
j = k;
|
||||
}
|
||||
for (var i = 0; i < count; i++) {
|
||||
uc = buf[i] & 0xff;
|
||||
|
||||
for (let i = 0; i < count; i++) {
|
||||
const uc = buf[i] & 0xff;
|
||||
buf[this.byteCount[uc]] |= i << 8;
|
||||
this.byteCount[uc]++;
|
||||
}
|
||||
var pos = 0,
|
||||
current = 0,
|
||||
run = 0;
|
||||
|
||||
// Output decoded data
|
||||
let pos = 0;
|
||||
let current = 0;
|
||||
let run = 0;
|
||||
|
||||
if (count) {
|
||||
pos = buf[origPtr];
|
||||
current = pos & 0xff;
|
||||
pos >>= 8;
|
||||
run = -1;
|
||||
}
|
||||
count = count;
|
||||
var copies, previous, outbyte;
|
||||
while (count) {
|
||||
count--;
|
||||
previous = current;
|
||||
|
||||
let remaining = count;
|
||||
while (remaining) {
|
||||
remaining--;
|
||||
const previous = current;
|
||||
pos = buf[pos];
|
||||
current = pos & 0xff;
|
||||
pos >>= 8;
|
||||
if (run++ == 3) {
|
||||
|
||||
let copies: number;
|
||||
let outbyte: number;
|
||||
|
||||
if (run++ === 3) {
|
||||
copies = current;
|
||||
outbyte = previous;
|
||||
current = -1;
|
||||
@@ -313,19 +429,21 @@ export class Bzip2 {
|
||||
copies = 1;
|
||||
outbyte = current;
|
||||
}
|
||||
|
||||
while (copies--) {
|
||||
crc =
|
||||
((crc << 8) ^ this.crcTable[((crc >> 24) ^ outbyte) & 0xff]) &
|
||||
0xffffffff; // crc32
|
||||
crc = ((crc << 8) ^ this.crcTable[((crc >> 24) ^ outbyte) & 0xff]) & 0xffffffff;
|
||||
stream(outbyte);
|
||||
}
|
||||
if (current != previous) run = 0;
|
||||
|
||||
if (current !== previous) run = 0;
|
||||
}
|
||||
|
||||
crc = (crc ^ -1) >>> 0;
|
||||
if ((crc | 0) != (crcblock | 0))
|
||||
messageArg.Error('Error in bzip2: crc32 do not match');
|
||||
streamCRC = (crc ^ ((streamCRC << 1) | (streamCRC >>> 31))) & 0xffffffff;
|
||||
return streamCRC;
|
||||
};
|
||||
if ((crc | 0) !== (crcblock | 0)) {
|
||||
throwError('CRC32 mismatch: block checksum verification failed', BZIP2_ERROR_CODES.CRC_MISMATCH);
|
||||
}
|
||||
|
||||
const newStreamCRC = (crc ^ (((streamCRC || 0) << 1) | ((streamCRC || 0) >>> 31))) & 0xffffffff;
|
||||
return newStreamCRC;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,51 +1,53 @@
|
||||
import * as plugins from '../plugins.js';
|
||||
import { Bzip2Error, BZIP2_ERROR_CODES } from '../errors.js';
|
||||
import type { IBitReader } from '../interfaces.js';
|
||||
|
||||
import { Bzip2 } from './bzip2.js';
|
||||
import { bitIterator } from './bititerator.js';
|
||||
|
||||
export function unbzip2Stream() {
|
||||
/**
|
||||
* Creates a streaming BZIP2 decompression transform
|
||||
*/
|
||||
export function unbzip2Stream(): plugins.smartstream.SmartDuplex<Buffer, Buffer> {
|
||||
const bzip2Instance = new Bzip2();
|
||||
var bufferQueue = [];
|
||||
var hasBytes = 0;
|
||||
var blockSize = 0;
|
||||
var broken = false;
|
||||
var done = false;
|
||||
var bitReader = null;
|
||||
var streamCRC = null;
|
||||
const bufferQueue: Buffer[] = [];
|
||||
let hasBytes = 0;
|
||||
let blockSize = 0;
|
||||
let broken = false;
|
||||
let bitReader: IBitReader | null = null;
|
||||
let streamCRC: number | null = null;
|
||||
|
||||
function decompressBlock() {
|
||||
function decompressBlock(): Buffer | undefined {
|
||||
if (!blockSize) {
|
||||
blockSize = bzip2Instance.header(bitReader);
|
||||
blockSize = bzip2Instance.header(bitReader!);
|
||||
streamCRC = 0;
|
||||
} else {
|
||||
var bufsize = 100000 * blockSize;
|
||||
var buf = new Int32Array(bufsize);
|
||||
|
||||
var chunk = [];
|
||||
var f = function (b) {
|
||||
chunk.push(b);
|
||||
};
|
||||
|
||||
streamCRC = bzip2Instance.decompress(
|
||||
bitReader,
|
||||
f,
|
||||
buf,
|
||||
bufsize,
|
||||
streamCRC,
|
||||
);
|
||||
if (streamCRC === null) {
|
||||
// reset for next bzip2 header
|
||||
blockSize = 0;
|
||||
return;
|
||||
} else {
|
||||
return Buffer.from(chunk);
|
||||
}
|
||||
return undefined;
|
||||
}
|
||||
|
||||
const bufsize = 100000 * blockSize;
|
||||
const buf = new Int32Array(bufsize);
|
||||
const chunk: number[] = [];
|
||||
|
||||
const outputFunc = (b: number): void => {
|
||||
chunk.push(b);
|
||||
};
|
||||
|
||||
streamCRC = bzip2Instance.decompress(bitReader!, outputFunc, buf, bufsize, streamCRC);
|
||||
|
||||
if (streamCRC === null) {
|
||||
// Reset for next bzip2 header
|
||||
blockSize = 0;
|
||||
return undefined;
|
||||
}
|
||||
|
||||
return Buffer.from(chunk);
|
||||
}
|
||||
|
||||
var outlength = 0;
|
||||
const decompressAndPush = async () => {
|
||||
if (broken) return;
|
||||
let outlength = 0;
|
||||
|
||||
const decompressAndPush = async (): Promise<Buffer | undefined> => {
|
||||
if (broken) return undefined;
|
||||
|
||||
try {
|
||||
const resultChunk = decompressBlock();
|
||||
if (resultChunk) {
|
||||
@@ -53,40 +55,39 @@ export function unbzip2Stream() {
|
||||
}
|
||||
return resultChunk;
|
||||
} catch (e) {
|
||||
console.error(e);
|
||||
broken = true;
|
||||
if (e instanceof Error) {
|
||||
throw new Bzip2Error(`Decompression failed: ${e.message}`, BZIP2_ERROR_CODES.INVALID_BLOCK_DATA);
|
||||
}
|
||||
throw e;
|
||||
}
|
||||
};
|
||||
let counter = 0;
|
||||
return new plugins.smartstream.SmartDuplex({
|
||||
|
||||
return new plugins.smartstream.SmartDuplex<Buffer, Buffer>({
|
||||
objectMode: true,
|
||||
name: 'bzip2',
|
||||
debug: false,
|
||||
highWaterMark: 1,
|
||||
writeFunction: async function (data, streamTools) {
|
||||
// console.log(`got chunk ${counter++}`)
|
||||
bufferQueue.push(data);
|
||||
hasBytes += data.length;
|
||||
|
||||
if (bitReader === null) {
|
||||
bitReader = bitIterator(function () {
|
||||
return bufferQueue.shift();
|
||||
return bufferQueue.shift()!;
|
||||
});
|
||||
}
|
||||
while (
|
||||
!broken &&
|
||||
hasBytes - bitReader.bytesRead + 1 >= (25000 + 100000 * blockSize || 4)
|
||||
) {
|
||||
//console.error('decompressing with', hasBytes - bitReader.bytesRead + 1, 'bytes in buffer');
|
||||
|
||||
const threshold = 25000 + 100000 * blockSize || 4;
|
||||
while (!broken && hasBytes - bitReader.bytesRead + 1 >= threshold) {
|
||||
const result = await decompressAndPush();
|
||||
if (!result) {
|
||||
continue;
|
||||
}
|
||||
// console.log(result.toString());
|
||||
await streamTools.push(result);
|
||||
}
|
||||
return null;
|
||||
},
|
||||
finalFunction: async function (streamTools) {
|
||||
//console.error(x,'last compressing with', hasBytes, 'bytes in buffer');
|
||||
while (!broken && bitReader && hasBytes > bitReader.bytesRead) {
|
||||
const result = await decompressAndPush();
|
||||
if (!result) {
|
||||
@@ -94,10 +95,11 @@ export function unbzip2Stream() {
|
||||
}
|
||||
await streamTools.push(result);
|
||||
}
|
||||
if (!broken) {
|
||||
if (streamCRC !== null)
|
||||
this.emit('error', new Error('input stream ended prematurely'));
|
||||
|
||||
if (!broken && streamCRC !== null) {
|
||||
this.emit('error', new Bzip2Error('Input stream ended prematurely', BZIP2_ERROR_CODES.PREMATURE_END));
|
||||
}
|
||||
return null;
|
||||
},
|
||||
});
|
||||
}
|
||||
|
||||
@@ -1,24 +1,41 @@
|
||||
import type { SmartArchive } from './classes.smartarchive.js';
|
||||
import type { TSupportedMime } from './interfaces.js';
|
||||
import * as plugins from './plugins.js';
|
||||
|
||||
/**
|
||||
* Type for decompression streams
|
||||
*/
|
||||
export type TDecompressionStream =
|
||||
| plugins.stream.Transform
|
||||
| plugins.stream.Duplex
|
||||
| plugins.tarStream.Extract;
|
||||
|
||||
/**
|
||||
* Result of archive analysis
|
||||
*/
|
||||
export interface IAnalyzedResult {
|
||||
fileType: plugins.fileType.FileTypeResult;
|
||||
fileType: plugins.fileType.FileTypeResult | undefined;
|
||||
isArchive: boolean;
|
||||
resultStream: plugins.smartstream.SmartDuplex;
|
||||
decompressionStream:
|
||||
| plugins.stream.Transform
|
||||
| plugins.stream.Duplex
|
||||
| plugins.tarStream.Extract;
|
||||
resultStream: plugins.smartstream.SmartDuplex<Buffer, Buffer>;
|
||||
decompressionStream: TDecompressionStream;
|
||||
}
|
||||
|
||||
/**
|
||||
* Analyzes archive streams to detect format and provide decompression
|
||||
*/
|
||||
export class ArchiveAnalyzer {
|
||||
smartArchiveRef: SmartArchive;
|
||||
private smartArchiveRef: SmartArchive;
|
||||
|
||||
constructor(smartArchiveRefArg: SmartArchive) {
|
||||
this.smartArchiveRef = smartArchiveRefArg;
|
||||
}
|
||||
|
||||
private async mimeTypeIsArchive(mimeType: string): Promise<boolean> {
|
||||
/**
|
||||
* Check if a MIME type represents an archive format
|
||||
*/
|
||||
private async mimeTypeIsArchive(mimeType: string | undefined): Promise<boolean> {
|
||||
if (!mimeType) return false;
|
||||
|
||||
const archiveMimeTypes: Set<string> = new Set([
|
||||
'application/zip',
|
||||
'application/x-rar-compressed',
|
||||
@@ -26,50 +43,46 @@ export class ArchiveAnalyzer {
|
||||
'application/gzip',
|
||||
'application/x-7z-compressed',
|
||||
'application/x-bzip2',
|
||||
// Add other archive mime types here
|
||||
]);
|
||||
|
||||
return archiveMimeTypes.has(mimeType);
|
||||
}
|
||||
|
||||
private async getDecompressionStream(
|
||||
mimeTypeArg: plugins.fileType.FileTypeResult['mime'],
|
||||
): Promise<
|
||||
plugins.stream.Transform | plugins.stream.Duplex | plugins.tarStream.Extract
|
||||
> {
|
||||
/**
|
||||
* Get the appropriate decompression stream for a MIME type
|
||||
*/
|
||||
private async getDecompressionStream(mimeTypeArg: TSupportedMime): Promise<TDecompressionStream> {
|
||||
switch (mimeTypeArg) {
|
||||
case 'application/gzip':
|
||||
return this.smartArchiveRef.gzipTools.getDecompressionStream();
|
||||
case 'application/zip':
|
||||
return this.smartArchiveRef.zipTools.getDecompressionStream();
|
||||
case 'application/x-bzip2':
|
||||
return await this.smartArchiveRef.bzip2Tools.getDecompressionStream(); // replace with your own bzip2 decompression stream
|
||||
return this.smartArchiveRef.bzip2Tools.getDecompressionStream();
|
||||
case 'application/x-tar':
|
||||
return this.smartArchiveRef.tarTools.getDecompressionStream(); // replace with your own tar decompression stream
|
||||
return this.smartArchiveRef.tarTools.getDecompressionStream();
|
||||
default:
|
||||
// Handle unsupported formats or no decompression needed
|
||||
return plugins.smartstream.createPassThrough();
|
||||
}
|
||||
}
|
||||
|
||||
public getAnalyzedStream() {
|
||||
/**
|
||||
* Create an analyzed stream that detects archive type and provides decompression
|
||||
* Emits a single IAnalyzedResult object
|
||||
*/
|
||||
public getAnalyzedStream(): plugins.smartstream.SmartDuplex<Buffer, IAnalyzedResult> {
|
||||
let firstRun = true;
|
||||
const resultStream = plugins.smartstream.createPassThrough();
|
||||
const analyzerstream = new plugins.smartstream.SmartDuplex<
|
||||
Buffer,
|
||||
IAnalyzedResult
|
||||
>({
|
||||
|
||||
const analyzerstream = new plugins.smartstream.SmartDuplex<Buffer, IAnalyzedResult>({
|
||||
readableObjectMode: true,
|
||||
writeFunction: async (chunkArg: Buffer, streamtools) => {
|
||||
if (firstRun) {
|
||||
firstRun = false;
|
||||
const fileType = await plugins.fileType.fileTypeFromBuffer(chunkArg);
|
||||
const decompressionStream = await this.getDecompressionStream(
|
||||
fileType?.mime as any,
|
||||
);
|
||||
/**
|
||||
* analyzed stream emits once with this object
|
||||
*/
|
||||
const decompressionStream = await this.getDecompressionStream(fileType?.mime as TSupportedMime);
|
||||
|
||||
const result: IAnalyzedResult = {
|
||||
fileType,
|
||||
isArchive: await this.mimeTypeIsArchive(fileType?.mime),
|
||||
@@ -81,11 +94,12 @@ export class ArchiveAnalyzer {
|
||||
await resultStream.backpressuredPush(chunkArg);
|
||||
return null;
|
||||
},
|
||||
finalFunction: async (tools) => {
|
||||
finalFunction: async () => {
|
||||
resultStream.push(null);
|
||||
return null;
|
||||
},
|
||||
});
|
||||
|
||||
return analyzerstream;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,42 +1,19 @@
|
||||
import type { SmartArchive } from './classes.smartarchive.js';
|
||||
import * as plugins from './plugins.js';
|
||||
import type { TCompressionLevel } from './interfaces.js';
|
||||
|
||||
// This class wraps fflate's gunzip in a Node.js Transform stream
|
||||
export class CompressGunzipTransform extends plugins.stream.Transform {
|
||||
constructor() {
|
||||
/**
|
||||
* Transform stream for GZIP compression using fflate
|
||||
*/
|
||||
export class GzipCompressionTransform extends plugins.stream.Transform {
|
||||
private gzip: plugins.fflate.Gzip;
|
||||
|
||||
constructor(level: TCompressionLevel = 6) {
|
||||
super();
|
||||
}
|
||||
|
||||
_transform(
|
||||
chunk: Buffer,
|
||||
encoding: BufferEncoding,
|
||||
callback: plugins.stream.TransformCallback,
|
||||
) {
|
||||
plugins.fflate.gunzip(chunk, (err, decompressed) => {
|
||||
if (err) {
|
||||
callback(err);
|
||||
} else {
|
||||
this.push(decompressed);
|
||||
callback();
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
// DecompressGunzipTransform class that extends the Node.js Transform stream to
|
||||
// create a stream that decompresses GZip-compressed data using fflate's gunzip function
|
||||
export class DecompressGunzipTransform extends plugins.stream.Transform {
|
||||
private gunzip: any; // fflate.Gunzip instance
|
||||
|
||||
constructor() {
|
||||
super();
|
||||
|
||||
// Create a streaming Gunzip decompressor
|
||||
this.gunzip = new plugins.fflate.Gunzip((chunk, final) => {
|
||||
// Push decompressed chunks to the output stream
|
||||
// Create a streaming Gzip compressor
|
||||
this.gzip = new plugins.fflate.Gzip({ level }, (chunk, final) => {
|
||||
this.push(Buffer.from(chunk));
|
||||
if (final) {
|
||||
// Signal end of stream when decompression is complete
|
||||
this.push(null);
|
||||
}
|
||||
});
|
||||
@@ -45,20 +22,59 @@ export class DecompressGunzipTransform extends plugins.stream.Transform {
|
||||
_transform(
|
||||
chunk: Buffer,
|
||||
encoding: BufferEncoding,
|
||||
callback: plugins.stream.TransformCallback,
|
||||
) {
|
||||
callback: plugins.stream.TransformCallback
|
||||
): void {
|
||||
try {
|
||||
this.gzip.push(chunk, false);
|
||||
callback();
|
||||
} catch (err) {
|
||||
callback(err as Error);
|
||||
}
|
||||
}
|
||||
|
||||
_flush(callback: plugins.stream.TransformCallback): void {
|
||||
try {
|
||||
this.gzip.push(new Uint8Array(0), true);
|
||||
callback();
|
||||
} catch (err) {
|
||||
callback(err as Error);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Transform stream for GZIP decompression using fflate
|
||||
*/
|
||||
export class GzipDecompressionTransform extends plugins.stream.Transform {
|
||||
private gunzip: plugins.fflate.Gunzip;
|
||||
|
||||
constructor() {
|
||||
super();
|
||||
|
||||
// Create a streaming Gunzip decompressor
|
||||
this.gunzip = new plugins.fflate.Gunzip((chunk, final) => {
|
||||
this.push(Buffer.from(chunk));
|
||||
if (final) {
|
||||
this.push(null);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
_transform(
|
||||
chunk: Buffer,
|
||||
encoding: BufferEncoding,
|
||||
callback: plugins.stream.TransformCallback
|
||||
): void {
|
||||
try {
|
||||
// Feed chunks to the gunzip stream
|
||||
this.gunzip.push(chunk, false);
|
||||
callback();
|
||||
} catch (err) {
|
||||
callback(err as Error);
|
||||
}
|
||||
}
|
||||
|
||||
_flush(callback: plugins.stream.TransformCallback) {
|
||||
|
||||
_flush(callback: plugins.stream.TransformCallback): void {
|
||||
try {
|
||||
// Signal end of input to gunzip
|
||||
this.gunzip.push(new Uint8Array(0), true);
|
||||
callback();
|
||||
} catch (err) {
|
||||
@@ -67,14 +83,61 @@ export class DecompressGunzipTransform extends plugins.stream.Transform {
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* GZIP compression and decompression utilities
|
||||
*/
|
||||
export class GzipTools {
|
||||
constructor() {}
|
||||
|
||||
public getCompressionStream() {
|
||||
return new CompressGunzipTransform();
|
||||
/**
|
||||
* Get a streaming compression transform
|
||||
*/
|
||||
public getCompressionStream(level?: TCompressionLevel): plugins.stream.Transform {
|
||||
return new GzipCompressionTransform(level);
|
||||
}
|
||||
|
||||
public getDecompressionStream() {
|
||||
return new DecompressGunzipTransform();
|
||||
/**
|
||||
* Get a streaming decompression transform
|
||||
*/
|
||||
public getDecompressionStream(): plugins.stream.Transform {
|
||||
return new GzipDecompressionTransform();
|
||||
}
|
||||
|
||||
/**
|
||||
* Compress data synchronously
|
||||
*/
|
||||
public compressSync(data: Buffer, level?: TCompressionLevel): Buffer {
|
||||
const options = level !== undefined ? { level } : undefined;
|
||||
return Buffer.from(plugins.fflate.gzipSync(data, options));
|
||||
}
|
||||
|
||||
/**
|
||||
* Decompress data synchronously
|
||||
*/
|
||||
public decompressSync(data: Buffer): Buffer {
|
||||
return Buffer.from(plugins.fflate.gunzipSync(data));
|
||||
}
|
||||
|
||||
/**
|
||||
* Compress data asynchronously
|
||||
*/
|
||||
public async compress(data: Buffer, level?: TCompressionLevel): Promise<Buffer> {
|
||||
return new Promise((resolve, reject) => {
|
||||
const options = level !== undefined ? { level } : undefined;
|
||||
plugins.fflate.gzip(data, options as plugins.fflate.AsyncGzipOptions, (err, result) => {
|
||||
if (err) reject(err);
|
||||
else resolve(Buffer.from(result));
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Decompress data asynchronously
|
||||
*/
|
||||
public async decompress(data: Buffer): Promise<Buffer> {
|
||||
return new Promise((resolve, reject) => {
|
||||
plugins.fflate.gunzip(data, (err, result) => {
|
||||
if (err) reject(err);
|
||||
else resolve(Buffer.from(result));
|
||||
});
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,75 +1,267 @@
|
||||
import * as plugins from './plugins.js';
|
||||
import * as paths from './paths.js';
|
||||
import type {
|
||||
IArchiveCreationOptions,
|
||||
IArchiveEntry,
|
||||
IArchiveExtractionOptions,
|
||||
IArchiveEntryInfo,
|
||||
IArchiveInfo,
|
||||
TArchiveFormat,
|
||||
TCompressionLevel,
|
||||
} from './interfaces.js';
|
||||
|
||||
import { Bzip2Tools } from './classes.bzip2tools.js';
|
||||
import { GzipTools } from './classes.gziptools.js';
|
||||
import { TarTools } from './classes.tartools.js';
|
||||
import { ZipTools } from './classes.ziptools.js';
|
||||
import { ArchiveAnalyzer, type IAnalyzedResult } from './classes.archiveanalyzer.js';
|
||||
|
||||
import {
|
||||
ArchiveAnalyzer,
|
||||
type IAnalyzedResult,
|
||||
} from './classes.archiveanalyzer.js';
|
||||
|
||||
import type { from } from '@push.rocks/smartrx/dist_ts/smartrx.plugins.rxjs.js';
|
||||
|
||||
/**
|
||||
* Main class for archive manipulation
|
||||
* Supports TAR, ZIP, GZIP, and BZIP2 formats
|
||||
*/
|
||||
export class SmartArchive {
|
||||
// STATIC
|
||||
public static async fromArchiveUrl(urlArg: string): Promise<SmartArchive> {
|
||||
// ============================================
|
||||
// STATIC FACTORY METHODS - EXTRACTION
|
||||
// ============================================
|
||||
|
||||
/**
|
||||
* Create SmartArchive from a URL
|
||||
*/
|
||||
public static async fromUrl(urlArg: string): Promise<SmartArchive> {
|
||||
const smartArchiveInstance = new SmartArchive();
|
||||
smartArchiveInstance.sourceUrl = urlArg;
|
||||
return smartArchiveInstance;
|
||||
}
|
||||
|
||||
public static async fromArchiveFile(
|
||||
filePathArg: string,
|
||||
): Promise<SmartArchive> {
|
||||
/**
|
||||
* Create SmartArchive from a local file path
|
||||
*/
|
||||
public static async fromFile(filePathArg: string): Promise<SmartArchive> {
|
||||
const smartArchiveInstance = new SmartArchive();
|
||||
smartArchiveInstance.sourceFilePath = filePathArg;
|
||||
return smartArchiveInstance;
|
||||
}
|
||||
|
||||
public static async fromArchiveStream(
|
||||
streamArg:
|
||||
| plugins.stream.Readable
|
||||
| plugins.stream.Duplex
|
||||
| plugins.stream.Transform,
|
||||
/**
|
||||
* Create SmartArchive from a readable stream
|
||||
*/
|
||||
public static async fromStream(
|
||||
streamArg: plugins.stream.Readable | plugins.stream.Duplex | plugins.stream.Transform
|
||||
): Promise<SmartArchive> {
|
||||
const smartArchiveInstance = new SmartArchive();
|
||||
smartArchiveInstance.sourceStream = streamArg;
|
||||
return smartArchiveInstance;
|
||||
}
|
||||
|
||||
// INSTANCE
|
||||
/**
|
||||
* Create SmartArchive from an in-memory buffer
|
||||
*/
|
||||
public static async fromBuffer(buffer: Buffer): Promise<SmartArchive> {
|
||||
const smartArchiveInstance = new SmartArchive();
|
||||
smartArchiveInstance.sourceStream = plugins.stream.Readable.from(buffer);
|
||||
return smartArchiveInstance;
|
||||
}
|
||||
|
||||
// ============================================
|
||||
// STATIC FACTORY METHODS - CREATION
|
||||
// ============================================
|
||||
|
||||
/**
|
||||
* Create a new archive from a directory
|
||||
*/
|
||||
public static async fromDirectory(
|
||||
directoryPath: string,
|
||||
options: IArchiveCreationOptions
|
||||
): Promise<SmartArchive> {
|
||||
const smartArchiveInstance = new SmartArchive();
|
||||
smartArchiveInstance.creationOptions = options;
|
||||
|
||||
const tarTools = new TarTools();
|
||||
|
||||
if (options.format === 'tar' || options.format === 'tar.gz' || options.format === 'tgz') {
|
||||
if (options.format === 'tar') {
|
||||
const pack = await tarTools.packDirectory(directoryPath);
|
||||
pack.finalize();
|
||||
smartArchiveInstance.archiveBuffer = await SmartArchive.streamToBuffer(pack);
|
||||
} else {
|
||||
smartArchiveInstance.archiveBuffer = await tarTools.packDirectoryToTarGz(
|
||||
directoryPath,
|
||||
options.compressionLevel
|
||||
);
|
||||
}
|
||||
} else if (options.format === 'zip') {
|
||||
const zipTools = new ZipTools();
|
||||
const fileTree = await plugins.listFileTree(directoryPath, '**/*');
|
||||
const entries: IArchiveEntry[] = [];
|
||||
|
||||
for (const filePath of fileTree) {
|
||||
const absolutePath = plugins.path.join(directoryPath, filePath);
|
||||
const content = await plugins.fsPromises.readFile(absolutePath);
|
||||
entries.push({
|
||||
archivePath: filePath,
|
||||
content,
|
||||
});
|
||||
}
|
||||
|
||||
smartArchiveInstance.archiveBuffer = await zipTools.createZip(entries, options.compressionLevel);
|
||||
} else {
|
||||
throw new Error(`Unsupported format for directory packing: ${options.format}`);
|
||||
}
|
||||
|
||||
return smartArchiveInstance;
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new archive from an array of entries
|
||||
*/
|
||||
public static async fromFiles(
|
||||
files: IArchiveEntry[],
|
||||
options: IArchiveCreationOptions
|
||||
): Promise<SmartArchive> {
|
||||
const smartArchiveInstance = new SmartArchive();
|
||||
smartArchiveInstance.creationOptions = options;
|
||||
|
||||
if (options.format === 'tar' || options.format === 'tar.gz' || options.format === 'tgz') {
|
||||
const tarTools = new TarTools();
|
||||
if (options.format === 'tar') {
|
||||
smartArchiveInstance.archiveBuffer = await tarTools.packFiles(files);
|
||||
} else {
|
||||
smartArchiveInstance.archiveBuffer = await tarTools.packFilesToTarGz(files, options.compressionLevel);
|
||||
}
|
||||
} else if (options.format === 'zip') {
|
||||
const zipTools = new ZipTools();
|
||||
smartArchiveInstance.archiveBuffer = await zipTools.createZip(files, options.compressionLevel);
|
||||
} else if (options.format === 'gz') {
|
||||
if (files.length !== 1) {
|
||||
throw new Error('GZIP format only supports a single file');
|
||||
}
|
||||
const gzipTools = new GzipTools();
|
||||
let content: Buffer;
|
||||
if (typeof files[0].content === 'string') {
|
||||
content = Buffer.from(files[0].content);
|
||||
} else if (Buffer.isBuffer(files[0].content)) {
|
||||
content = files[0].content;
|
||||
} else {
|
||||
throw new Error('GZIP format requires string or Buffer content');
|
||||
}
|
||||
smartArchiveInstance.archiveBuffer = await gzipTools.compress(content, options.compressionLevel);
|
||||
} else {
|
||||
throw new Error(`Unsupported format: ${options.format}`);
|
||||
}
|
||||
|
||||
return smartArchiveInstance;
|
||||
}
|
||||
|
||||
/**
|
||||
* Start building an archive incrementally using a builder pattern
|
||||
*/
|
||||
public static create(options: IArchiveCreationOptions): SmartArchive {
|
||||
const smartArchiveInstance = new SmartArchive();
|
||||
smartArchiveInstance.creationOptions = options;
|
||||
smartArchiveInstance.pendingEntries = [];
|
||||
return smartArchiveInstance;
|
||||
}
|
||||
|
||||
/**
|
||||
* Helper to convert a stream to buffer
|
||||
*/
|
||||
private static async streamToBuffer(stream: plugins.stream.Readable): Promise<Buffer> {
|
||||
const chunks: Buffer[] = [];
|
||||
return new Promise((resolve, reject) => {
|
||||
stream.on('data', (chunk) => chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk)));
|
||||
stream.on('end', () => resolve(Buffer.concat(chunks)));
|
||||
stream.on('error', reject);
|
||||
});
|
||||
}
|
||||
|
||||
// ============================================
|
||||
// INSTANCE PROPERTIES
|
||||
// ============================================
|
||||
|
||||
public tarTools = new TarTools();
|
||||
public zipTools = new ZipTools();
|
||||
public gzipTools = new GzipTools();
|
||||
public bzip2Tools = new Bzip2Tools(this);
|
||||
public archiveAnalyzer = new ArchiveAnalyzer(this);
|
||||
|
||||
public sourceUrl: string;
|
||||
public sourceFilePath: string;
|
||||
public sourceStream:
|
||||
| plugins.stream.Readable
|
||||
| plugins.stream.Duplex
|
||||
| plugins.stream.Transform;
|
||||
public sourceUrl?: string;
|
||||
public sourceFilePath?: string;
|
||||
public sourceStream?: plugins.stream.Readable | plugins.stream.Duplex | plugins.stream.Transform;
|
||||
|
||||
public archiveName: string;
|
||||
public singleFileMode: boolean = false;
|
||||
|
||||
public addedDirectories: string[] = [];
|
||||
public addedFiles: (
|
||||
| plugins.smartfile.SmartFile
|
||||
| plugins.smartfile.StreamFile
|
||||
)[] = [];
|
||||
public addedUrls: string[] = [];
|
||||
private archiveBuffer?: Buffer;
|
||||
private creationOptions?: IArchiveCreationOptions;
|
||||
private pendingEntries?: IArchiveEntry[];
|
||||
|
||||
constructor() {}
|
||||
|
||||
// ============================================
|
||||
// BUILDER METHODS (for incremental creation)
|
||||
// ============================================
|
||||
|
||||
/**
|
||||
* gets the original archive stream
|
||||
* Add a file to the archive (builder pattern)
|
||||
*/
|
||||
public async getArchiveStream() {
|
||||
public addFile(archivePath: string, content: string | Buffer): this {
|
||||
if (!this.pendingEntries) {
|
||||
throw new Error('addFile can only be called on archives created with SmartArchive.create()');
|
||||
}
|
||||
this.pendingEntries.push({ archivePath, content });
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* Add a SmartFile to the archive (builder pattern)
|
||||
*/
|
||||
public addSmartFile(file: plugins.smartfile.SmartFile, archivePath?: string): this {
|
||||
if (!this.pendingEntries) {
|
||||
throw new Error('addSmartFile can only be called on archives created with SmartArchive.create()');
|
||||
}
|
||||
this.pendingEntries.push({
|
||||
archivePath: archivePath || file.relative,
|
||||
content: file,
|
||||
});
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* Add a StreamFile to the archive (builder pattern)
|
||||
*/
|
||||
public addStreamFile(file: plugins.smartfile.StreamFile, archivePath?: string): this {
|
||||
if (!this.pendingEntries) {
|
||||
throw new Error('addStreamFile can only be called on archives created with SmartArchive.create()');
|
||||
}
|
||||
this.pendingEntries.push({
|
||||
archivePath: archivePath || file.relativeFilePath,
|
||||
content: file,
|
||||
});
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* Build the archive from pending entries
|
||||
*/
|
||||
public async build(): Promise<SmartArchive> {
|
||||
if (!this.pendingEntries || !this.creationOptions) {
|
||||
throw new Error('build can only be called on archives created with SmartArchive.create()');
|
||||
}
|
||||
|
||||
const built = await SmartArchive.fromFiles(this.pendingEntries, this.creationOptions);
|
||||
this.archiveBuffer = built.archiveBuffer;
|
||||
this.pendingEntries = undefined;
|
||||
return this;
|
||||
}
|
||||
|
||||
// ============================================
|
||||
// EXTRACTION METHODS
|
||||
// ============================================
|
||||
|
||||
/**
|
||||
* Get the original archive stream
|
||||
*/
|
||||
public async toStream(): Promise<plugins.stream.Readable> {
|
||||
if (this.archiveBuffer) {
|
||||
return plugins.stream.Readable.from(this.archiveBuffer);
|
||||
}
|
||||
if (this.sourceStream) {
|
||||
return this.sourceStream;
|
||||
}
|
||||
@@ -78,161 +270,320 @@ export class SmartArchive {
|
||||
.url(this.sourceUrl)
|
||||
.get();
|
||||
const webStream = response.stream();
|
||||
// @ts-ignore - Web stream to Node.js stream conversion
|
||||
const urlStream = plugins.stream.Readable.fromWeb(webStream);
|
||||
return urlStream;
|
||||
return plugins.stream.Readable.fromWeb(webStream as any);
|
||||
}
|
||||
if (this.sourceFilePath) {
|
||||
const fileStream = plugins.fs.createReadStream(this.sourceFilePath);
|
||||
return fileStream;
|
||||
return plugins.fs.createReadStream(this.sourceFilePath);
|
||||
}
|
||||
throw new Error('No archive source configured');
|
||||
}
|
||||
|
||||
public async exportToTarGzStream() {
|
||||
const tarPackStream = await this.tarTools.getPackStream();
|
||||
const gzipStream = await this.gzipTools.getCompressionStream();
|
||||
// const archiveStream = tarPackStream.pipe(gzipStream);
|
||||
// return archiveStream;
|
||||
/**
|
||||
* Get archive as a Buffer
|
||||
*/
|
||||
public async toBuffer(): Promise<Buffer> {
|
||||
if (this.archiveBuffer) {
|
||||
return this.archiveBuffer;
|
||||
}
|
||||
const stream = await this.toStream();
|
||||
return SmartArchive.streamToBuffer(stream);
|
||||
}
|
||||
|
||||
public async exportToFs(
|
||||
/**
|
||||
* Write archive to a file
|
||||
*/
|
||||
public async toFile(filePath: string): Promise<void> {
|
||||
const buffer = await this.toBuffer();
|
||||
await plugins.fsPromises.mkdir(plugins.path.dirname(filePath), { recursive: true });
|
||||
await plugins.fsPromises.writeFile(filePath, buffer);
|
||||
}
|
||||
|
||||
/**
|
||||
* Extract archive to filesystem
|
||||
*/
|
||||
public async extractToDirectory(
|
||||
targetDir: string,
|
||||
fileNameArg?: string,
|
||||
options?: Partial<IArchiveExtractionOptions>
|
||||
): Promise<void> {
|
||||
const done = plugins.smartpromise.defer<void>();
|
||||
const streamFileStream = await this.exportToStreamOfStreamFiles();
|
||||
const streamFileStream = await this.extractToStream();
|
||||
|
||||
streamFileStream.pipe(
|
||||
new plugins.smartstream.SmartDuplex({
|
||||
objectMode: true,
|
||||
writeFunction: async (
|
||||
streamFileArg: plugins.smartfile.StreamFile,
|
||||
streamtools,
|
||||
) => {
|
||||
const done = plugins.smartpromise.defer<void>();
|
||||
console.log(
|
||||
streamFileArg.relativeFilePath
|
||||
? streamFileArg.relativeFilePath
|
||||
: 'no relative path',
|
||||
);
|
||||
writeFunction: async (streamFileArg: plugins.smartfile.StreamFile) => {
|
||||
const innerDone = plugins.smartpromise.defer<void>();
|
||||
const streamFile = streamFileArg;
|
||||
let relativePath = streamFile.relativeFilePath || options?.fileName || 'extracted_file';
|
||||
|
||||
// Apply stripComponents if specified
|
||||
if (options?.stripComponents && options.stripComponents > 0) {
|
||||
const parts = relativePath.split('/');
|
||||
relativePath = parts.slice(options.stripComponents).join('/');
|
||||
if (!relativePath) {
|
||||
innerDone.resolve();
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
// Apply filter if specified
|
||||
if (options?.filter) {
|
||||
const entryInfo: IArchiveEntryInfo = {
|
||||
path: relativePath,
|
||||
size: 0,
|
||||
isDirectory: false,
|
||||
isFile: true,
|
||||
};
|
||||
if (!options.filter(entryInfo)) {
|
||||
innerDone.resolve();
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
const readStream = await streamFile.createReadStream();
|
||||
await plugins.fsPromises.mkdir(targetDir, { recursive: true });
|
||||
const writePath = plugins.path.join(
|
||||
targetDir,
|
||||
streamFile.relativeFilePath || fileNameArg,
|
||||
);
|
||||
const writePath = plugins.path.join(targetDir, relativePath);
|
||||
await plugins.fsPromises.mkdir(plugins.path.dirname(writePath), { recursive: true });
|
||||
const writeStream = plugins.fs.createWriteStream(writePath);
|
||||
readStream.pipe(writeStream);
|
||||
writeStream.on('finish', () => {
|
||||
done.resolve();
|
||||
innerDone.resolve();
|
||||
});
|
||||
await done.promise;
|
||||
await innerDone.promise;
|
||||
},
|
||||
finalFunction: async () => {
|
||||
done.resolve();
|
||||
},
|
||||
}),
|
||||
})
|
||||
);
|
||||
|
||||
return done.promise;
|
||||
}
|
||||
|
||||
public async exportToStreamOfStreamFiles() {
|
||||
const streamFileIntake =
|
||||
new plugins.smartstream.StreamIntake<plugins.smartfile.StreamFile>({
|
||||
objectMode: true,
|
||||
});
|
||||
const archiveStream = await this.getArchiveStream();
|
||||
/**
|
||||
* Extract archive to a stream of StreamFile objects
|
||||
*/
|
||||
public async extractToStream(): Promise<plugins.smartstream.StreamIntake<plugins.smartfile.StreamFile>> {
|
||||
const streamFileIntake = new plugins.smartstream.StreamIntake<plugins.smartfile.StreamFile>({
|
||||
objectMode: true,
|
||||
});
|
||||
|
||||
// Guard to prevent multiple signalEnd calls
|
||||
let hasSignaledEnd = false;
|
||||
const safeSignalEnd = () => {
|
||||
if (!hasSignaledEnd) {
|
||||
hasSignaledEnd = true;
|
||||
streamFileIntake.signalEnd();
|
||||
}
|
||||
};
|
||||
|
||||
const archiveStream = await this.toStream();
|
||||
const createAnalyzedStream = () => this.archiveAnalyzer.getAnalyzedStream();
|
||||
|
||||
// lets create a function that can be called multiple times to unpack layers of archives
|
||||
const createUnpackStream = () =>
|
||||
plugins.smartstream.createTransformFunction<IAnalyzedResult, any>(
|
||||
plugins.smartstream.createTransformFunction<IAnalyzedResult, void>(
|
||||
async (analyzedResultChunk) => {
|
||||
if (analyzedResultChunk.fileType?.mime === 'application/x-tar') {
|
||||
const tarStream =
|
||||
analyzedResultChunk.decompressionStream as plugins.tarStream.Extract;
|
||||
const tarStream = analyzedResultChunk.decompressionStream as plugins.tarStream.Extract;
|
||||
|
||||
tarStream.on('entry', async (header, stream, next) => {
|
||||
if (header.type === 'directory') {
|
||||
console.log(
|
||||
`tar stream directory: ${header.name} ... skipping!`,
|
||||
);
|
||||
stream.resume(); // Consume directory stream
|
||||
stream.resume();
|
||||
stream.on('end', () => next());
|
||||
return;
|
||||
}
|
||||
console.log(`tar stream file: ${header.name}`);
|
||||
|
||||
// Create a PassThrough stream to buffer the data
|
||||
|
||||
const passThrough = new plugins.stream.PassThrough();
|
||||
const streamfile = plugins.smartfile.StreamFile.fromStream(
|
||||
passThrough,
|
||||
header.name,
|
||||
);
|
||||
|
||||
// Push the streamfile immediately
|
||||
const streamfile = plugins.smartfile.StreamFile.fromStream(passThrough, header.name);
|
||||
streamFileIntake.push(streamfile);
|
||||
|
||||
// Pipe the tar entry stream to the passthrough
|
||||
stream.pipe(passThrough);
|
||||
|
||||
// Move to next entry when this one ends
|
||||
stream.on('end', () => {
|
||||
passThrough.end();
|
||||
next();
|
||||
});
|
||||
});
|
||||
tarStream.on('finish', function () {
|
||||
console.log('tar extraction finished');
|
||||
// Only signal end if this is the final stream
|
||||
streamFileIntake.signalEnd();
|
||||
|
||||
tarStream.on('finish', () => {
|
||||
safeSignalEnd();
|
||||
});
|
||||
analyzedResultChunk.resultStream.pipe(
|
||||
analyzedResultChunk.decompressionStream,
|
||||
);
|
||||
|
||||
analyzedResultChunk.resultStream.pipe(analyzedResultChunk.decompressionStream);
|
||||
} else if (analyzedResultChunk.fileType?.mime === 'application/zip') {
|
||||
analyzedResultChunk.resultStream
|
||||
.pipe(analyzedResultChunk.decompressionStream)
|
||||
.pipe(
|
||||
new plugins.smartstream.SmartDuplex({
|
||||
objectMode: true,
|
||||
writeFunction: async (
|
||||
streamFileArg: plugins.smartfile.StreamFile,
|
||||
streamtools,
|
||||
) => {
|
||||
writeFunction: async (streamFileArg: plugins.smartfile.StreamFile) => {
|
||||
streamFileIntake.push(streamFileArg);
|
||||
},
|
||||
finalFunction: async () => {
|
||||
streamFileIntake.signalEnd();
|
||||
safeSignalEnd();
|
||||
},
|
||||
}),
|
||||
})
|
||||
);
|
||||
} else if (
|
||||
analyzedResultChunk.isArchive &&
|
||||
analyzedResultChunk.decompressionStream
|
||||
) {
|
||||
} else if (analyzedResultChunk.isArchive && analyzedResultChunk.decompressionStream) {
|
||||
// For nested archives (like gzip containing tar)
|
||||
const nestedStream = analyzedResultChunk.resultStream
|
||||
analyzedResultChunk.resultStream
|
||||
.pipe(analyzedResultChunk.decompressionStream)
|
||||
.pipe(createAnalyzedStream())
|
||||
.pipe(createUnpackStream());
|
||||
|
||||
// Don't signal end here - let the nested unpacker handle it
|
||||
} else {
|
||||
const streamFile = plugins.smartfile.StreamFile.fromStream(
|
||||
analyzedResultChunk.resultStream,
|
||||
analyzedResultChunk.fileType?.ext,
|
||||
analyzedResultChunk.fileType?.ext
|
||||
);
|
||||
streamFileIntake.push(streamFile);
|
||||
streamFileIntake.signalEnd();
|
||||
safeSignalEnd();
|
||||
}
|
||||
},
|
||||
{
|
||||
objectMode: true,
|
||||
},
|
||||
{ objectMode: true }
|
||||
);
|
||||
|
||||
archiveStream.pipe(createAnalyzedStream()).pipe(createUnpackStream());
|
||||
return streamFileIntake;
|
||||
}
|
||||
|
||||
/**
|
||||
* Extract archive to an array of SmartFile objects (in-memory)
|
||||
*/
|
||||
public async extractToSmartFiles(): Promise<plugins.smartfile.SmartFile[]> {
|
||||
const streamFiles = await this.extractToStream();
|
||||
const smartFiles: plugins.smartfile.SmartFile[] = [];
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
streamFiles.on('data', async (streamFile: plugins.smartfile.StreamFile) => {
|
||||
try {
|
||||
const smartFile = await streamFile.toSmartFile();
|
||||
smartFiles.push(smartFile);
|
||||
} catch (err) {
|
||||
reject(err);
|
||||
}
|
||||
});
|
||||
streamFiles.on('end', () => resolve(smartFiles));
|
||||
streamFiles.on('error', reject);
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Extract a single file from the archive by path
|
||||
*/
|
||||
public async extractFile(filePath: string): Promise<plugins.smartfile.SmartFile | null> {
|
||||
const streamFiles = await this.extractToStream();
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
let found = false;
|
||||
|
||||
streamFiles.on('data', async (streamFile: plugins.smartfile.StreamFile) => {
|
||||
if (streamFile.relativeFilePath === filePath || streamFile.relativeFilePath?.endsWith(filePath)) {
|
||||
found = true;
|
||||
try {
|
||||
const smartFile = await streamFile.toSmartFile();
|
||||
resolve(smartFile);
|
||||
} catch (err) {
|
||||
reject(err);
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
streamFiles.on('end', () => {
|
||||
if (!found) {
|
||||
resolve(null);
|
||||
}
|
||||
});
|
||||
|
||||
streamFiles.on('error', reject);
|
||||
});
|
||||
}
|
||||
|
||||
// ============================================
|
||||
// ANALYSIS METHODS
|
||||
// ============================================
|
||||
|
||||
/**
|
||||
* Analyze the archive and return metadata
|
||||
*/
|
||||
public async analyze(): Promise<IArchiveInfo> {
|
||||
const stream = await this.toStream();
|
||||
const firstChunk = await this.readFirstChunk(stream);
|
||||
const fileType = await plugins.fileType.fileTypeFromBuffer(firstChunk);
|
||||
|
||||
let format: TArchiveFormat | null = null;
|
||||
let isCompressed = false;
|
||||
let isArchive = false;
|
||||
|
||||
if (fileType) {
|
||||
switch (fileType.mime) {
|
||||
case 'application/gzip':
|
||||
format = 'gz';
|
||||
isCompressed = true;
|
||||
isArchive = true;
|
||||
break;
|
||||
case 'application/zip':
|
||||
format = 'zip';
|
||||
isCompressed = true;
|
||||
isArchive = true;
|
||||
break;
|
||||
case 'application/x-tar':
|
||||
format = 'tar';
|
||||
isArchive = true;
|
||||
break;
|
||||
case 'application/x-bzip2':
|
||||
format = 'bz2';
|
||||
isCompressed = true;
|
||||
isArchive = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
return {
|
||||
format,
|
||||
isCompressed,
|
||||
isArchive,
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* List all entries in the archive without extracting
|
||||
*/
|
||||
public async listEntries(): Promise<IArchiveEntryInfo[]> {
|
||||
const entries: IArchiveEntryInfo[] = [];
|
||||
const streamFiles = await this.extractToStream();
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
streamFiles.on('data', (streamFile: plugins.smartfile.StreamFile) => {
|
||||
entries.push({
|
||||
path: streamFile.relativeFilePath || 'unknown',
|
||||
size: 0, // Size not available without reading
|
||||
isDirectory: false,
|
||||
isFile: true,
|
||||
});
|
||||
});
|
||||
streamFiles.on('end', () => resolve(entries));
|
||||
streamFiles.on('error', reject);
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Check if a specific file exists in the archive
|
||||
*/
|
||||
public async hasFile(filePath: string): Promise<boolean> {
|
||||
const entries = await this.listEntries();
|
||||
return entries.some((e) => e.path === filePath || e.path.endsWith(filePath));
|
||||
}
|
||||
|
||||
/**
|
||||
* Helper to read first chunk from stream
|
||||
*/
|
||||
private async readFirstChunk(stream: plugins.stream.Readable): Promise<Buffer> {
|
||||
return new Promise((resolve, reject) => {
|
||||
const onData = (chunk: Buffer) => {
|
||||
stream.removeListener('data', onData);
|
||||
stream.removeListener('error', reject);
|
||||
resolve(chunk);
|
||||
};
|
||||
stream.on('data', onData);
|
||||
stream.on('error', reject);
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,11 +1,14 @@
|
||||
import type { SmartArchive } from './classes.smartarchive.js';
|
||||
import * as plugins from './plugins.js';
|
||||
import type { IArchiveEntry, TCompressionLevel } from './interfaces.js';
|
||||
import { GzipTools } from './classes.gziptools.js';
|
||||
|
||||
/**
|
||||
* TAR archive creation and extraction utilities
|
||||
*/
|
||||
export class TarTools {
|
||||
// INSTANCE
|
||||
constructor() {}
|
||||
|
||||
// packing
|
||||
/**
|
||||
* Add a file to a TAR pack stream
|
||||
*/
|
||||
public async addFileToPack(
|
||||
pack: plugins.tarStream.Pack,
|
||||
optionsArg: {
|
||||
@@ -13,12 +16,12 @@ export class TarTools {
|
||||
content?:
|
||||
| string
|
||||
| Buffer
|
||||
| plugins.smartstream.stream.Readable
|
||||
| plugins.stream.Readable
|
||||
| plugins.smartfile.SmartFile
|
||||
| plugins.smartfile.StreamFile;
|
||||
byteLength?: number;
|
||||
filePath?: string;
|
||||
},
|
||||
}
|
||||
): Promise<void> {
|
||||
return new Promise<void>(async (resolve, reject) => {
|
||||
let fileName: string | null = null;
|
||||
@@ -26,18 +29,20 @@ export class TarTools {
|
||||
if (optionsArg.fileName) {
|
||||
fileName = optionsArg.fileName;
|
||||
} else if (optionsArg.content instanceof plugins.smartfile.SmartFile) {
|
||||
fileName = (optionsArg.content as plugins.smartfile.SmartFile).relative;
|
||||
fileName = optionsArg.content.relative;
|
||||
} else if (optionsArg.content instanceof plugins.smartfile.StreamFile) {
|
||||
fileName = (optionsArg.content as plugins.smartfile.StreamFile)
|
||||
.relativeFilePath;
|
||||
fileName = optionsArg.content.relativeFilePath;
|
||||
} else if (optionsArg.filePath) {
|
||||
fileName = optionsArg.filePath;
|
||||
}
|
||||
|
||||
/**
|
||||
* contentByteLength is used to set the size of the entry in the tar file
|
||||
*/
|
||||
let contentByteLength: number;
|
||||
if (!fileName) {
|
||||
reject(new Error('No filename specified for TAR entry'));
|
||||
return;
|
||||
}
|
||||
|
||||
// Determine content byte length
|
||||
let contentByteLength: number | undefined;
|
||||
if (optionsArg.byteLength) {
|
||||
contentByteLength = optionsArg.byteLength;
|
||||
} else if (typeof optionsArg.content === 'string') {
|
||||
@@ -45,72 +50,59 @@ export class TarTools {
|
||||
} else if (Buffer.isBuffer(optionsArg.content)) {
|
||||
contentByteLength = optionsArg.content.length;
|
||||
} else if (optionsArg.content instanceof plugins.smartfile.SmartFile) {
|
||||
contentByteLength = await optionsArg.content.getSize(); // assuming SmartFile has getSize method
|
||||
contentByteLength = await optionsArg.content.getSize();
|
||||
} else if (optionsArg.content instanceof plugins.smartfile.StreamFile) {
|
||||
contentByteLength = await optionsArg.content.getSize(); // assuming StreamFile has getSize method
|
||||
} else if (
|
||||
optionsArg.content instanceof plugins.smartstream.stream.Readable
|
||||
) {
|
||||
console.warn(
|
||||
'@push.rocks/smartarchive: When streaming, it is recommended to provide byteLength, if known.',
|
||||
);
|
||||
contentByteLength = await optionsArg.content.getSize();
|
||||
} else if (optionsArg.filePath) {
|
||||
const fileStat = await plugins.fsPromises.stat(optionsArg.filePath);
|
||||
contentByteLength = fileStat.size;
|
||||
}
|
||||
|
||||
/**
|
||||
* here we try to harmonize all kind of entries towards a readable stream
|
||||
*/
|
||||
let content: plugins.smartstream.stream.Readable;
|
||||
// Convert all content types to Readable stream
|
||||
let content: plugins.stream.Readable;
|
||||
if (Buffer.isBuffer(optionsArg.content)) {
|
||||
content = plugins.smartstream.stream.Readable.from(optionsArg.content);
|
||||
content = plugins.stream.Readable.from(optionsArg.content);
|
||||
} else if (typeof optionsArg.content === 'string') {
|
||||
content = plugins.smartstream.stream.Readable.from(
|
||||
Buffer.from(optionsArg.content),
|
||||
);
|
||||
content = plugins.stream.Readable.from(Buffer.from(optionsArg.content));
|
||||
} else if (optionsArg.content instanceof plugins.smartfile.SmartFile) {
|
||||
content = plugins.smartstream.stream.Readable.from(
|
||||
optionsArg.content.contents,
|
||||
);
|
||||
content = plugins.stream.Readable.from(optionsArg.content.contents);
|
||||
} else if (optionsArg.content instanceof plugins.smartfile.StreamFile) {
|
||||
content = await optionsArg.content.createReadStream();
|
||||
} else if (
|
||||
optionsArg.content instanceof plugins.smartstream.stream.Readable
|
||||
) {
|
||||
} else if (optionsArg.content instanceof plugins.stream.Readable) {
|
||||
content = optionsArg.content;
|
||||
} else if (optionsArg.filePath) {
|
||||
content = plugins.fs.createReadStream(optionsArg.filePath);
|
||||
} else {
|
||||
reject(new Error('No content or filePath specified for TAR entry'));
|
||||
return;
|
||||
}
|
||||
|
||||
const entry = pack.entry(
|
||||
{
|
||||
name: fileName,
|
||||
...(contentByteLength
|
||||
? {
|
||||
size: contentByteLength,
|
||||
}
|
||||
: null),
|
||||
...(contentByteLength !== undefined ? { size: contentByteLength } : {}),
|
||||
},
|
||||
(err: Error) => {
|
||||
(err: Error | null) => {
|
||||
if (err) {
|
||||
reject(err);
|
||||
} else {
|
||||
resolve();
|
||||
}
|
||||
},
|
||||
}
|
||||
);
|
||||
|
||||
content.pipe(entry);
|
||||
resolve();
|
||||
// Note: resolve() is called in the callback above when pipe completes
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* packs a directory from disk into a tar stream
|
||||
* @param directoryPath
|
||||
* Pack a directory into a TAR stream
|
||||
*/
|
||||
public async packDirectory(directoryPath: string) {
|
||||
public async packDirectory(directoryPath: string): Promise<plugins.tarStream.Pack> {
|
||||
const fileTree = await plugins.listFileTree(directoryPath, '**/*');
|
||||
const pack = await this.getPackStream();
|
||||
|
||||
for (const filePath of fileTree) {
|
||||
const absolutePath = plugins.path.join(directoryPath, filePath);
|
||||
const fileStat = await plugins.fsPromises.stat(absolutePath);
|
||||
@@ -121,16 +113,96 @@ export class TarTools {
|
||||
content: plugins.fs.createReadStream(absolutePath),
|
||||
});
|
||||
}
|
||||
|
||||
return pack;
|
||||
}
|
||||
|
||||
public async getPackStream() {
|
||||
const pack = plugins.tarStream.pack();
|
||||
return pack;
|
||||
/**
|
||||
* Get a new TAR pack stream
|
||||
*/
|
||||
public async getPackStream(): Promise<plugins.tarStream.Pack> {
|
||||
return plugins.tarStream.pack();
|
||||
}
|
||||
|
||||
// extracting
|
||||
getDecompressionStream() {
|
||||
/**
|
||||
* Get a TAR extraction stream
|
||||
*/
|
||||
public getDecompressionStream(): plugins.tarStream.Extract {
|
||||
return plugins.tarStream.extract();
|
||||
}
|
||||
|
||||
/**
|
||||
* Pack files into a TAR buffer
|
||||
*/
|
||||
public async packFiles(files: IArchiveEntry[]): Promise<Buffer> {
|
||||
const pack = await this.getPackStream();
|
||||
|
||||
for (const file of files) {
|
||||
await this.addFileToPack(pack, {
|
||||
fileName: file.archivePath,
|
||||
content: file.content as string | Buffer | plugins.stream.Readable | plugins.smartfile.SmartFile | plugins.smartfile.StreamFile,
|
||||
byteLength: file.size,
|
||||
});
|
||||
}
|
||||
|
||||
pack.finalize();
|
||||
|
||||
const chunks: Buffer[] = [];
|
||||
return new Promise((resolve, reject) => {
|
||||
pack.on('data', (chunk: Buffer) => chunks.push(chunk));
|
||||
pack.on('end', () => resolve(Buffer.concat(chunks)));
|
||||
pack.on('error', reject);
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Pack a directory into a TAR.GZ buffer
|
||||
*/
|
||||
public async packDirectoryToTarGz(
|
||||
directoryPath: string,
|
||||
compressionLevel?: TCompressionLevel
|
||||
): Promise<Buffer> {
|
||||
const pack = await this.packDirectory(directoryPath);
|
||||
pack.finalize();
|
||||
|
||||
const gzipTools = new GzipTools();
|
||||
const gzipStream = gzipTools.getCompressionStream(compressionLevel);
|
||||
|
||||
const chunks: Buffer[] = [];
|
||||
return new Promise((resolve, reject) => {
|
||||
pack
|
||||
.pipe(gzipStream)
|
||||
.on('data', (chunk: Buffer) => chunks.push(chunk))
|
||||
.on('end', () => resolve(Buffer.concat(chunks)))
|
||||
.on('error', reject);
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Pack a directory into a TAR.GZ stream
|
||||
*/
|
||||
public async packDirectoryToTarGzStream(
|
||||
directoryPath: string,
|
||||
compressionLevel?: TCompressionLevel
|
||||
): Promise<plugins.stream.Readable> {
|
||||
const pack = await this.packDirectory(directoryPath);
|
||||
pack.finalize();
|
||||
|
||||
const gzipTools = new GzipTools();
|
||||
const gzipStream = gzipTools.getCompressionStream(compressionLevel);
|
||||
|
||||
return pack.pipe(gzipStream);
|
||||
}
|
||||
|
||||
/**
|
||||
* Pack files into a TAR.GZ buffer
|
||||
*/
|
||||
public async packFilesToTarGz(
|
||||
files: IArchiveEntry[],
|
||||
compressionLevel?: TCompressionLevel
|
||||
): Promise<Buffer> {
|
||||
const tarBuffer = await this.packFiles(files);
|
||||
const gzipTools = new GzipTools();
|
||||
return gzipTools.compress(tarBuffer, compressionLevel);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,83 +1,209 @@
|
||||
import type { SmartArchive } from './classes.smartarchive.js';
|
||||
import * as plugins from './plugins.js';
|
||||
import type { IArchiveEntry, TCompressionLevel } from './interfaces.js';
|
||||
|
||||
class DecompressZipTransform extends plugins.smartstream
|
||||
.SmartDuplex<ArrayBufferLike> {
|
||||
private streamtools: plugins.smartstream.IStreamTools;
|
||||
/**
|
||||
* Transform stream for ZIP decompression using fflate
|
||||
* Emits StreamFile objects for each file in the archive
|
||||
*/
|
||||
export class ZipDecompressionTransform extends plugins.smartstream.SmartDuplex<Buffer, plugins.smartfile.StreamFile> {
|
||||
private streamtools!: plugins.smartstream.IStreamTools;
|
||||
private unzipper = new plugins.fflate.Unzip(async (fileArg) => {
|
||||
let resultBuffer: Buffer;
|
||||
fileArg.ondata = async (flateError, dat, final) => {
|
||||
fileArg.ondata = async (_flateError, dat, final) => {
|
||||
resultBuffer
|
||||
? (resultBuffer = Buffer.concat([resultBuffer, Buffer.from(dat)]))
|
||||
: (resultBuffer = Buffer.from(dat));
|
||||
if (final) {
|
||||
const streamFile =
|
||||
plugins.smartfile.StreamFile.fromBuffer(resultBuffer);
|
||||
const streamFile = plugins.smartfile.StreamFile.fromBuffer(resultBuffer);
|
||||
streamFile.relativeFilePath = fileArg.name;
|
||||
this.streamtools.push(streamFile);
|
||||
}
|
||||
};
|
||||
fileArg.start();
|
||||
});
|
||||
|
||||
constructor() {
|
||||
super({
|
||||
objectMode: true,
|
||||
writeFunction: async (chunkArg, streamtoolsArg) => {
|
||||
this.streamtools ? null : (this.streamtools = streamtoolsArg);
|
||||
this.unzipper.push(
|
||||
Buffer.isBuffer(chunkArg) ? chunkArg : Buffer.from(chunkArg),
|
||||
false,
|
||||
Buffer.isBuffer(chunkArg) ? chunkArg : Buffer.from(chunkArg as unknown as ArrayBuffer),
|
||||
false
|
||||
);
|
||||
return null;
|
||||
},
|
||||
finalFunction: async () => {
|
||||
this.unzipper.push(Buffer.from(''), true);
|
||||
await plugins.smartdelay.delayFor(0);
|
||||
await this.streamtools.push(null);
|
||||
return null;
|
||||
},
|
||||
});
|
||||
this.unzipper.register(plugins.fflate.UnzipInflate);
|
||||
}
|
||||
}
|
||||
|
||||
// This class wraps fflate's zip in a Node.js Transform stream for compression
|
||||
export class CompressZipTransform extends plugins.stream.Transform {
|
||||
files: { [fileName: string]: Uint8Array };
|
||||
/**
|
||||
* Streaming ZIP compression using fflate
|
||||
* Allows adding multiple entries before finalizing
|
||||
*/
|
||||
export class ZipCompressionStream extends plugins.stream.Duplex {
|
||||
private files: Map<string, { data: Uint8Array; options?: plugins.fflate.ZipOptions }> = new Map();
|
||||
private finalized = false;
|
||||
|
||||
constructor() {
|
||||
super();
|
||||
this.files = {};
|
||||
}
|
||||
|
||||
_transform(
|
||||
chunk: Buffer,
|
||||
encoding: BufferEncoding,
|
||||
callback: plugins.stream.TransformCallback,
|
||||
) {
|
||||
// Simple example: storing chunks in memory before finalizing ZIP in _flush
|
||||
this.files['file.txt'] = new Uint8Array(chunk);
|
||||
callback();
|
||||
}
|
||||
/**
|
||||
* Add a file entry to the ZIP archive
|
||||
*/
|
||||
public async addEntry(
|
||||
fileName: string,
|
||||
content: Buffer | plugins.stream.Readable,
|
||||
options?: { compressionLevel?: TCompressionLevel }
|
||||
): Promise<void> {
|
||||
if (this.finalized) {
|
||||
throw new Error('Cannot add entries to a finalized ZIP archive');
|
||||
}
|
||||
|
||||
_flush(callback: plugins.stream.TransformCallback) {
|
||||
plugins.fflate.zip(this.files, (err, zipped) => {
|
||||
if (err) {
|
||||
callback(err);
|
||||
} else {
|
||||
this.push(Buffer.from(zipped));
|
||||
callback();
|
||||
let data: Buffer;
|
||||
if (Buffer.isBuffer(content)) {
|
||||
data = content;
|
||||
} else {
|
||||
// Collect stream to buffer
|
||||
const chunks: Buffer[] = [];
|
||||
for await (const chunk of content) {
|
||||
chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));
|
||||
}
|
||||
data = Buffer.concat(chunks);
|
||||
}
|
||||
|
||||
this.files.set(fileName, {
|
||||
data: new Uint8Array(data),
|
||||
options: options?.compressionLevel !== undefined ? { level: options.compressionLevel } : undefined,
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Finalize the ZIP archive and emit the compressed data
|
||||
*/
|
||||
public async finalize(): Promise<void> {
|
||||
if (this.finalized) {
|
||||
return;
|
||||
}
|
||||
this.finalized = true;
|
||||
|
||||
const filesObj: plugins.fflate.Zippable = {};
|
||||
for (const [name, { data, options }] of this.files) {
|
||||
filesObj[name] = options ? [data, options] : data;
|
||||
}
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
plugins.fflate.zip(filesObj, (err, result) => {
|
||||
if (err) {
|
||||
reject(err);
|
||||
} else {
|
||||
this.push(Buffer.from(result));
|
||||
this.push(null);
|
||||
resolve();
|
||||
}
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
_read(): void {
|
||||
// No-op: data is pushed when finalize() is called
|
||||
}
|
||||
|
||||
_write(
|
||||
_chunk: Buffer,
|
||||
_encoding: BufferEncoding,
|
||||
callback: (error?: Error | null) => void
|
||||
): void {
|
||||
// Not used for ZIP creation - use addEntry() instead
|
||||
callback(new Error('Use addEntry() to add files to the ZIP archive'));
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* ZIP compression and decompression utilities
|
||||
*/
|
||||
export class ZipTools {
|
||||
/**
|
||||
* Get a streaming compression object for creating ZIP archives
|
||||
*/
|
||||
public getCompressionStream(): ZipCompressionStream {
|
||||
return new ZipCompressionStream();
|
||||
}
|
||||
|
||||
/**
|
||||
* Get a streaming decompression transform for extracting ZIP archives
|
||||
*/
|
||||
public getDecompressionStream(): ZipDecompressionTransform {
|
||||
return new ZipDecompressionTransform();
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a ZIP archive from an array of entries
|
||||
*/
|
||||
public async createZip(entries: IArchiveEntry[], compressionLevel?: TCompressionLevel): Promise<Buffer> {
|
||||
const filesObj: plugins.fflate.Zippable = {};
|
||||
|
||||
for (const entry of entries) {
|
||||
let data: Uint8Array;
|
||||
|
||||
if (typeof entry.content === 'string') {
|
||||
data = new TextEncoder().encode(entry.content);
|
||||
} else if (Buffer.isBuffer(entry.content)) {
|
||||
data = new Uint8Array(entry.content);
|
||||
} else if (entry.content instanceof plugins.smartfile.SmartFile) {
|
||||
data = new Uint8Array(entry.content.contents);
|
||||
} else if (entry.content instanceof plugins.smartfile.StreamFile) {
|
||||
const buffer = await entry.content.getContentAsBuffer();
|
||||
data = new Uint8Array(buffer);
|
||||
} else {
|
||||
// Readable stream
|
||||
const chunks: Buffer[] = [];
|
||||
for await (const chunk of entry.content as plugins.stream.Readable) {
|
||||
chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));
|
||||
}
|
||||
data = new Uint8Array(Buffer.concat(chunks));
|
||||
}
|
||||
|
||||
if (compressionLevel !== undefined) {
|
||||
filesObj[entry.archivePath] = [data, { level: compressionLevel }];
|
||||
} else {
|
||||
filesObj[entry.archivePath] = data;
|
||||
}
|
||||
}
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
plugins.fflate.zip(filesObj, (err, result) => {
|
||||
if (err) reject(err);
|
||||
else resolve(Buffer.from(result));
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Extract a ZIP buffer to an array of entries
|
||||
*/
|
||||
public async extractZip(data: Buffer): Promise<Array<{ path: string; content: Buffer }>> {
|
||||
return new Promise((resolve, reject) => {
|
||||
plugins.fflate.unzip(data, (err, result) => {
|
||||
if (err) {
|
||||
reject(err);
|
||||
return;
|
||||
}
|
||||
|
||||
const entries: Array<{ path: string; content: Buffer }> = [];
|
||||
for (const [path, content] of Object.entries(result)) {
|
||||
entries.push({ path, content: Buffer.from(content) });
|
||||
}
|
||||
resolve(entries);
|
||||
});
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
export class ZipTools {
|
||||
constructor() {}
|
||||
|
||||
public getCompressionStream() {
|
||||
return new CompressZipTransform();
|
||||
}
|
||||
|
||||
public getDecompressionStream() {
|
||||
return new DecompressZipTransform();
|
||||
}
|
||||
}
|
||||
|
||||
70
ts/errors.ts
Normal file
70
ts/errors.ts
Normal file
@@ -0,0 +1,70 @@
|
||||
/**
|
||||
* Base error class for smartarchive
|
||||
*/
|
||||
export class SmartArchiveError extends Error {
|
||||
public readonly code: string;
|
||||
|
||||
constructor(message: string, code: string) {
|
||||
super(message);
|
||||
this.name = 'SmartArchiveError';
|
||||
this.code = code;
|
||||
// Maintains proper stack trace for where error was thrown (V8)
|
||||
if (Error.captureStackTrace) {
|
||||
Error.captureStackTrace(this, this.constructor);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* BZIP2-specific decompression errors
|
||||
*/
|
||||
export class Bzip2Error extends SmartArchiveError {
|
||||
constructor(message: string, code: string = 'BZIP2_ERROR') {
|
||||
super(message, code);
|
||||
this.name = 'Bzip2Error';
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Archive format detection errors
|
||||
*/
|
||||
export class ArchiveFormatError extends SmartArchiveError {
|
||||
constructor(message: string) {
|
||||
super(message, 'ARCHIVE_FORMAT_ERROR');
|
||||
this.name = 'ArchiveFormatError';
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Stream processing errors
|
||||
*/
|
||||
export class StreamError extends SmartArchiveError {
|
||||
constructor(message: string) {
|
||||
super(message, 'STREAM_ERROR');
|
||||
this.name = 'StreamError';
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* BZIP2 error codes for programmatic error handling
|
||||
*/
|
||||
export const BZIP2_ERROR_CODES = {
|
||||
NO_MAGIC_NUMBER: 'BZIP2_NO_MAGIC',
|
||||
INVALID_ARCHIVE: 'BZIP2_INVALID_ARCHIVE',
|
||||
CRC_MISMATCH: 'BZIP2_CRC_MISMATCH',
|
||||
INVALID_BLOCK_DATA: 'BZIP2_INVALID_BLOCK',
|
||||
BUFFER_OVERFLOW: 'BZIP2_BUFFER_OVERFLOW',
|
||||
INVALID_HUFFMAN: 'BZIP2_INVALID_HUFFMAN',
|
||||
INVALID_SELECTOR: 'BZIP2_INVALID_SELECTOR',
|
||||
INVALID_POSITION: 'BZIP2_INVALID_POSITION',
|
||||
PREMATURE_END: 'BZIP2_PREMATURE_END',
|
||||
} as const;
|
||||
|
||||
export type TBzip2ErrorCode = typeof BZIP2_ERROR_CODES[keyof typeof BZIP2_ERROR_CODES];
|
||||
|
||||
/**
|
||||
* Throw a BZIP2 error with a specific code
|
||||
*/
|
||||
export function throwBzip2Error(message: string, code: TBzip2ErrorCode): never {
|
||||
throw new Bzip2Error(message, code);
|
||||
}
|
||||
11
ts/index.ts
11
ts/index.ts
@@ -1,4 +1,15 @@
|
||||
// Core types and errors
|
||||
export * from './interfaces.js';
|
||||
export * from './errors.js';
|
||||
|
||||
// Main archive class
|
||||
export * from './classes.smartarchive.js';
|
||||
|
||||
// Format-specific tools
|
||||
export * from './classes.tartools.js';
|
||||
export * from './classes.ziptools.js';
|
||||
export * from './classes.gziptools.js';
|
||||
export * from './classes.bzip2tools.js';
|
||||
|
||||
// Archive analysis
|
||||
export * from './classes.archiveanalyzer.js';
|
||||
|
||||
131
ts/interfaces.ts
Normal file
131
ts/interfaces.ts
Normal file
@@ -0,0 +1,131 @@
|
||||
import type * as stream from 'node:stream';
|
||||
import type { SmartFile, StreamFile } from '@push.rocks/smartfile';
|
||||
|
||||
/**
|
||||
* Supported archive formats
|
||||
*/
|
||||
export type TArchiveFormat = 'tar' | 'tar.gz' | 'tgz' | 'zip' | 'gz' | 'bz2';
|
||||
|
||||
/**
|
||||
* Compression level (0 = no compression, 9 = maximum compression)
|
||||
*/
|
||||
export type TCompressionLevel = 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9;
|
||||
|
||||
/**
|
||||
* Supported MIME types for archive detection
|
||||
*/
|
||||
export type TSupportedMime =
|
||||
| 'application/gzip'
|
||||
| 'application/zip'
|
||||
| 'application/x-bzip2'
|
||||
| 'application/x-tar'
|
||||
| undefined;
|
||||
|
||||
/**
|
||||
* Entry to add to an archive during creation
|
||||
*/
|
||||
export interface IArchiveEntry {
|
||||
/** Path within the archive */
|
||||
archivePath: string;
|
||||
/** Content: string, Buffer, Readable stream, SmartFile, or StreamFile */
|
||||
content: string | Buffer | stream.Readable | SmartFile | StreamFile;
|
||||
/** Optional size hint for streams (improves performance) */
|
||||
size?: number;
|
||||
/** Optional file mode/permissions */
|
||||
mode?: number;
|
||||
/** Optional modification time */
|
||||
mtime?: Date;
|
||||
}
|
||||
|
||||
/**
|
||||
* Options for creating archives
|
||||
*/
|
||||
export interface IArchiveCreationOptions {
|
||||
/** Target archive format */
|
||||
format: TArchiveFormat;
|
||||
/** Compression level (0-9, default depends on format) */
|
||||
compressionLevel?: TCompressionLevel;
|
||||
/** Base path to strip from file paths in archive */
|
||||
basePath?: string;
|
||||
}
|
||||
|
||||
/**
|
||||
* Options for extracting archives
|
||||
*/
|
||||
export interface IArchiveExtractionOptions {
|
||||
/** Target directory for extraction */
|
||||
targetDir: string;
|
||||
/** Optional filename for single-file archives (gz, bz2) */
|
||||
fileName?: string;
|
||||
/** Number of leading path components to strip */
|
||||
stripComponents?: number;
|
||||
/** Filter function to select which entries to extract */
|
||||
filter?: (entry: IArchiveEntryInfo) => boolean;
|
||||
/** Whether to overwrite existing files */
|
||||
overwrite?: boolean;
|
||||
}
|
||||
|
||||
/**
|
||||
* Information about an archive entry
|
||||
*/
|
||||
export interface IArchiveEntryInfo {
|
||||
/** Path of the entry within the archive */
|
||||
path: string;
|
||||
/** Size in bytes */
|
||||
size: number;
|
||||
/** Whether this entry is a directory */
|
||||
isDirectory: boolean;
|
||||
/** Whether this entry is a file */
|
||||
isFile: boolean;
|
||||
/** Modification time */
|
||||
mtime?: Date;
|
||||
/** File mode/permissions */
|
||||
mode?: number;
|
||||
}
|
||||
|
||||
/**
|
||||
* Result of archive analysis
|
||||
*/
|
||||
export interface IArchiveInfo {
|
||||
/** Detected archive format */
|
||||
format: TArchiveFormat | null;
|
||||
/** Whether the archive is compressed */
|
||||
isCompressed: boolean;
|
||||
/** Whether this is a recognized archive format */
|
||||
isArchive: boolean;
|
||||
/** List of entries (if available without full extraction) */
|
||||
entries?: IArchiveEntryInfo[];
|
||||
}
|
||||
|
||||
/**
|
||||
* Options for adding a file to a TAR pack stream
|
||||
*/
|
||||
export interface IAddFileOptions {
|
||||
/** Filename within the archive */
|
||||
fileName?: string;
|
||||
/** File content */
|
||||
content?: string | Buffer | stream.Readable | SmartFile | StreamFile;
|
||||
/** Size in bytes (required for streams) */
|
||||
byteLength?: number;
|
||||
/** Path to file on disk (alternative to content) */
|
||||
filePath?: string;
|
||||
}
|
||||
|
||||
/**
|
||||
* Bit reader interface for BZIP2 decompression
|
||||
*/
|
||||
export interface IBitReader {
|
||||
(n: number | null): number | void;
|
||||
bytesRead: number;
|
||||
}
|
||||
|
||||
/**
|
||||
* Huffman group for BZIP2 decompression
|
||||
*/
|
||||
export interface IHuffmanGroup {
|
||||
permute: Int32Array;
|
||||
limit: Int32Array;
|
||||
base: Int32Array;
|
||||
minLen: number;
|
||||
maxLen: number;
|
||||
}
|
||||
Reference in New Issue
Block a user