// loaders.gl
// SPDX-License-Identifier: MIT
// Copyright (c) vis.gl contributors
import { Compression } from "./compression.js";
import { registerJSModules, checkJSModule, getJSModule, getJSModuleOrNull } from '@loaders.gl/loader-utils';
// import {ZstdCodec} from 'zstd-codec'; // https://bundlephobia.com/package/zstd-codec
const CHUNK_SIZE = 1000000; // Tested value
let zstdPromise;
let zstd;
/**
 * Zstandard compression / decompression
 */
export class ZstdCompression extends Compression {
    name = 'zstd';
    extensions = [];
    contentEncodings = [];
    isSupported = true;
    options;
    /**
     * zstd-codec is an injectable dependency due to big size
     * @param options
     */
    constructor(options) {
        super(options);
        this.options = options;
        registerJSModules(options?.modules);
    }
    async preload(modules = {}) {
        registerJSModules(modules);
        checkJSModule('zstd-codec', this.name);
        const ZstdCodec = getJSModuleOrNull('zstd-codec');
        // eslint-disable-next-line  @typescript-eslint/no-misused-promises
        if (!zstdPromise && ZstdCodec) {
            zstdPromise = new Promise((resolve) => ZstdCodec.run((zstd) => resolve(zstd)));
            zstd = await zstdPromise;
        }
    }
    compressSync(input) {
        getJSModule('zstd-codec', this.name);
        const simpleZstd = new zstd.Simple();
        const inputArray = new Uint8Array(input);
        return simpleZstd.compress(inputArray).buffer;
    }
    decompressSync(input) {
        getJSModule('zstd-codec', this.name);
        const simpleZstd = new zstd.Simple();
        // var ddict = new zstd.Dict.Decompression(dictData);
        // var jsonBytes = simpleZstd.decompressUsingDict(jsonZstData, ddict);
        const inputArray = new Uint8Array(input);
        return simpleZstd.decompress(inputArray).buffer;
    }
    async decompress(input, size) {
        await this.preload();
        const simpleZstd = new zstd.Streaming();
        const inputArray = new Uint8Array(input);
        const chunks = [];
        for (let i = 0; i <= inputArray.length; i += CHUNK_SIZE) {
            chunks.push(inputArray.subarray(i, i + CHUNK_SIZE));
        }
        const decompressResult = await simpleZstd.decompressChunks(chunks);
        return decompressResult.buffer;
    }
}