mirror of
https://github.com/denoland/deno.git
synced 2024-12-22 15:24:46 -05:00
dda0f1c343
`ZeroCopyBuf` was convenient to use, but sometimes it did hide details that some copies were necessary in certain cases. Also it made it way to easy for the caller to pass around and convert into different values. This commit splits `ZeroCopyBuf` into `JsBuffer` (an array buffer coming from V8) and `ToJsBuffer` (a Rust buffer that will be converted into a V8 array buffer). As a result some magical conversions were removed (they were never used) limiting the API surface and preparing for changes in #19534.
128 lines
3.4 KiB
Rust
128 lines
3.4 KiB
Rust
// Copyright 2018-2023 the Deno authors. All rights reserved. MIT license.
|
|
|
|
use deno_core::error::AnyError;
|
|
use deno_core::op;
|
|
use deno_core::OpState;
|
|
use deno_core::Resource;
|
|
use deno_core::ResourceId;
|
|
use deno_core::ToJsBuffer;
|
|
use flate2::write::DeflateDecoder;
|
|
use flate2::write::DeflateEncoder;
|
|
use flate2::write::GzDecoder;
|
|
use flate2::write::GzEncoder;
|
|
use flate2::write::ZlibDecoder;
|
|
use flate2::write::ZlibEncoder;
|
|
use flate2::Compression;
|
|
use std::borrow::Cow;
|
|
use std::cell::RefCell;
|
|
use std::io::Write;
|
|
use std::rc::Rc;
|
|
|
|
#[derive(Debug)]
|
|
struct CompressionResource(RefCell<Inner>);
|
|
|
|
/// https://wicg.github.io/compression/#supported-formats
|
|
#[derive(Debug)]
|
|
enum Inner {
|
|
DeflateDecoder(ZlibDecoder<Vec<u8>>),
|
|
DeflateEncoder(ZlibEncoder<Vec<u8>>),
|
|
DeflateRawDecoder(DeflateDecoder<Vec<u8>>),
|
|
DeflateRawEncoder(DeflateEncoder<Vec<u8>>),
|
|
GzDecoder(GzDecoder<Vec<u8>>),
|
|
GzEncoder(GzEncoder<Vec<u8>>),
|
|
}
|
|
|
|
impl Resource for CompressionResource {
|
|
fn name(&self) -> Cow<str> {
|
|
"compression".into()
|
|
}
|
|
}
|
|
|
|
#[op]
|
|
pub fn op_compression_new(
|
|
state: &mut OpState,
|
|
format: &str,
|
|
is_decoder: bool,
|
|
) -> ResourceId {
|
|
let w = Vec::new();
|
|
let inner = match (format, is_decoder) {
|
|
("deflate", true) => Inner::DeflateDecoder(ZlibDecoder::new(w)),
|
|
("deflate", false) => {
|
|
Inner::DeflateEncoder(ZlibEncoder::new(w, Compression::default()))
|
|
}
|
|
("deflate-raw", true) => Inner::DeflateRawDecoder(DeflateDecoder::new(w)),
|
|
("deflate-raw", false) => {
|
|
Inner::DeflateRawEncoder(DeflateEncoder::new(w, Compression::default()))
|
|
}
|
|
("gzip", true) => Inner::GzDecoder(GzDecoder::new(w)),
|
|
("gzip", false) => {
|
|
Inner::GzEncoder(GzEncoder::new(w, Compression::default()))
|
|
}
|
|
_ => unreachable!(),
|
|
};
|
|
let resource = CompressionResource(RefCell::new(inner));
|
|
state.resource_table.add(resource)
|
|
}
|
|
|
|
#[op]
|
|
pub fn op_compression_write(
|
|
state: &mut OpState,
|
|
rid: ResourceId,
|
|
input: &[u8],
|
|
) -> Result<ToJsBuffer, AnyError> {
|
|
let resource = state.resource_table.get::<CompressionResource>(rid)?;
|
|
let mut inner = resource.0.borrow_mut();
|
|
let out: Vec<u8> = match &mut *inner {
|
|
Inner::DeflateDecoder(d) => {
|
|
d.write_all(input)?;
|
|
d.flush()?;
|
|
d.get_mut().drain(..)
|
|
}
|
|
Inner::DeflateEncoder(d) => {
|
|
d.write_all(input)?;
|
|
d.flush()?;
|
|
d.get_mut().drain(..)
|
|
}
|
|
Inner::DeflateRawDecoder(d) => {
|
|
d.write_all(input)?;
|
|
d.flush()?;
|
|
d.get_mut().drain(..)
|
|
}
|
|
Inner::DeflateRawEncoder(d) => {
|
|
d.write_all(input)?;
|
|
d.flush()?;
|
|
d.get_mut().drain(..)
|
|
}
|
|
Inner::GzDecoder(d) => {
|
|
d.write_all(input)?;
|
|
d.flush()?;
|
|
d.get_mut().drain(..)
|
|
}
|
|
Inner::GzEncoder(d) => {
|
|
d.write_all(input)?;
|
|
d.flush()?;
|
|
d.get_mut().drain(..)
|
|
}
|
|
}
|
|
.collect();
|
|
Ok(out.into())
|
|
}
|
|
|
|
#[op]
|
|
pub fn op_compression_finish(
|
|
state: &mut OpState,
|
|
rid: ResourceId,
|
|
) -> Result<ToJsBuffer, AnyError> {
|
|
let resource = state.resource_table.take::<CompressionResource>(rid)?;
|
|
let resource = Rc::try_unwrap(resource).unwrap();
|
|
let inner = resource.0.into_inner();
|
|
let out: Vec<u8> = match inner {
|
|
Inner::DeflateDecoder(d) => d.finish()?,
|
|
Inner::DeflateEncoder(d) => d.finish()?,
|
|
Inner::DeflateRawDecoder(d) => d.finish()?,
|
|
Inner::DeflateRawEncoder(d) => d.finish()?,
|
|
Inner::GzDecoder(d) => d.finish()?,
|
|
Inner::GzEncoder(d) => d.finish()?,
|
|
};
|
|
Ok(out.into())
|
|
}
|