use dom_struct::dom_struct;
use js::context::JSContext;
use js::gc::CustomAutoRooterGuard;
use js::typedarray::{ArrayBufferView, ArrayBufferViewU8};
use script_bindings::trace::RootedTraceableBox;
use crate::dom::bindings::buffer_source::HeapBufferSource;
use crate::dom::bindings::cell::DomRefCell;
use crate::dom::bindings::codegen::Bindings::ReadableStreamBYOBRequestBinding::ReadableStreamBYOBRequestMethods;
use crate::dom::bindings::error::{Error, Fallible};
use crate::dom::bindings::reflector::{Reflector, reflect_dom_object};
use crate::dom::bindings::root::{DomRoot, MutNullableDom};
use crate::dom::stream::readablebytestreamcontroller::ReadableByteStreamController;
use crate::dom::types::GlobalScope;
use crate::script_runtime::{CanGc, JSContext as SafeJSContext};
#[dom_struct]
pub(crate) struct ReadableStreamBYOBRequest {
reflector_: Reflector,
controller: MutNullableDom<ReadableByteStreamController>,
#[ignore_malloc_size_of = "mozjs"]
view: DomRefCell<HeapBufferSource<ArrayBufferViewU8>>,
}
impl ReadableStreamBYOBRequest {
fn new_inherited() -> ReadableStreamBYOBRequest {
ReadableStreamBYOBRequest {
reflector_: Reflector::new(),
controller: MutNullableDom::new(None),
view: DomRefCell::new(HeapBufferSource::<ArrayBufferViewU8>::default()),
}
}
pub(crate) fn new(global: &GlobalScope, can_gc: CanGc) -> DomRoot<ReadableStreamBYOBRequest> {
reflect_dom_object(Box::new(Self::new_inherited()), global, can_gc)
}
pub(crate) fn set_controller(&self, controller: Option<&ReadableByteStreamController>) {
self.controller.set(controller);
}
pub(crate) fn set_view(&self, view: Option<HeapBufferSource<ArrayBufferViewU8>>) {
match view {
Some(view) => {
*self.view.borrow_mut() = view;
},
None => {
*self.view.borrow_mut() = HeapBufferSource::<ArrayBufferViewU8>::default();
},
}
}
pub(crate) fn get_view(&self) -> HeapBufferSource<ArrayBufferViewU8> {
self.view.borrow().clone()
}
}
impl ReadableStreamBYOBRequestMethods<crate::DomTypeHolder> for ReadableStreamBYOBRequest {
fn GetView(
&self,
_cx: SafeJSContext,
) -> Option<RootedTraceableBox<js::typedarray::HeapArrayBufferView>> {
self.view.borrow().typed_array_to_option()
}
fn Respond(&self, cx: &mut JSContext, bytes_written: u64) -> Fallible<()> {
let controller = if let Some(controller) = self.controller.get() {
controller
} else {
return Err(Error::Type(c"controller is undefined".to_owned()));
};
{
let view = self.view.borrow();
if view
.get_array_buffer_view_buffer(cx.into())
.is_detached_buffer(cx.into())
{
return Err(Error::Type(c"buffer is detached".to_owned()));
}
assert!(view.byte_length() > 0);
assert!(view.viewed_buffer_array_byte_length(cx.into()) > 0);
}
controller.respond(cx, bytes_written)
}
fn RespondWithNewView(
&self,
cx: &mut JSContext,
view: CustomAutoRooterGuard<ArrayBufferView>,
) -> Fallible<()> {
let view = HeapBufferSource::<ArrayBufferViewU8>::from_view(view);
let controller = if let Some(controller) = self.controller.get() {
controller
} else {
return Err(Error::Type(c"controller is undefined".to_owned()));
};
if view.is_detached_buffer(cx.into()) {
return Err(Error::Type(c"buffer is detached".to_owned()));
}
controller.respond_with_new_view(cx, view)
}
}