ipfs-unixfs-importer/dist/src/chunker/rabin.js

JavaScript implementation of the UnixFs importer used by IPFS

// @ts-expect-error no types
import { create } from 'rabin-wasm';
import { Uint8ArrayList } from 'uint8arraylist';
import { InvalidAvgChunkSizeError, InvalidChunkSizeError, InvalidMinChunkSizeError } from '../errors.js';
const DEFAULT_MIN_CHUNK_SIZE = 262144;
const DEFAULT_MAX_CHUNK_SIZE = 262144;
const DEFAULT_AVG_CHUNK_SIZE = 262144;
const DEFAULT_WINDOW = 16;
async function* chunker(source, r) {
    const buffers = new Uint8ArrayList();
    for await (const chunk of source) {
        buffers.append(chunk);
        const sizes = r.fingerprint(chunk);
        for (let i = 0; i < sizes.length; i++) {
            const size = sizes[i];
            const buf = buffers.slice(0, size);
            buffers.consume(size);
            yield buf;
        }
    }
    if (buffers.length > 0) {
        yield buffers.subarray(0);
    }
}
export const rabin = (options = {}) => {
    let min = options.minChunkSize ?? DEFAULT_MIN_CHUNK_SIZE;
    let max = options.maxChunkSize ?? DEFAULT_MAX_CHUNK_SIZE;
    let avg = options.avgChunkSize ?? DEFAULT_AVG_CHUNK_SIZE;
    const window = options.window ?? DEFAULT_WINDOW;
    // if only avg was passed, calculate min/max from that
    if (options.avgChunkSize != null && options.minChunkSize == null && options.maxChunkSize == null) {
        min = avg / 3;
        max = avg + (avg / 2);
    }
    const isInvalidChunkSizes = [min, avg, max].some((size) => size == null || isNaN(size));
    if (isInvalidChunkSizes) {
        if (options.avgChunkSize != null) {
            throw new InvalidAvgChunkSizeError('please specify a valid average chunk size number');
        }
        throw new InvalidChunkSizeError('please specify valid numbers for (min|max|avg)ChunkSize');
    }
    // validate min/max/avg in the same way as go
    if (min < 16) {
        throw new InvalidMinChunkSizeError('rabin min must be greater than 16');
    }
    if (max < min) {
        max = min;
    }
    if (avg < min) {
        avg = min;
    }
    const sizepow = Math.floor(Math.log2(avg));
    return async function* rabinChunker(source) {
        const r = await create(sizepow, min, max, window);
        for await (const chunk of chunker(source, r)) {
            yield chunk;
        }
    };
};
//# sourceMappingURL=rabin.js.map