ferrijs-std 0.2.4

// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
use crate::buffer::Buffer;
use crate::context::CtxExtension;
use crate::utils::{bytes::ObjectBytes, object::ObjectExt, result::ResultExt};
use rquickjs::{
    prelude::{Opt, Rest},
    Ctx, Error, Exception, Function, IntoJs, Null, Result, Value,
};

use crate::{define_cb_function, define_sync_function};
use super::{max_output_length, read_to_end_limited};

enum ZstdCommand {
    Compress,
    Decompress,
}

fn zstd_converter<'js>(
    ctx: Ctx<'js>,
    bytes: ObjectBytes<'js>,
    options: Opt<Value<'js>>,
    command: ZstdCommand,
) -> Result<Value<'js>> {
    let src = bytes.as_bytes(&ctx)?;

    let mut level = crate::compression::zstd::DEFAULT_COMPRESSION_LEVEL;
    if let Some(options) = options.0.as_ref() {
        if let Some(opt) = options.get_optional("level")? {
            level = opt;
        }
    }
    let limit = max_output_length(&options)?;

    let dst = match command {
        ZstdCommand::Compress => read_to_end_limited(
            &ctx,
            crate::compression::zstd::encoder(src, level)?,
            limit,
            src.len(),
        )?,
        ZstdCommand::Decompress => read_to_end_limited(
            &ctx,
            crate::compression::zstd::decoder(src)?,
            limit,
            src.len(),
        )?,
    };

    Buffer(dst).into_js(&ctx)
}

define_cb_function!(zstd_comp, zstd_converter, ZstdCommand::Compress);
define_sync_function!(zstd_comp_sync, zstd_converter, ZstdCommand::Compress);

define_cb_function!(zstd_decomp, zstd_converter, ZstdCommand::Decompress);
define_sync_function!(zstd_decomp_sync, zstd_converter, ZstdCommand::Decompress);