Revert "Update js functions to new rquickjs version" (#2262)

This commit is contained in:
Tobie Morgan Hitchcock 2023-07-14 17:01:12 +01:00 committed by GitHub
parent 4f4339848e
commit 5c08be973d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
24 changed files with 1496 additions and 1360 deletions

389
Cargo.lock generated

File diff suppressed because it is too large Load diff

View file

@ -75,7 +75,7 @@ fuzzy-matcher = "0.3.7"
geo = { version = "0.25.1", features = ["use-serde"] } geo = { version = "0.25.1", features = ["use-serde"] }
indexmap = { version = "1.9.3", features = ["serde"] } indexmap = { version = "1.9.3", features = ["serde"] }
indxdb = { version = "0.3.0", optional = true } indxdb = { version = "0.3.0", optional = true }
js = { version = "0.4.0-beta.0" , package = "rquickjs", features = ["array-buffer", "bindgen", "classes", "futures", "loader", "macro", "parallel", "properties","rust-alloc"], optional = true } js = { version = "0.3.1" , package = "rquickjs", features = ["array-buffer", "bindgen", "classes", "futures", "loader", "macro", "parallel", "properties","rust-alloc"], optional = true }
jsonwebtoken = "8.3.0" jsonwebtoken = "8.3.0"
lexicmp = "0.1.0" lexicmp = "0.1.0"
lru = "0.10.1" lru = "0.10.1"

View file

@ -1,24 +1,28 @@
use js::class::Trace; #[js::bind(object, public)]
#[quickjs(bare)]
#[allow(non_snake_case)]
#[allow(unused_variables)]
#[allow(clippy::module_inception)]
pub mod duration {
use crate::sql::duration; use crate::sql::duration;
use crate::sql::value::Value;
use js::{class::Ref, function::Rest};
#[derive(Clone, Trace)] #[derive(Clone)]
#[js::class] #[quickjs(cloneable)]
pub struct Duration { pub struct Duration {
#[qjs(skip_trace)]
pub(crate) value: Option<duration::Duration>, pub(crate) value: Option<duration::Duration>,
} }
#[js::methods]
impl Duration { impl Duration {
#[qjs(constructor)] #[quickjs(constructor)]
pub fn new(value: String) -> Self { pub fn new(value: String, args: Rest<Value>) -> Self {
Self { Self {
value: duration::Duration::try_from(value).ok(), value: duration::Duration::try_from(value).ok(),
} }
} }
#[quickjs(get)]
#[qjs(get)]
pub fn value(&self) -> String { pub fn value(&self) -> String {
match &self.value { match &self.value {
Some(v) => v.to_raw(), Some(v) => v.to_raw(),
@ -26,23 +30,22 @@ impl Duration {
} }
} }
// Compare two Duration instances // Compare two Duration instances
pub fn is(a: &Duration, b: &Duration) -> bool { pub fn is(a: Ref<Duration>, b: Ref<Duration>, args: Rest<()>) -> bool {
a.value.is_some() && b.value.is_some() && a.value == b.value a.value.is_some() && b.value.is_some() && a.value == b.value
} }
/// Convert the object to a string /// Convert the object to a string
#[qjs(rename = "toString")] pub fn toString(&self, args: Rest<()>) -> String {
pub fn js_to_string(&self) -> String {
match &self.value { match &self.value {
Some(v) => v.to_raw(), Some(v) => v.to_raw(),
None => String::from("Invalid Duration"), None => String::from("Invalid Duration"),
} }
} }
/// Convert the object to JSON /// Convert the object to JSON
#[qjs(rename = "toJSON")] pub fn toJSON(&self, args: Rest<()>) -> String {
pub fn to_json(&self) -> String {
match &self.value { match &self.value {
Some(v) => v.to_raw(), Some(v) => v.to_raw(),
None => String::from("Invalid Duration"), None => String::from("Invalid Duration"),
} }
} }
} }
}

View file

@ -1,13 +1,13 @@
use js::{Class, Ctx, Result}; use js::{Ctx, Result};
pub mod duration; pub mod duration;
pub mod record; pub mod record;
pub mod uuid; pub mod uuid;
pub fn init(ctx: &Ctx<'_>) -> Result<()> { pub fn init(ctx: Ctx<'_>) -> Result<()> {
let globals = ctx.globals(); let globals = ctx.globals();
Class::<duration::Duration>::define(&globals)?; globals.init_def::<duration::Duration>()?;
Class::<record::Record>::define(&globals)?; globals.init_def::<record::Record>()?;
Class::<uuid::Uuid>::define(&globals)?; globals.init_def::<uuid::Uuid>()?;
Ok(()) Ok(())
} }

View file

@ -1,18 +1,23 @@
#[js::bind(object, public)]
#[quickjs(bare)]
#[allow(non_snake_case)]
#[allow(unused_variables)]
#[allow(clippy::module_inception)]
pub mod record {
use crate::sql::thing; use crate::sql::thing;
use crate::sql::value::Value; use crate::sql::value::Value;
use js::class::Trace; use js::{class::Ref, function::Rest};
#[derive(Clone, Trace)] #[derive(Clone)]
#[js::class] #[quickjs(cloneable)]
pub struct Record { pub struct Record {
#[qjs(skip_trace)]
pub(crate) value: thing::Thing, pub(crate) value: thing::Thing,
} }
#[js::methods]
impl Record { impl Record {
#[qjs(constructor)] #[quickjs(constructor)]
pub fn new(tb: String, id: Value) -> Self { pub fn new(tb: String, id: Value, args: Rest<()>) -> Self {
Self { Self {
value: thing::Thing { value: thing::Thing {
tb, tb,
@ -26,27 +31,26 @@ impl Record {
} }
} }
#[qjs(get)] #[quickjs(get)]
pub fn tb(&self) -> String { pub fn tb(&self) -> String {
self.value.tb.clone() self.value.tb.clone()
} }
#[qjs(get)] #[quickjs(get)]
pub fn id(&self) -> String { pub fn id(&self) -> String {
self.value.id.to_raw() self.value.id.to_raw()
} }
// Compare two Record instances // Compare two Record instances
pub fn is(a: &Record, b: &Record) -> bool { pub fn is<'js>(a: Ref<'js, Record>, b: Ref<'js, Record>, args: Rest<()>) -> bool {
a.value == b.value a.value == b.value
} }
/// Convert the object to a string /// Convert the object to a string
#[qjs(rename = "toString")] pub fn toString(&self, args: Rest<()>) -> String {
pub fn js_to_string(&self) -> String {
self.value.to_raw() self.value.to_raw()
} }
/// Convert the object to JSON /// Convert the object to JSON
#[qjs(rename = "toJSON")] pub fn toJSON(&self, args: Rest<()>) -> String {
pub fn to_json(&self) -> String {
self.value.to_raw() self.value.to_raw()
} }
} }
}

View file

@ -1,22 +1,28 @@
use crate::sql::uuid; #[js::bind(object, public)]
use js::class::Trace; #[quickjs(bare)]
#[allow(non_snake_case)]
#[allow(unused_variables)]
#[allow(clippy::module_inception)]
pub mod uuid {
#[derive(Clone, Trace)] use crate::sql::uuid;
#[js::class] use crate::sql::value::Value;
use js::{class::Ref, function::Rest};
#[derive(Clone)]
#[quickjs(cloneable)]
pub struct Uuid { pub struct Uuid {
#[qjs(skip_trace)]
pub(crate) value: Option<uuid::Uuid>, pub(crate) value: Option<uuid::Uuid>,
} }
#[js::methods]
impl Uuid { impl Uuid {
#[qjs(constructor)] #[quickjs(constructor)]
pub fn new(value: String) -> Self { pub fn new(value: String, args: Rest<Value>) -> Self {
Self { Self {
value: uuid::Uuid::try_from(value).ok(), value: uuid::Uuid::try_from(value).ok(),
} }
} }
#[qjs(get)] #[quickjs(get)]
pub fn value(&self) -> String { pub fn value(&self) -> String {
match &self.value { match &self.value {
Some(v) => v.to_raw(), Some(v) => v.to_raw(),
@ -24,23 +30,22 @@ impl Uuid {
} }
} }
// Compare two Uuid instances // Compare two Uuid instances
pub fn is(a: &Uuid, b: &Uuid) -> bool { pub fn is<'js>(a: Ref<'js, Uuid>, b: Ref<'js, Uuid>, _args: Rest<()>) -> bool {
a.value.is_some() && b.value.is_some() && a.value == b.value a.value.is_some() && b.value.is_some() && a.value == b.value
} }
/// Convert the object to a string /// Convert the object to a string
#[qjs(rename = "toString")] pub fn toString(&self, _args: Rest<()>) -> String {
pub fn js_to_string(&self) -> String {
match &self.value { match &self.value {
Some(v) => v.to_raw(), Some(v) => v.to_raw(),
None => String::from("Invalid Uuid"), None => String::from("Invalid Uuid"),
} }
} }
/// Convert the object to JSON /// Convert the object to JSON
#[qjs(rename = "toJSON")] pub fn toJSON(&self, _args: Rest<()>) -> String {
pub fn to_json(&self) -> String {
match &self.value { match &self.value {
Some(v) => v.to_raw(), Some(v) => v.to_raw(),
None => String::from("Invalid Uuid"), None => String::from("Invalid Uuid"),
} }
} }
} }
}

View file

@ -1,4 +1,4 @@
use crate::fnc::script::fetch::{stream::ReadableStream, RequestError}; use crate::fnc::script::fetch::{classes::BlobClass, stream::ReadableStream, RequestError};
use bytes::{Bytes, BytesMut}; use bytes::{Bytes, BytesMut};
use futures::{future, Stream, TryStreamExt}; use futures::{future, Stream, TryStreamExt};
use js::{ArrayBuffer, Class, Ctx, Error, Exception, FromJs, Result, Type, TypedArray, Value}; use js::{ArrayBuffer, Class, Ctx, Error, Exception, FromJs, Result, Type, TypedArray, Value};
@ -7,8 +7,6 @@ use std::{
result::Result as StdResult, result::Result as StdResult,
}; };
use super::classes::Blob;
pub type StreamItem = StdResult<Bytes, RequestError>; pub type StreamItem = StdResult<Bytes, RequestError>;
#[derive(Clone)] #[derive(Clone)]
@ -129,7 +127,7 @@ impl Body {
} }
impl<'js> FromJs<'js> for Body { impl<'js> FromJs<'js> for Body {
fn from_js(ctx: &Ctx<'js>, value: Value<'js>) -> Result<Self> { fn from_js(ctx: Ctx<'js>, value: Value<'js>) -> Result<Self> {
let object = match value.type_of() { let object = match value.type_of() {
Type::String => { Type::String => {
let string = value.as_string().unwrap().to_string()?; let string = value.as_string().unwrap().to_string()?;
@ -144,7 +142,7 @@ impl<'js> FromJs<'js> for Body {
}) })
} }
}; };
if let Some(x) = Class::<Blob>::from_object(object.clone()) { if let Ok(x) = Class::<BlobClass>::from_object(object.clone()) {
let borrow = x.borrow(); let borrow = x.borrow();
return Ok(Body::buffer(BodyKind::Blob(borrow.mime.clone()), borrow.data.clone())); return Ok(Body::buffer(BodyKind::Blob(borrow.mime.clone()), borrow.data.clone()));
} }
@ -196,7 +194,7 @@ impl<'js> FromJs<'js> for Body {
.ok_or_else(|| Exception::throw_type(ctx, "Buffer is already detached"))?; .ok_or_else(|| Exception::throw_type(ctx, "Buffer is already detached"))?;
return Ok(Body::buffer(BodyKind::Buffer, Bytes::copy_from_slice(bytes))); return Ok(Body::buffer(BodyKind::Buffer, Bytes::copy_from_slice(bytes)));
} }
if let Some(x) = ArrayBuffer::from_object(object.clone()) { if let Ok(x) = ArrayBuffer::from_object(object.clone()) {
let bytes = x let bytes = x
.as_bytes() .as_bytes()
.ok_or_else(|| Exception::throw_type(ctx, "Buffer is already detached"))?; .ok_or_else(|| Exception::throw_type(ctx, "Buffer is already detached"))?;

View file

@ -1,11 +1,9 @@
//! Blob class implementation //! Blob class implementation
use bytes::{Bytes, BytesMut}; use bytes::BytesMut;
use js::{ use js::{bind, prelude::Coerced, ArrayBuffer, Class, Ctx, Exception, FromJs, Result, Value};
class::Trace,
prelude::{Coerced, Opt}, pub use blob::Blob as BlobClass;
ArrayBuffer, Class, Ctx, Exception, FromJs, Object, Result, Value,
};
#[derive(Clone, Copy)] #[derive(Clone, Copy)]
pub enum EndingType { pub enum EndingType {
@ -14,7 +12,7 @@ pub enum EndingType {
} }
fn append_blob_part<'js>( fn append_blob_part<'js>(
ctx: &Ctx<'js>, ctx: Ctx<'js>,
value: Value<'js>, value: Value<'js>,
ending: EndingType, ending: EndingType,
data: &mut BytesMut, data: &mut BytesMut,
@ -25,11 +23,11 @@ fn append_blob_part<'js>(
const LINE_ENDING: &[u8] = b"\n"; const LINE_ENDING: &[u8] = b"\n";
if let Some(object) = value.as_object() { if let Some(object) = value.as_object() {
if let Some(x) = Class::<Blob>::from_object(object.clone()) { if let Ok(x) = Class::<BlobClass>::from_object(object.clone()) {
data.extend_from_slice(&x.borrow().data); data.extend_from_slice(&x.borrow().data);
return Ok(()); return Ok(());
} }
if let Some(x) = ArrayBuffer::from_object(object.clone()) { if let Ok(x) = ArrayBuffer::from_object(object.clone()) {
data.extend_from_slice(x.as_bytes().ok_or_else(|| { data.extend_from_slice(x.as_bytes().ok_or_else(|| {
Exception::throw_type(ctx, "Tried to construct blob with detached buffer") Exception::throw_type(ctx, "Tried to construct blob with detached buffer")
})?); })?);
@ -76,26 +74,39 @@ fn normalize_type(mut ty: String) -> String {
} }
} }
#[derive(Clone, Trace)] #[bind(object, public)]
#[js::class] #[quickjs(bare)]
#[allow(non_snake_case)]
#[allow(unused_variables)]
#[allow(clippy::module_inception)]
mod blob {
use super::*;
use bytes::{Bytes, BytesMut};
use js::{
function::{Opt, Rest},
ArrayBuffer, Ctx, Exception, Object, Result, Value,
};
#[derive(Clone)]
#[quickjs(cloneable)]
pub struct Blob { pub struct Blob {
pub(crate) mime: String, pub(crate) mime: String,
// TODO: make bytes? // TODO: make bytes?
#[qjs(skip_trace)]
pub(crate) data: Bytes, pub(crate) data: Bytes,
} }
#[js::methods]
impl Blob { impl Blob {
// ------------------------------ // ------------------------------
// Constructor // Constructor
// ------------------------------ // ------------------------------
#[qjs(constructor)] #[quickjs(constructor)]
pub fn new<'js>( pub fn new<'js>(
ctx: Ctx<'js>, ctx: Ctx<'js>,
parts: Opt<Value<'js>>, parts: Opt<Value<'js>>,
options: Opt<Object<'js>>, options: Opt<Object<'js>>,
_rest: Rest<()>,
) -> Result<Self> { ) -> Result<Self> {
let mut r#type = String::new(); let mut r#type = String::new();
let mut endings = EndingType::Transparent; let mut endings = EndingType::Transparent;
@ -109,7 +120,7 @@ impl Blob {
endings = EndingType::Native; endings = EndingType::Native;
} else if x != "transparent" { } else if x != "transparent" {
return Err(Exception::throw_type( return Err(Exception::throw_type(
&ctx, ctx,
",expected endings to be either 'transparent' or 'native'", ",expected endings to be either 'transparent' or 'native'",
)); ));
} }
@ -119,13 +130,13 @@ impl Blob {
let data = if let Some(parts) = parts.into_inner() { let data = if let Some(parts) = parts.into_inner() {
let array = parts let array = parts
.into_array() .into_array()
.ok_or_else(|| Exception::throw_type(&ctx, "Blob parts are not a sequence"))?; .ok_or_else(|| Exception::throw_type(ctx, "Blob parts are not a sequence"))?;
let mut buffer = BytesMut::new(); let mut buffer = BytesMut::new();
for elem in array.iter::<Value>() { for elem in array.iter::<Value>() {
let elem = elem?; let elem = elem?;
append_blob_part(&ctx, elem, endings, &mut buffer)?; append_blob_part(ctx, elem, endings, &mut buffer)?;
} }
buffer.freeze() buffer.freeze()
} else { } else {
@ -141,17 +152,24 @@ impl Blob {
// Instance properties // Instance properties
// ------------------------------ // ------------------------------
#[qjs(get)] #[quickjs(get)]
pub fn size(&self) -> usize { pub fn size(&self) -> usize {
self.data.len() self.data.len()
} }
#[qjs(get, rename = "type")] #[quickjs(get)]
#[quickjs(rename = "type")]
pub fn r#type(&self) -> String { pub fn r#type(&self) -> String {
self.mime.clone() self.mime.clone()
} }
pub fn slice(&self, start: Opt<isize>, end: Opt<isize>, content_type: Opt<String>) -> Blob { pub fn slice(
&self,
start: Opt<isize>,
end: Opt<isize>,
content_type: Opt<String>,
_rest: Rest<()>,
) -> Blob {
// see https://w3c.github.io/FileAPI/#slice-blob // see https://w3c.github.io/FileAPI/#slice-blob
let start = start.into_inner().unwrap_or_default(); let start = start.into_inner().unwrap_or_default();
let start = if start < 0 { let start = if start < 0 {
@ -173,13 +191,16 @@ impl Blob {
} }
} }
pub async fn text(&self) -> Result<String> { pub async fn text(&self, _rest: Rest<()>) -> Result<String> {
let text = String::from_utf8(self.data.to_vec())?; let text = String::from_utf8(self.data.to_vec())?;
Ok(text) Ok(text)
} }
#[qjs(rename = "arrayBuffer")] pub async fn arrayBuffer<'js>(
pub async fn array_buffer<'js>(&self, ctx: Ctx<'js>) -> Result<ArrayBuffer<'js>> { &self,
ctx: Ctx<'js>,
_rest: Rest<()>,
) -> Result<ArrayBuffer<'js>> {
ArrayBuffer::new(ctx, self.data.to_vec()) ArrayBuffer::new(ctx, self.data.to_vec())
} }
@ -188,11 +209,11 @@ impl Blob {
// ------------------------------ // ------------------------------
// Convert the object to a string // Convert the object to a string
#[qjs(rename = "toString")] pub fn toString(&self, _rest: Rest<()>) -> String {
pub fn js_to_string(&self) -> String {
String::from("[object Blob]") String::from("[object Blob]")
} }
} }
}
#[cfg(test)] #[cfg(test)]
mod test { mod test {
@ -231,7 +252,7 @@ mod test {
assert.mustThrow(() => new Blob("text")); assert.mustThrow(() => new Blob("text"));
assert.mustThrow(() => new Blob(["text"], {endings: "invalid value"})); assert.mustThrow(() => new Blob(["text"], {endings: "invalid value"}));
})() })()
"#).catch(&ctx).unwrap().await.catch(&ctx).unwrap(); "#).catch(ctx).unwrap().await.catch(ctx).unwrap();
}) })
.await .await
} }

View file

@ -1,36 +1,34 @@
//! FormData class implementation //! FormData class implementation
use js::{ use js::{
class::{Class, Trace}, bind, function::Opt, prelude::Coerced, Class, Ctx, Exception, FromJs, Persistent, Result,
function::{Opt, Rest}, String, Value,
prelude::Coerced,
Ctx, Exception, FromJs, Result, String, Value,
}; };
use reqwest::multipart::{Form, Part}; use std::string::String as StdString;
use std::{collections::HashMap, string::String as StdString};
use crate::fnc::script::fetch::classes::Blob; use crate::fnc::script::fetch::classes::BlobClass;
#[derive(Clone)] #[derive(Clone)]
pub enum FormDataValue<'js> { pub enum FormDataValue {
String(String<'js>), String(Persistent<String<'static>>),
Blob { Blob {
data: Class<'js, Blob>, data: Persistent<Class<'static, BlobClass>>,
filename: Option<String<'js>>, filename: Option<Persistent<String<'static>>>,
}, },
} }
impl<'js> FormDataValue<'js> { impl FormDataValue {
fn from_arguments( fn from_arguments<'js>(
ctx: &Ctx<'js>, ctx: Ctx<'js>,
value: Value<'js>, value: Value<'js>,
filename: Opt<Coerced<String<'js>>>, filename: Opt<Coerced<String<'js>>>,
error: &'static str, error: &'static str,
) -> Result<Self> { ) -> Result<FormDataValue> {
if let Some(blob) = if let Some(blob) =
value.as_object().and_then(|value| Class::<Blob>::from_object(value.clone())) value.as_object().and_then(|value| Class::<BlobClass>::from_object(value.clone()).ok())
{ {
let filename = filename.into_inner().map(|x| x.0); let blob = Persistent::save(ctx, blob);
let filename = filename.into_inner().map(|x| Persistent::save(ctx, x.0));
Ok(FormDataValue::Blob { Ok(FormDataValue::Blob {
data: blob, data: blob,
@ -40,20 +38,36 @@ impl<'js> FormDataValue<'js> {
return Err(Exception::throw_type(ctx, error)); return Err(Exception::throw_type(ctx, error));
} else { } else {
let value = Coerced::<String>::from_js(ctx, value)?; let value = Coerced::<String>::from_js(ctx, value)?;
Ok(FormDataValue::String(value.0)) let value = Persistent::save(ctx, value.0);
Ok(FormDataValue::String(value))
} }
} }
} }
#[js::class] pub use form_data::FormData as FormDataClass;
#[derive(Clone, Trace)] #[bind(object, public)]
pub struct FormData<'js> { #[quickjs(bare)]
#[qjs(skip_trace)] #[allow(non_snake_case)]
pub(crate) values: HashMap<StdString, Vec<FormDataValue<'js>>>, #[allow(unused_variables)]
#[allow(clippy::module_inception)]
pub mod form_data {
use super::*;
use std::{cell::RefCell, collections::HashMap};
use js::{
function::Opt,
prelude::{Coerced, Rest},
Ctx, Result, String, Value,
};
use reqwest::multipart::{Form, Part};
#[derive(Clone)]
#[quickjs(cloneable)]
pub struct FormData {
pub(crate) values: RefCell<HashMap<StdString, Vec<FormDataValue>>>,
} }
#[js::methods] impl FormData {
impl<'js> FormData<'js> {
// ------------------------------ // ------------------------------
// Constructor // Constructor
// ------------------------------ // ------------------------------
@ -61,81 +75,83 @@ impl<'js> FormData<'js> {
// FormData spec states that FormDa takes two html elements as arguments // FormData spec states that FormDa takes two html elements as arguments
// which does not make sense implementing fetch outside a browser. // which does not make sense implementing fetch outside a browser.
// So we ignore those arguments. // So we ignore those arguments.
#[qjs(constructor)] #[quickjs(constructor)]
pub fn new(ctx: Ctx<'js>, args: Rest<()>) -> Result<Self> { pub fn new(ctx: Ctx<'_>, args: Rest<()>) -> Result<FormData> {
if args.len() > 0 { if args.len() > 0 {
return Err(Exception::throw_internal( return Err(Exception::throw_internal(ctx,"Cant call FormData with arguments as the dom elements required are not available"));
&ctx,
"Cant call FormData with arguments as the dom elements required are not available",
));
} }
Ok(FormData { Ok(FormData {
values: HashMap::new(), values: RefCell::new(HashMap::new()),
}) })
} }
pub fn append( pub fn append<'js>(
&mut self, &self,
ctx: Ctx<'js>, ctx: Ctx<'js>,
name: Coerced<StdString>, name: Coerced<StdString>,
value: Value<'js>, value: Value<'js>,
filename: Opt<Coerced<String<'js>>>, filename: Opt<Coerced<String<'js>>>,
) -> Result<()> { ) -> Result<()> {
let value = FormDataValue::from_arguments( let value = FormDataValue::from_arguments(
&ctx, ctx,
value, value,
filename, filename,
"Can't call `append` on `FormData` with a filename when value isn't of type `Blob`", "Can't call `append` on `FormData` with a filename when value isn't of type `Blob`",
)?; )?;
self.values.entry(name.0).or_insert_with(Vec::new).push(value); self.values.borrow_mut().entry(name.0).or_insert_with(Vec::new).push(value);
Ok(()) Ok(())
} }
pub fn set( pub fn set<'js>(
&mut self, &self,
ctx: Ctx<'js>, ctx: Ctx<'js>,
name: Coerced<StdString>, name: Coerced<StdString>,
value: Value<'js>, value: Value<'js>,
filename: Opt<Coerced<String<'js>>>, filename: Opt<Coerced<String<'js>>>,
) -> Result<()> { ) -> Result<()> {
let value = FormDataValue::from_arguments( let value = FormDataValue::from_arguments(
&ctx, ctx,
value, value,
filename, filename,
"Can't call `set` on `FormData` with a filename when value isn't of type `Blob`", "Can't call `set` on `FormData` with a filename when value isn't of type `Blob`",
)?; )?;
self.values.insert(name.0, vec![value]); self.values.borrow_mut().insert(name.0, vec![value]);
Ok(()) Ok(())
} }
pub fn has(&self, name: Coerced<StdString>) -> bool { pub fn has(&self, ctx: Ctx<'_>, name: Coerced<StdString>) -> bool {
self.values.contains_key(&name.0) self.values.borrow().contains_key(&name.0)
} }
pub fn delete(&mut self, name: Coerced<StdString>) { pub fn delete(&self, ctx: Ctx<'_>, name: Coerced<StdString>) {
self.values.remove(&name.0); self.values.borrow_mut().remove(&name.0);
} }
#[qjs(skip)] #[quickjs(skip)]
pub fn to_form(&self) -> Result<Form> { pub fn to_form(&self, ctx: Ctx<'_>) -> Result<Form> {
let lock = self.values.borrow();
let mut res = Form::new(); let mut res = Form::new();
for (k, v) in self.values.iter() { for (k, v) in lock.iter() {
for v in v { for v in v {
match v { match v {
FormDataValue::String(x) => { FormDataValue::String(x) => {
let x = x.clone().restore(ctx).unwrap();
res = res.text(k.clone(), x.to_string()?); res = res.text(k.clone(), x.to_string()?);
} }
FormDataValue::Blob { FormDataValue::Blob {
data, data,
filename, filename,
} => { } => {
let mut part = Part::bytes(data.borrow().data.to_vec()); let mut part = Part::bytes(
data.clone().restore(ctx).unwrap().borrow().data.to_vec(),
);
if let Some(filename) = filename { if let Some(filename) = filename {
let filename = filename.to_string()?; let filename =
filename.clone().restore(ctx).unwrap().to_string()?;
part = part.file_name(filename); part = part.file_name(filename);
} }
res = res.part(k.clone(), part); res = res.part(k.clone(), part);
@ -146,3 +162,4 @@ impl<'js> FormData<'js> {
Ok(res) Ok(res)
} }
} }
}

View file

@ -1,30 +1,35 @@
//! Headers class implementation //! Headers class implementation
use std::str::FromStr; use js::bind;
use js::{ pub use headers::Headers as HeadersClass;
class::Trace,
prelude::{Coerced, List}, #[bind(object, public)]
Array, Ctx, Exception, Result, Value, #[quickjs(bare)]
}; #[allow(non_snake_case)]
#[allow(unused_variables)]
#[allow(clippy::module_inception)]
mod headers {
use std::{cell::RefCell, str::FromStr};
use js::{function::Rest, prelude::Coerced, Array, Ctx, Exception, Result, Value};
use reqwest::header::{HeaderMap, HeaderName, HeaderValue}; use reqwest::header::{HeaderMap, HeaderName, HeaderValue};
#[derive(Clone, Trace)] #[derive(Clone)]
#[js::class] #[quickjs(cloneable)]
#[allow(dead_code)]
pub struct Headers { pub struct Headers {
#[qjs(skip_trace)] pub(crate) inner: RefCell<HeaderMap>,
pub(crate) inner: HeaderMap,
} }
#[js::methods]
impl Headers { impl Headers {
// ------------------------------ // ------------------------------
// Constructor // Constructor
// ------------------------------ // ------------------------------
#[qjs(constructor)] #[quickjs(constructor)]
pub fn new<'js>(ctx: Ctx<'js>, init: Value<'js>) -> Result<Self> { pub fn new<'js>(ctx: Ctx<'js>, init: Value<'js>, args: Rest<()>) -> Result<Self> {
Headers::new_inner(&ctx, init) Headers::new_inner(ctx, init)
} }
// ------------------------------ // ------------------------------
@ -32,41 +37,41 @@ impl Headers {
// ------------------------------ // ------------------------------
// Convert the object to a string // Convert the object to a string
#[qjs(rename = "toString")] pub fn toString(&self, args: Rest<()>) -> String {
pub fn js_to_string(&self) -> String {
String::from("[object Header]") String::from("[object Header]")
} }
// Adds or appends a new value to a header // Adds or appends a new value to a header
pub fn append(&mut self, ctx: Ctx<'_>, key: String, val: String) -> Result<()> { pub fn append(&self, ctx: Ctx<'_>, key: String, val: String, args: Rest<()>) -> Result<()> {
self.append_inner(&ctx, &key, &val) self.append_inner(ctx, &key, &val)
} }
// Deletes a header from the header set // Deletes a header from the header set
pub fn delete(&mut self, ctx: Ctx<'_>, key: String) -> Result<()> { pub fn delete(&self, ctx: Ctx<'_>, key: String, args: Rest<()>) -> Result<()> {
// Process and check the header name is valid // Process and check the header name is valid
let key = let key = HeaderName::from_str(&key)
HeaderName::from_str(&key).map_err(|e| Exception::throw_type(&ctx, &format!("{e}")))?; .map_err(|e| Exception::throw_type(ctx, &format!("{e}")))?;
// Remove the header entry from the map // Remove the header entry from the map
self.inner.remove(&key); self.inner.borrow_mut().remove(&key);
// Everything ok // Everything ok
Ok(()) Ok(())
} }
// Returns all header entries in the header set // Returns all header entries in the header set
pub fn entries(&self) -> Vec<List<(String, String)>> { pub fn entries(&self, args: Rest<()>) -> Vec<(String, String)> {
let mut res = Vec::<List<(String, String)>>::with_capacity(self.inner.len()); let lock = self.inner.borrow();
let mut res = Vec::<(String, String)>::with_capacity(lock.len());
for (k, v) in self.inner.iter() { for (k, v) in lock.iter() {
let k = k.as_str(); let k = k.as_str();
if Some(k) == res.last().map(|x| x.0 .0.as_str()) { if Some(k) == res.last().map(|x| x.0.as_str()) {
let ent = res.last_mut().unwrap(); let ent = res.last_mut().unwrap();
ent.0 .1.push_str(", "); ent.1.push_str(", ");
// Header value came from a string, so it should also be able to be cast back // Header value came from a string, so it should also be able to be cast back
// to a string // to a string
ent.0 .1.push_str(v.to_str().unwrap()); ent.1.push_str(v.to_str().unwrap());
} else { } else {
res.push(List((k.to_owned(), v.to_str().unwrap().to_owned()))); res.push((k.to_owned(), v.to_str().unwrap().to_owned()));
} }
} }
@ -74,12 +79,13 @@ impl Headers {
} }
// Returns all values of a header in the header set // Returns all values of a header in the header set
pub fn get(&self, ctx: Ctx<'_>, key: String) -> Result<Option<String>> { pub fn get(&self, ctx: Ctx<'_>, key: String, args: Rest<()>) -> Result<Option<String>> {
// Process and check the header name is valid // Process and check the header name is valid
let key = let key = HeaderName::from_str(&key)
HeaderName::from_str(&key).map_err(|e| Exception::throw_type(&ctx, &format!("{e}")))?; .map_err(|e| Exception::throw_type(ctx, &format!("{e}")))?;
// Convert the header values to strings // Convert the header values to strings
let all = self.inner.get_all(&key); let lock = self.inner.borrow();
let all = lock.get_all(&key);
// Header value came from a string, so it should also be able to be cast back // Header value came from a string, so it should also be able to be cast back
// to a string // to a string
@ -98,48 +104,54 @@ impl Headers {
} }
// Returns all values for the `Set-Cookie` header. // Returns all values for the `Set-Cookie` header.
#[qjs(rename = "getSetCookie")] #[quickjs(rename = "getSetCookie")]
pub fn get_set_cookie(&self) -> Vec<String> { pub fn get_set_cookie(&self, args: Rest<()>) -> Vec<String> {
// This should always be a correct cookie; // This should always be a correct cookie;
let key = HeaderName::from_str("set-cookie").unwrap(); let key = HeaderName::from_str("set-cookie").unwrap();
self.inner.get_all(key).iter().map(|x| x.to_str().unwrap().to_owned()).collect() self.inner
.borrow()
.get_all(key)
.iter()
.map(|x| x.to_str().unwrap().to_owned())
.collect()
} }
// Checks to see if the header set contains a header // Checks to see if the header set contains a header
pub fn has(&self, ctx: Ctx<'_>, key: String) -> Result<bool> { pub fn has(&self, ctx: Ctx<'_>, key: String, args: Rest<()>) -> Result<bool> {
// Process and check the header name is valid // Process and check the header name is valid
let key = let key = HeaderName::from_str(&key)
HeaderName::from_str(&key).map_err(|e| Exception::throw_type(&ctx, &format!("{e}")))?; .map_err(|e| Exception::throw_type(ctx, &format!("{e}")))?;
// Check if the header entry exists // Check if the header entry exists
Ok(self.inner.contains_key(&key)) Ok(self.inner.borrow().contains_key(&key))
} }
// Returns all header keys contained in the header set // Returns all header keys contained in the header set
pub fn keys(&self) -> Vec<String> { pub fn keys(&self, args: Rest<()>) -> Vec<String> {
// TODO: Incorrect, should return an iterator but iterators are not supported yet by quickjs // TODO: Incorrect, should return an iterator but iterators are not supported yet by quickjs
self.inner.keys().map(|v| v.as_str().to_owned()).collect::<Vec<String>>() self.inner.borrow().keys().map(|v| v.as_str().to_owned()).collect::<Vec<String>>()
} }
// Sets a new value or adds a header to the header set // Sets a new value or adds a header to the header set
pub fn set(&mut self, ctx: Ctx<'_>, key: String, val: String) -> Result<()> { pub fn set(&self, ctx: Ctx<'_>, key: String, val: String, args: Rest<()>) -> Result<()> {
// Process and check the header name is valid // Process and check the header name is valid
let key = HeaderName::from_str(&key) let key = HeaderName::from_str(&key)
.map_err(|e| Exception::throw_type(&ctx, &format!("Invalid header name: {e}")))?; .map_err(|e| Exception::throw_type(ctx, &format!("Invalid header name: {e}")))?;
// Process and check the header name is valid // Process and check the header name is valid
let val = HeaderValue::from_str(&val) let val = HeaderValue::from_str(&val)
.map_err(|e| Exception::throw_type(&ctx, &format!("Invalid header value: {e}")))?; .map_err(|e| Exception::throw_type(ctx, &format!("Invalid header value: {e}")))?;
// Insert and overwrite the header entry // Insert and overwrite the header entry
self.inner.insert(key, val); self.inner.borrow_mut().insert(key, val);
// Everything ok // Everything ok
Ok(()) Ok(())
} }
// Returns all header values contained in the header set // Returns all header values contained in the header set
pub fn values(&self) -> Vec<String> { pub fn values(&self, args: Rest<()>) -> Vec<String> {
let mut res = Vec::<String>::with_capacity(self.inner.len()); let lock = self.inner.borrow();
let mut res = Vec::<String>::with_capacity(lock.len());
let mut pref = None; let mut pref = None;
for (k, v) in self.inner.iter() { for (k, v) in lock.iter() {
if Some(k) == pref { if Some(k) == pref {
let ent = res.last_mut().unwrap(); let ent = res.last_mut().unwrap();
ent.push_str(", "); ent.push_str(", ");
@ -154,10 +166,11 @@ impl Headers {
} }
} }
#[quickjs(skip)]
impl Headers { impl Headers {
pub fn from_map(map: HeaderMap) -> Self { pub fn from_map(map: HeaderMap) -> Self {
Self { Self {
inner: map, inner: RefCell::new(map),
} }
} }
@ -165,9 +178,9 @@ impl Headers {
Self::from_map(HeaderMap::new()) Self::from_map(HeaderMap::new())
} }
pub fn new_inner<'js>(ctx: &Ctx<'js>, val: Value<'js>) -> Result<Self> { pub fn new_inner<'js>(ctx: Ctx<'js>, val: Value<'js>) -> Result<Self> {
static INVALID_ERROR: &str = "Headers constructor: init was neither sequence<sequence<ByteString>> or record<ByteString, ByteString>"; static INVALID_ERROR: &str = "Headers constructor: init was neither sequence<sequence<ByteString>> or record<ByteString, ByteString>";
let mut res = Self::new_empty(); let res = Self::new_empty();
// TODO Set and Map, // TODO Set and Map,
if let Some(array) = val.as_array() { if let Some(array) = val.as_array() {
@ -223,7 +236,7 @@ impl Headers {
Ok(res) Ok(res)
} }
fn append_inner(&mut self, ctx: &Ctx<'_>, key: &str, val: &str) -> Result<()> { fn append_inner(&self, ctx: Ctx<'_>, key: &str, val: &str) -> Result<()> {
// Unsure what to do exactly here. // Unsure what to do exactly here.
// Spec dictates normalizing string before adding it as a header value, i.e. removing // Spec dictates normalizing string before adding it as a header value, i.e. removing
// any leading and trailing whitespace: // any leading and trailing whitespace:
@ -250,11 +263,12 @@ impl Headers {
} }
}; };
self.inner.append(key, val); self.inner.borrow_mut().append(key, val);
Ok(()) Ok(())
} }
} }
}
#[cfg(test)] #[cfg(test)]
mod test { mod test {
@ -313,7 +327,7 @@ mod test {
}); });
assert.seq(headers.get("f"), "g"); assert.seq(headers.get("f"), "g");
assert.seq(headers.get("h"), "j"); assert.seq(headers.get("h"), "j");
"#).catch(&ctx).unwrap(); "#).catch(ctx).unwrap();
}) })
.await .await
} }

View file

@ -1,9 +1,18 @@
//! Request class implementation //! Request class implementation
//!
use js::{class::Trace, prelude::Coerced, Class, Ctx, Exception, FromJs, Object, Result, Value}; use js::{
bind,
class::{HasRefs, RefsMarker},
prelude::Coerced,
Class, Ctx, Exception, FromJs, Object, Persistent, Result, Value,
};
use reqwest::Method; use reqwest::Method;
use crate::fnc::script::fetch::{body::Body, RequestError}; use crate::fnc::script::fetch::{
body::Body,
classes::{BlobClass, HeadersClass},
RequestError,
};
#[derive(Clone, Copy, Eq, PartialEq)] #[derive(Clone, Copy, Eq, PartialEq)]
pub enum RequestMode { pub enum RequestMode {
@ -14,7 +23,7 @@ pub enum RequestMode {
} }
impl<'js> FromJs<'js> for RequestMode { impl<'js> FromJs<'js> for RequestMode {
fn from_js(ctx: &Ctx<'js>, value: Value<'js>) -> Result<Self> { fn from_js(ctx: Ctx<'js>, value: Value<'js>) -> Result<Self> {
let res = if let Some(Coerced(x)) = <Option<Coerced<String>>>::from_js(ctx, value)? { let res = if let Some(Coerced(x)) = <Option<Coerced<String>>>::from_js(ctx, value)? {
match x.as_str() { match x.as_str() {
"navigate" => RequestMode::Navigate, "navigate" => RequestMode::Navigate,
@ -50,7 +59,7 @@ pub enum RequestCredentials {
} }
impl<'js> FromJs<'js> for RequestCredentials { impl<'js> FromJs<'js> for RequestCredentials {
fn from_js(ctx: &Ctx<'js>, value: Value<'js>) -> Result<Self> { fn from_js(ctx: Ctx<'js>, value: Value<'js>) -> Result<Self> {
let res = if let Some(Coerced(x)) = <Option<Coerced<String>>>::from_js(ctx, value)? { let res = if let Some(Coerced(x)) = <Option<Coerced<String>>>::from_js(ctx, value)? {
match x.as_str() { match x.as_str() {
"omit" => RequestCredentials::Omit, "omit" => RequestCredentials::Omit,
@ -87,7 +96,7 @@ pub enum RequestCache {
} }
impl<'js> FromJs<'js> for RequestCache { impl<'js> FromJs<'js> for RequestCache {
fn from_js(ctx: &Ctx<'js>, value: Value<'js>) -> Result<Self> { fn from_js(ctx: Ctx<'js>, value: Value<'js>) -> Result<Self> {
let res = if let Some(Coerced(x)) = <Option<Coerced<String>>>::from_js(ctx, value)? { let res = if let Some(Coerced(x)) = <Option<Coerced<String>>>::from_js(ctx, value)? {
match x.as_str() { match x.as_str() {
"default" => RequestCache::Default, "default" => RequestCache::Default,
@ -127,7 +136,7 @@ pub enum RequestRedirect {
} }
impl<'js> FromJs<'js> for RequestRedirect { impl<'js> FromJs<'js> for RequestRedirect {
fn from_js(ctx: &Ctx<'js>, value: Value<'js>) -> Result<Self> { fn from_js(ctx: Ctx<'js>, value: Value<'js>) -> Result<Self> {
let res = if let Some(Coerced(x)) = <Option<Coerced<String>>>::from_js(ctx, value)? { let res = if let Some(Coerced(x)) = <Option<Coerced<String>>>::from_js(ctx, value)? {
match x.as_str() { match x.as_str() {
"follow" => RequestRedirect::Follow, "follow" => RequestRedirect::Follow,
@ -167,7 +176,7 @@ pub enum ReferrerPolicy {
} }
impl<'js> FromJs<'js> for ReferrerPolicy { impl<'js> FromJs<'js> for ReferrerPolicy {
fn from_js(ctx: &Ctx<'js>, value: Value<'js>) -> Result<Self> { fn from_js(ctx: Ctx<'js>, value: Value<'js>) -> Result<Self> {
let res = if let Some(Coerced(x)) = <Option<Coerced<String>>>::from_js(ctx, value)? { let res = if let Some(Coerced(x)) = <Option<Coerced<String>>>::from_js(ctx, value)? {
match x.as_str() { match x.as_str() {
"" => ReferrerPolicy::Empty, "" => ReferrerPolicy::Empty,
@ -204,9 +213,9 @@ impl<'js> FromJs<'js> for ReferrerPolicy {
} }
} }
pub struct RequestInit<'js> { pub struct RequestInit {
pub method: Method, pub method: Method,
pub headers: Class<'js, Headers>, pub headers: Persistent<Class<'static, HeadersClass>>,
pub body: Option<Body>, pub body: Option<Body>,
pub referrer: String, pub referrer: String,
pub referrer_policy: ReferrerPolicy, pub referrer_policy: ReferrerPolicy,
@ -218,15 +227,15 @@ pub struct RequestInit<'js> {
pub keep_alive: bool, pub keep_alive: bool,
} }
impl<'js> Trace<'js> for RequestInit<'js> { impl HasRefs for RequestInit {
fn trace<'a>(&self, tracer: js::class::Tracer<'a, 'js>) { fn mark_refs(&self, marker: &RefsMarker) {
self.headers.trace(tracer); self.headers.mark_refs(marker);
} }
} }
impl<'js> RequestInit<'js> { impl RequestInit {
pub fn default(ctx: Ctx<'js>) -> Result<Self> { pub fn default(ctx: Ctx<'_>) -> Result<Self> {
let headers = Class::instance(ctx, Headers::new_empty())?; let headers = Persistent::save(ctx, Class::instance(ctx, HeadersClass::new_empty())?);
Ok(RequestInit { Ok(RequestInit {
method: Method::GET, method: Method::GET,
headers, headers,
@ -242,9 +251,9 @@ impl<'js> RequestInit<'js> {
}) })
} }
pub fn clone_js(&self, ctx: Ctx<'js>) -> Result<Self> { pub fn clone_js(&self, ctx: Ctx<'_>) -> Result<Self> {
let headers = self.headers.clone(); let headers = self.headers.clone().restore(ctx).unwrap();
let headers = Class::instance(ctx.clone(), headers.borrow().clone())?; let headers = Persistent::save(ctx, Class::instance(ctx, headers.borrow().clone())?);
let body = self.body.as_ref().map(|x| x.clone_js(ctx)); let body = self.body.as_ref().map(|x| x.clone_js(ctx));
@ -265,7 +274,7 @@ impl<'js> RequestInit<'js> {
} }
// Normalize method string according to spec. // Normalize method string according to spec.
fn normalize_method(ctx: &Ctx<'_>, m: String) -> Result<Method> { fn normalize_method(ctx: Ctx<'_>, m: String) -> Result<Method> {
if m.as_bytes().eq_ignore_ascii_case(b"CONNECT") if m.as_bytes().eq_ignore_ascii_case(b"CONNECT")
|| m.as_bytes().eq_ignore_ascii_case(b"TRACE") || m.as_bytes().eq_ignore_ascii_case(b"TRACE")
|| m.as_bytes().eq_ignore_ascii_case(b"TRACK") || m.as_bytes().eq_ignore_ascii_case(b"TRACK")
@ -300,8 +309,8 @@ fn normalize_method(ctx: &Ctx<'_>, m: String) -> Result<Method> {
} }
} }
impl<'js> FromJs<'js> for RequestInit<'js> { impl<'js> FromJs<'js> for RequestInit {
fn from_js(ctx: &Ctx<'js>, value: Value<'js>) -> Result<Self> { fn from_js(ctx: Ctx<'js>, value: Value<'js>) -> Result<Self> {
let object = Object::from_js(ctx, value)?; let object = Object::from_js(ctx, value)?;
let referrer = object let referrer = object
@ -337,14 +346,15 @@ impl<'js> FromJs<'js> for RequestInit<'js> {
} }
let headers = if let Some(hdrs) = object.get::<_, Option<Object>>("headers")? { let headers = if let Some(hdrs) = object.get::<_, Option<Object>>("headers")? {
if let Some(cls) = Class::<Headers>::from_object(hdrs.clone()) { if let Ok(cls) = Class::<HeadersClass>::from_object(hdrs.clone()) {
cls cls
} else { } else {
Class::instance(ctx.clone(), Headers::new_inner(ctx, hdrs.into_value())?)? Class::instance(ctx, HeadersClass::new_inner(ctx, hdrs.into_value())?)?
} }
} else { } else {
Class::instance(ctx.clone(), Headers::new_empty())? Class::instance(ctx, HeadersClass::new_empty())?
}; };
let headers = Persistent::save(ctx, headers);
let body = object.get::<_, Option<Body>>("body")?; let body = object.get::<_, Option<Body>>("body")?;
@ -366,44 +376,61 @@ impl<'js> FromJs<'js> for RequestInit<'js> {
pub use request::Request as RequestClass; pub use request::Request as RequestClass;
#[bind(object, public)]
#[quickjs(bare)]
#[allow(non_snake_case)]
#[allow(unused_variables)]
#[allow(clippy::module_inception)]
mod request {
pub use super::*; pub use super::*;
use bytes::Bytes; use bytes::Bytes;
use js::function::Opt; use js::{
function::{Opt, Rest},
Class, Ctx, Exception, HasRefs, Result, Value,
};
// TODO: change implementation based on features. // TODO: change implementation based on features.
use reqwest::{header::HeaderName, Url}; use reqwest::{header::HeaderName, Url};
#[allow(dead_code)] #[allow(dead_code)]
#[js::class] #[derive(HasRefs)]
#[derive(Trace)] #[quickjs(has_refs)]
pub struct Request<'js> { pub struct Request {
#[qjs(skip_trace)]
pub(crate) url: Url, pub(crate) url: Url,
pub(crate) init: RequestInit<'js>, #[quickjs(has_refs)]
pub(crate) init: RequestInit,
} }
#[js::methods] impl Request {
impl<'js> Request<'js> {
// ------------------------------ // ------------------------------
// Constructor // Constructor
// ------------------------------ // ------------------------------
#[qjs(constructor)] #[quickjs(constructor)]
pub fn new(ctx: Ctx<'js>, input: Value<'js>, init: Opt<RequestInit<'js>>) -> Result<Self> { pub fn new<'js>(
ctx: Ctx<'js>,
input: Value<'js>,
init: Opt<RequestInit>,
args: Rest<()>,
) -> Result<Self> {
if let Some(url) = input.as_string() { if let Some(url) = input.as_string() {
// url string // url string
let url_str = url.to_string()?; let url_str = url.to_string()?;
let url = Url::parse(&url_str) let url = Url::parse(&url_str).map_err(|e| {
.map_err(|e| Exception::throw_type(&ctx, &format!("failed to parse url: {e}")))?; Exception::throw_type(ctx, &format!("failed to parse url: {e}"))
if !url.username().is_empty() || !url.password().map(str::is_empty).unwrap_or(true) { })?;
if !url.username().is_empty() || !url.password().map(str::is_empty).unwrap_or(true)
{
// url cannot contain non empty username and passwords // url cannot contain non empty username and passwords
return Err(Exception::throw_type(&ctx, "Url contained credentials.")); return Err(Exception::throw_type(ctx, "Url contained credentials."));
} }
let init = init.into_inner().map_or_else(|| RequestInit::default(ctx.clone()), Ok)?; let init = init.into_inner().map_or_else(|| RequestInit::default(ctx), Ok)?;
// HEAD and GET methods can't have a body // HEAD and GET methods can't have a body
if init.body.is_some() && init.method == Method::GET || init.method == Method::HEAD { if init.body.is_some() && init.method == Method::GET || init.method == Method::HEAD
{
return Err(Exception::throw_type( return Err(Exception::throw_type(
&ctx, ctx,
&format!("Request with method `{}` cannot have a body", init.method), &format!("Request with method `{}` cannot have a body", init.method),
)); ));
} }
@ -412,20 +439,23 @@ impl<'js> Request<'js> {
url, url,
init, init,
}) })
} else if let Some(request) = input.into_object().and_then(Class::<Self>::from_object) { } else if let Some(request) = input
.into_object()
.and_then(|obj| Class::<Self>::from_object(obj).ok().map(|x| x.borrow()))
{
// existing request object, just return it // existing request object, just return it
request.try_borrow()?.clone_js(ctx.clone()) request.clone_js(ctx, Default::default())
} else { } else {
Err(Exception::throw_type( Err(Exception::throw_type(
&ctx, ctx,
"request `init` paramater must either be a request object or a string", "request `init` paramater must either be a request object or a string",
)) ))
} }
} }
/// Clone the response, teeing any possible underlying streams. /// Clone the response, teeing any possible underlying streams.
#[qjs(rename = "clone")] #[quickjs(rename = "clone")]
pub fn clone_js(&self, ctx: Ctx<'js>) -> Result<Self> { pub fn clone_js(&self, ctx: Ctx<'_>, _rest: Rest<()>) -> Result<Self> {
Ok(Self { Ok(Self {
url: self.url.clone(), url: self.url.clone(),
init: self.init.clone_js(ctx)?, init: self.init.clone_js(ctx)?,
@ -435,28 +465,28 @@ impl<'js> Request<'js> {
// ------------------------------ // ------------------------------
// Instance properties // Instance properties
// ------------------------------ // ------------------------------
#[qjs(get, rename = "body_used")] #[quickjs(get)]
pub fn body_used(&self) -> bool { pub fn bodyUsed(&self) -> bool {
self.init.body.as_ref().map(Body::used).unwrap_or(true) self.init.body.as_ref().map(Body::used).unwrap_or(true)
} }
#[qjs(get)] #[quickjs(get)]
pub fn method(&self) -> String { pub fn method(&self) -> String {
self.init.method.to_string() self.init.method.to_string()
} }
#[qjs(get)] #[quickjs(get)]
pub fn url(&self) -> String { pub fn url(&self) -> String {
self.url.to_string() self.url.to_string()
} }
#[qjs(get)] #[quickjs(get)]
pub fn headers(&self) -> Class<'js, Headers> { pub fn headers<'js>(&self, ctx: Ctx<'js>) -> Class<'js, HeadersClass> {
self.init.headers.clone() self.init.headers.clone().restore(ctx).unwrap()
} }
#[qjs(get)] #[quickjs(get)]
pub fn referrer(&self) -> String { pub fn referrer(&self, ctx: Ctx<'_>) -> String {
self.init.referrer.clone() self.init.referrer.clone()
} }
// TODO // TODO
@ -466,16 +496,15 @@ impl<'js> Request<'js> {
// ------------------------------ // ------------------------------
// Convert the object to a string // Convert the object to a string
#[qjs(rename = "toString")] pub fn toString(&self) -> String {
pub fn js_to_string(&self) -> String {
String::from("[object Request]") String::from("[object Request]")
} }
/// Takes the buffer from the body leaving it used. /// Takes the buffer from the body leaving it used.
#[qjs(skip)] #[quickjs(skip)]
async fn take_buffer(&self, ctx: &Ctx<'js>) -> Result<Bytes> { async fn take_buffer<'js>(&self, ctx: Ctx<'js>) -> Result<Bytes> {
let Some(body) = self.init.body.as_ref() else { let Some(body) = self.init.body.as_ref() else {
return Ok(Bytes::new()); return Ok(Bytes::new())
}; };
match body.to_buffer().await { match body.to_buffer().await {
Ok(Some(x)) => Ok(x), Ok(Some(x)) => Ok(x),
@ -489,11 +518,11 @@ impl<'js> Request<'js> {
} }
// Returns a promise with the request body as a Blob // Returns a promise with the request body as a Blob
pub async fn blob(&self, ctx: Ctx<'js>) -> Result<Blob> { pub async fn blob(&self, ctx: Ctx<'_>, args: Rest<()>) -> Result<BlobClass> {
let headers = self.init.headers.clone(); let headers = self.init.headers.clone().restore(ctx).unwrap();
let mime = { let mime = {
let headers = headers.borrow(); let headers = headers.borrow();
let headers = &headers.inner; let headers = headers.inner.borrow();
let key = HeaderName::from_static("content-type"); let key = HeaderName::from_static("content-type");
let types = headers.get_all(key); let types = headers.get_all(key);
// TODO: This is not according to spec. // TODO: This is not according to spec.
@ -505,28 +534,27 @@ impl<'js> Request<'js> {
.to_owned() .to_owned()
}; };
let data = self.take_buffer(&ctx).await?; let data = self.take_buffer(ctx).await?;
Ok(Blob { Ok(BlobClass {
mime, mime,
data, data,
}) })
} }
// Returns a promise with the request body as FormData // Returns a promise with the request body as FormData
#[qjs(rename = "formData")] pub async fn formData<'js>(&self, ctx: Ctx<'js>, args: Rest<()>) -> Result<Value<'js>> {
pub async fn form_data(&self, ctx: Ctx<'js>) -> Result<Value<'js>> { Err(Exception::throw_internal(ctx, "Not yet implemented"))
Err(Exception::throw_internal(&ctx, "Not yet implemented"))
} }
// Returns a promise with the request body as JSON // Returns a promise with the request body as JSON
pub async fn json(&self, ctx: Ctx<'js>) -> Result<Value<'js>> { pub async fn json<'js>(&self, ctx: Ctx<'js>, args: Rest<()>) -> Result<Value<'js>> {
let text = self.text(ctx.clone()).await?; let text = self.text(ctx, args).await?;
ctx.json_parse(text) ctx.json_parse(text)
} }
// Returns a promise with the request body as text // Returns a promise with the request body as text
pub async fn text(&self, ctx: Ctx<'js>) -> Result<String> { pub async fn text<'js>(&self, ctx: Ctx<'js>, args: Rest<()>) -> Result<String> {
let data = self.take_buffer(&ctx).await?; let data = self.take_buffer(ctx).await?;
// Skip UTF-BOM // Skip UTF-BOM
if data.starts_with(&[0xEF, 0xBB, 0xBF]) { if data.starts_with(&[0xEF, 0xBB, 0xBF]) {
@ -536,6 +564,7 @@ impl<'js> Request<'js> {
} }
} }
} }
}
#[cfg(test)] #[cfg(test)]
mod test { mod test {
@ -615,7 +644,7 @@ mod test {
assert.seq(await req_2.text(),"some text"); assert.seq(await req_2.text(),"some text");
})() })()
"#).catch(&ctx).unwrap().await.catch(&ctx).unwrap(); "#).catch(ctx).unwrap().await.catch(ctx).unwrap();
}) })
.await; .await;
} }

View file

@ -1,33 +1,34 @@
use std::string::String as StdString; use std::string::String as StdString;
use js::{ use js::{
class::{Trace, Tracer}, class::{HasRefs, RefsMarker},
prelude::*, prelude::*,
Class, Ctx, Exception, FromJs, Object, Result, Value, Class, Ctx, Exception, FromJs, Object, Persistent, Result, Value,
}; };
use crate::fnc::script::fetch::{classes::Headers, util}; use crate::fnc::script::fetch::{classes::HeadersClass, util};
/// Struct containing data from the init argument from the Response constructor. /// Struct containing data from the init argument from the Response constructor.
#[derive(Clone)] #[derive(Clone)]
pub struct ResponseInit<'js> { pub struct ResponseInit {
// u16 instead of reqwest::StatusCode since javascript allows non valid status codes in some // u16 instead of reqwest::StatusCode since javascript allows non valid status codes in some
// circumstances. // circumstances.
pub status: u16, pub status: u16,
pub status_text: StdString, pub status_text: StdString,
pub headers: Class<'js, Headers>, pub headers: Persistent<Class<'static, HeadersClass>>,
} }
impl<'js> Trace<'js> for ResponseInit<'js> { impl HasRefs for ResponseInit {
fn trace<'a>(&self, tracer: Tracer<'a, 'js>) { fn mark_refs(&self, marker: &RefsMarker) {
self.headers.trace(tracer); self.headers.mark_refs(marker);
} }
} }
impl<'js> ResponseInit<'js> { impl ResponseInit {
/// Returns a ResponseInit object with all values as the default value. /// Returns a ResponseInit object with all values as the default value.
pub fn default(ctx: Ctx<'js>) -> Result<Self> { pub fn default(ctx: Ctx<'_>) -> Result<ResponseInit> {
let headers = Class::instance(ctx, Headers::new_empty())?; let headers = Class::instance(ctx, HeadersClass::new_empty())?;
let headers = Persistent::save(ctx, headers);
Ok(ResponseInit { Ok(ResponseInit {
status: 200, status: 200,
status_text: StdString::new(), status_text: StdString::new(),
@ -36,8 +37,8 @@ impl<'js> ResponseInit<'js> {
} }
} }
impl<'js> FromJs<'js> for ResponseInit<'js> { impl<'js> FromJs<'js> for ResponseInit {
fn from_js(ctx: &Ctx<'js>, value: Value<'js>) -> Result<Self> { fn from_js(ctx: Ctx<'js>, value: Value<'js>) -> Result<Self> {
let object = Object::from_js(ctx, value)?; let object = Object::from_js(ctx, value)?;
// Extract status. // Extract status.
@ -65,11 +66,12 @@ impl<'js> FromJs<'js> for ResponseInit<'js> {
// Extract headers. // Extract headers.
let headers = if let Some(headers) = object.get::<_, Option<Value>>("headers")? { let headers = if let Some(headers) = object.get::<_, Option<Value>>("headers")? {
let headers = Headers::new_inner(ctx, headers)?; let headers = HeadersClass::new_inner(ctx, headers)?;
Class::instance(ctx.clone(), headers)? Class::instance(ctx, headers)?
} else { } else {
Class::instance(ctx.clone(), Headers::new_empty())? Class::instance(ctx, HeadersClass::new_empty())?
}; };
let headers = Persistent::save(ctx, headers);
Ok(ResponseInit { Ok(ResponseInit {
status, status,

View file

@ -1,10 +1,11 @@
//! Response class implementation //! Response class implementation
use js::bind;
mod init; mod init;
use bytes::Bytes;
pub use init::ResponseInit; pub use init::ResponseInit;
use js::{class::Trace, prelude::Opt, ArrayBuffer, Class, Ctx, Exception, Result, Value}; pub use response::Response as ResponseClass;
#[allow(dead_code)] #[allow(dead_code)]
#[derive(Clone, Copy)] #[derive(Clone, Copy)]
@ -17,51 +18,65 @@ pub enum ResponseType {
OpaqueRedirect, OpaqueRedirect,
} }
use reqwest::Url; #[bind(object, public)]
#[quickjs(bare)]
#[allow(non_snake_case)]
#[allow(unused_variables)]
#[allow(clippy::module_inception)]
pub mod response {
use crate::fnc::script::fetch::{ use crate::fnc::script::fetch::{
body::{Body, BodyKind}, body::{Body, BodyKind},
classes::{BlobClass, HeadersClass},
util, RequestError, util, RequestError,
}; };
use super::{Blob, Headers}; use super::{ResponseInit, ResponseType};
use bytes::Bytes;
use js::{
function::{Opt, Rest},
ArrayBuffer, Class, Ctx, Exception, HasRefs, Persistent, Result, Value,
};
use reqwest::Url;
#[derive(HasRefs)]
#[allow(dead_code)] #[allow(dead_code)]
#[derive(Trace)] #[quickjs(has_refs)]
#[js::class] pub struct Response {
pub struct Response<'js> { #[quickjs(has_refs)]
#[qjs(skip_trace)]
pub(crate) body: Body, pub(crate) body: Body,
pub(crate) init: ResponseInit<'js>, #[quickjs(has_refs)]
#[qjs(skip_trace)] pub(crate) init: ResponseInit,
pub(crate) url: Option<Url>, pub(crate) url: Option<Url>,
#[qjs(skip_trace)]
pub(crate) r#type: ResponseType, pub(crate) r#type: ResponseType,
pub(crate) was_redirected: bool, pub(crate) was_redirected: bool,
} }
#[js::methods] impl Response {
impl<'js> Response<'js> {
// ------------------------------ // ------------------------------
// Constructor // Constructor
// ------------------------------ // ------------------------------
#[qjs(constructor)] #[quickjs(constructor)]
pub fn new( pub fn new(
ctx: Ctx<'js>, ctx: Ctx<'_>,
body: Opt<Option<Body>>, body: Opt<Option<Body>>,
init: Opt<ResponseInit<'js>>, init: Opt<ResponseInit>,
args: Rest<()>,
) -> Result<Self> { ) -> Result<Self> {
let init = match init.into_inner() { let init = match init.into_inner() {
Some(x) => x, Some(x) => x,
None => ResponseInit::default(ctx.clone())?, None => ResponseInit::default(ctx)?,
}; };
let body = body.into_inner().and_then(|x| x); let body = body.into_inner().and_then(|x| x);
if body.is_some() && util::is_null_body_status(init.status) { if body.is_some() && util::is_null_body_status(init.status) {
// Null body statuses are not allowed to have a body. // Null body statuses are not allowed to have a body.
return Err(Exception::throw_type( return Err(Exception::throw_type(
&ctx, ctx,
&format!("Response with status `{}` is not allowed to have a body", init.status), &format!(
"Response with status `{}` is not allowed to have a body",
init.status
),
)); ));
} }
let body = body.unwrap_or_default(); let body = body.unwrap_or_default();
@ -79,32 +94,32 @@ impl<'js> Response<'js> {
// Instance properties // Instance properties
// ------------------------------ // ------------------------------
#[qjs(get, rename = "bodyUsed")] #[quickjs(get)]
pub fn body_used(&self) -> bool { pub fn bodyUsed(&self) -> bool {
self.body.used() self.body.used()
} }
#[qjs(get)] #[quickjs(get)]
pub fn status(&self) -> u16 { pub fn status(&self) -> u16 {
self.init.status self.init.status
} }
#[qjs(get)] #[quickjs(get)]
pub fn ok(&self) -> bool { pub fn ok(&self) -> bool {
util::is_ok_status(self.init.status) util::is_ok_status(self.init.status)
} }
#[qjs(get)] #[quickjs(get)]
pub fn redirected(&self) -> bool { pub fn redirected(&self) -> bool {
self.was_redirected self.was_redirected
} }
#[qjs(get, rename = "statusText")] #[quickjs(get)]
pub fn status_text(&self) -> String { pub fn statusText(&self) -> String {
self.init.status_text.clone() self.init.status_text.clone()
} }
#[qjs(get, rename = "type")] #[quickjs(get)]
pub fn r#type(&self) -> &'static str { pub fn r#type(&self) -> &'static str {
match self.r#type { match self.r#type {
ResponseType::Basic => "basic", ResponseType::Basic => "basic",
@ -116,12 +131,12 @@ impl<'js> Response<'js> {
} }
} }
#[qjs(get)] #[quickjs(get)]
pub fn headers(&self) -> Class<'js, Headers> { pub fn headers<'js>(&self, ctx: Ctx<'js>) -> Class<'js, HeadersClass> {
self.init.headers.clone() self.init.headers.clone().restore(ctx).unwrap()
} }
#[qjs(get)] #[quickjs(get)]
pub fn url(&self) -> Option<String> { pub fn url(&self) -> Option<String> {
self.url.as_ref().map(|x| { self.url.as_ref().map(|x| {
if x.fragment().is_some() { if x.fragment().is_some() {
@ -139,14 +154,13 @@ impl<'js> Response<'js> {
// ------------------------------ // ------------------------------
// Convert the object to a string // Convert the object to a string
#[qjs(rename = "toString")] pub fn toString(&self, args: Rest<()>) -> String {
pub fn js_to_string(&self) -> String {
String::from("[object Response]") String::from("[object Response]")
} }
// Creates a copy of the request object // Creates a copy of the request object
#[qjs(rename = "clone")] #[quickjs(rename = "clone")]
pub fn clone_js(&self, ctx: Ctx<'js>) -> Self { pub fn clone_js(&self, ctx: Ctx<'_>, args: Rest<()>) -> Response {
Response { Response {
body: self.body.clone_js(ctx), body: self.body.clone_js(ctx),
init: self.init.clone(), init: self.init.clone(),
@ -156,8 +170,8 @@ impl<'js> Response<'js> {
} }
} }
#[qjs(skip)] #[quickjs(skip)]
async fn take_buffer(&self, ctx: &Ctx<'js>) -> Result<Bytes> { async fn take_buffer<'js>(&self, ctx: Ctx<'js>) -> Result<Bytes> {
match self.body.to_buffer().await { match self.body.to_buffer().await {
Ok(Some(x)) => Ok(x), Ok(Some(x)) => Ok(x),
Ok(None) => Err(Exception::throw_type(ctx, "Body unusable")), Ok(None) => Err(Exception::throw_type(ctx, "Body unusable")),
@ -170,11 +184,11 @@ impl<'js> Response<'js> {
} }
// Returns a promise with the response body as a Blob // Returns a promise with the response body as a Blob
pub async fn blob(&self, ctx: Ctx<'js>) -> Result<Blob> { pub async fn blob<'js>(&self, ctx: Ctx<'js>, args: Rest<()>) -> Result<BlobClass> {
let headers = self.init.headers.clone(); let headers = self.init.headers.clone().restore(ctx).unwrap();
let mime = { let mime = {
let headers = headers.borrow(); let headers = headers.borrow();
let headers = &headers.inner; let headers = headers.inner.borrow();
let types = headers.get_all(reqwest::header::CONTENT_TYPE); let types = headers.get_all(reqwest::header::CONTENT_TYPE);
// TODO: This is not according to spec. // TODO: This is not according to spec.
types types
@ -185,28 +199,27 @@ impl<'js> Response<'js> {
.to_owned() .to_owned()
}; };
let data = self.take_buffer(&ctx).await?; let data = self.take_buffer(ctx).await?;
Ok(Blob { Ok(BlobClass {
mime, mime,
data, data,
}) })
} }
// Returns a promise with the response body as FormData // Returns a promise with the response body as FormData
#[qjs(rename = "formData")] pub async fn formData<'js>(&self, ctx: Ctx<'js>, args: Rest<()>) -> Result<Value<'js>> {
pub async fn form_data(&self, ctx: Ctx<'js>) -> Result<Value<'js>> { Err(Exception::throw_internal(ctx, "Not yet implemented"))
Err(Exception::throw_internal(&ctx, "Not yet implemented"))
} }
// Returns a promise with the response body as JSON // Returns a promise with the response body as JSON
pub async fn json(&self, ctx: Ctx<'js>) -> Result<Value<'js>> { pub async fn json<'js>(&self, ctx: Ctx<'js>, args: Rest<()>) -> Result<Value<'js>> {
let text = self.text(ctx.clone()).await?; let text = self.text(ctx, args).await?;
ctx.json_parse(text) ctx.json_parse(text)
} }
// Returns a promise with the response body as text // Returns a promise with the response body as text
pub async fn text(&self, ctx: Ctx<'js>) -> Result<String> { pub async fn text<'js>(&self, ctx: Ctx<'js>, args: Rest<()>) -> Result<String> {
let data = self.take_buffer(&ctx).await?; let data = self.take_buffer(ctx).await?;
// Skip UTF-BOM // Skip UTF-BOM
if data.starts_with(&[0xEF, 0xBB, 0xBF]) { if data.starts_with(&[0xEF, 0xBB, 0xBF]) {
@ -217,9 +230,12 @@ impl<'js> Response<'js> {
} }
// Returns a promise with the response body as text // Returns a promise with the response body as text
#[qjs(rename = "arrayBuffer")] pub async fn arrayBuffer<'js>(
pub async fn array_buffer(&self, ctx: Ctx<'js>) -> Result<ArrayBuffer<'js>> { &self,
let data = self.take_buffer(&ctx).await?; ctx: Ctx<'js>,
args: Rest<()>,
) -> Result<ArrayBuffer<'js>> {
let data = self.take_buffer(ctx).await?;
ArrayBuffer::new(ctx, data) ArrayBuffer::new(ctx, data)
} }
@ -227,15 +243,16 @@ impl<'js> Response<'js> {
// Static methods // Static methods
// ------------------------------ // ------------------------------
#[qjs(r#static, rename = "json")] #[quickjs(rename = "json")]
pub fn static_json( pub fn static_json<'js>(
ctx: Ctx<'js>, ctx: Ctx<'js>,
data: Value<'js>, data: Value<'js>,
init: Opt<ResponseInit<'js>>, init: Opt<ResponseInit>,
args: Rest<()>,
) -> Result<Self> { ) -> Result<Self> {
let json = ctx.json_stringify(data)?; let json = ctx.json_stringify(data)?;
let json = let json =
json.ok_or_else(|| Exception::throw_type(&ctx, "Value is not JSON serializable"))?; json.ok_or_else(|| Exception::throw_type(ctx, "Value is not JSON serializable"))?;
let json = json.to_string()?; let json = json.to_string()?;
let init = if let Some(init) = init.into_inner() { let init = if let Some(init) = init.into_inner() {
@ -254,9 +271,8 @@ impl<'js> Response<'js> {
} }
// Returns a new response representing a network error // Returns a new response representing a network error
#[qjs(r#static)] pub fn error(ctx: Ctx<'_>, args: Rest<()>) -> Result<Self> {
pub fn error(ctx: Ctx<'js>) -> Result<Self> { let headers = Persistent::save(ctx, Class::instance(ctx, HeadersClass::new_empty())?);
let headers = Class::instance(ctx, Headers::new_empty())?;
Ok(Response { Ok(Response {
url: None, url: None,
body: Body::new(), body: Body::new(),
@ -271,18 +287,22 @@ impl<'js> Response<'js> {
} }
// Creates a new response with a different URL // Creates a new response with a different URL
#[qjs(r#static)] pub fn redirect(
pub fn redirect(ctx: Ctx<'_>, url: String, status: Opt<u32>) -> Result<Response> { ctx: Ctx<'_>,
url: String,
status: Opt<u32>,
args: Rest<()>,
) -> Result<Response> {
let url = url let url = url
.parse::<Url>() .parse::<Url>()
.map_err(|e| Exception::throw_type(&ctx, &format!("Invalid url: {e}")))?; .map_err(|e| Exception::throw_type(ctx, &format!("Invalid url: {e}")))?;
let status = status.into_inner().unwrap_or(302) as u16; let status = status.into_inner().unwrap_or(302) as u16;
if !util::is_redirect_status(status) { if !util::is_redirect_status(status) {
return Err(Exception::throw_range(&ctx, "Status code is not a redirect status")); return Err(Exception::throw_range(ctx, "Status code is not a redirect status"));
} }
let headers = Class::instance(ctx, Headers::new_empty())?; let headers = Persistent::save(ctx, Class::instance(ctx, HeadersClass::new_empty())?);
Ok(Response { Ok(Response {
url: Some(url), url: Some(url),
@ -297,6 +317,7 @@ impl<'js> Response<'js> {
}) })
} }
} }
}
#[cfg(test)] #[cfg(test)]
mod test { mod test {
@ -367,7 +388,7 @@ mod test {
})() })()
"#).catch(&ctx).unwrap().await.catch(&ctx).unwrap(); "#).catch(ctx).unwrap().await.catch(ctx).unwrap();
}) })
.await; .await;
} }

View file

@ -2,28 +2,29 @@
use crate::fnc::script::fetch::{ use crate::fnc::script::fetch::{
body::{Body, BodyData, BodyKind}, body::{Body, BodyData, BodyKind},
classes::{self, Request, RequestInit, Response, ResponseInit, ResponseType}, classes::{
self, HeadersClass, RequestClass, RequestInit, ResponseClass, ResponseInit, ResponseType,
},
RequestError, RequestError,
}; };
use futures::TryStreamExt; use futures::TryStreamExt;
use js::{function::Opt, Class, Ctx, Exception, Result, Value}; use js::{bind, function::Opt, prelude::*, Class, Ctx, Exception, Persistent, Result, Value};
use reqwest::{ use reqwest::{
header::{HeaderValue, CONTENT_TYPE}, header::{HeaderValue, CONTENT_TYPE},
redirect, Body as ReqBody, redirect, Body as ReqBody,
}; };
use std::sync::Arc; use std::sync::Arc;
use super::classes::Headers; #[bind(object, public)]
#[js::function]
#[allow(unused_variables)] #[allow(unused_variables)]
pub async fn fetch<'js>( pub async fn fetch<'js>(
ctx: Ctx<'js>, ctx: Ctx<'js>,
input: Value<'js>, input: Value<'js>,
init: Opt<RequestInit<'js>>, init: Opt<RequestInit>,
) -> Result<Response<'js>> { args: Rest<()>,
) -> Result<ResponseClass> {
// Create a request from the input. // Create a request from the input.
let js_req = Request::new(ctx.clone(), input, init)?; let js_req = RequestClass::new(ctx, input, init, args)?;
let url = js_req.url; let url = js_req.url;
@ -31,9 +32,9 @@ pub async fn fetch<'js>(
// SurrealDB Implementation keeps all javascript parts inside the context::with scope so this // SurrealDB Implementation keeps all javascript parts inside the context::with scope so this
// unwrap should never panic. // unwrap should never panic.
let headers = js_req.init.headers; let headers = js_req.init.headers.restore(ctx).unwrap();
let headers = headers.borrow(); let headers = headers.borrow();
let mut headers = headers.inner.clone(); let mut headers = headers.inner.borrow().clone();
let redirect = js_req.init.request_redirect; let redirect = js_req.init.request_redirect;
@ -54,7 +55,7 @@ pub async fn fetch<'js>(
}); });
let client = reqwest::Client::builder().redirect(policy).build().map_err(|e| { let client = reqwest::Client::builder().redirect(policy).build().map_err(|e| {
Exception::throw_internal(&ctx, &format!("Could not initialize http client: {e}")) Exception::throw_internal(ctx, &format!("Could not initialize http client: {e}"))
})?; })?;
// Set the body for the request. // Set the body for the request.
@ -69,7 +70,7 @@ pub async fn fetch<'js>(
let body = ReqBody::from(x); let body = ReqBody::from(x);
req_builder = req_builder.body(body); req_builder = req_builder.body(body);
} }
BodyData::Used => return Err(Exception::throw_type(&ctx, "Body unusable")), BodyData::Used => return Err(Exception::throw_type(ctx, "Body unusable")),
}; };
match body.kind { match body.kind {
BodyKind::Buffer => {} BodyKind::Buffer => {}
@ -93,11 +94,12 @@ pub async fn fetch<'js>(
.headers(headers) .headers(headers)
.send() .send()
.await .await
.map_err(|e| Exception::throw_type(&ctx, &e.to_string()))?; .map_err(|e| Exception::throw_type(ctx, &e.to_string()))?;
// Extract the headers // Extract the headers
let headers = Headers::from_map(response.headers().clone()); let headers = HeadersClass::from_map(response.headers().clone());
let headers = Class::instance(ctx, headers)?; let headers = Class::instance(ctx, headers)?;
let headers = Persistent::save(ctx, headers);
let init = ResponseInit { let init = ResponseInit {
headers, headers,
status: response.status().as_u16(), status: response.status().as_u16(),
@ -109,7 +111,7 @@ pub async fn fetch<'js>(
BodyKind::Buffer, BodyKind::Buffer,
response.bytes_stream().map_err(Arc::new).map_err(RequestError::Reqwest), response.bytes_stream().map_err(Arc::new).map_err(RequestError::Reqwest),
); );
let response = Response { let response = ResponseClass {
body, body,
init, init,
url: Some(url), url: Some(url),

View file

@ -1,6 +1,6 @@
use std::{error::Error, fmt, sync::Arc}; use std::{error::Error, fmt, sync::Arc};
use js::{Class, Ctx, Result}; use js::{Ctx, Result};
mod body; mod body;
mod classes; mod classes;
@ -9,7 +9,7 @@ mod stream;
mod util; mod util;
use classes::{Blob, FormData, Headers, Request, Response}; use classes::{Blob, FormData, Headers, Request, Response};
use func::js_fetch; use func::Fetch;
// Anoyingly errors aren't clone, // Anoyingly errors aren't clone,
// But with how we implement streams RequestError must be clone. // But with how we implement streams RequestError must be clone.
@ -30,14 +30,15 @@ impl fmt::Display for RequestError {
impl Error for RequestError {} impl Error for RequestError {}
/// Register the fetch types in the context. /// Register the fetch types in the context.
pub fn register(ctx: &Ctx<'_>) -> Result<()> { pub fn register(ctx: Ctx<'_>) -> Result<()> {
let globals = ctx.globals(); let globals = ctx.globals();
globals.set("fetch", js_fetch)?; globals.init_def::<Fetch>()?;
Class::<Response>::define(&globals)?;
Class::<Request>::define(&globals)?; globals.init_def::<Response>()?;
Class::<Blob>::define(&globals)?; globals.init_def::<Request>()?;
Class::<FormData>::define(&globals)?; globals.init_def::<Blob>()?;
Class::<Headers>::define(&globals)?; globals.init_def::<FormData>()?;
globals.init_def::<Headers>()?;
Ok(()) Ok(())
} }
@ -54,9 +55,9 @@ mod test {
let ctx = js::AsyncContext::full(&rt).await.unwrap(); let ctx = js::AsyncContext::full(&rt).await.unwrap();
js::async_with!(ctx => |$ctx|{ js::async_with!(ctx => |$ctx|{
crate::fnc::script::fetch::register(&$ctx).unwrap(); crate::fnc::script::fetch::register($ctx).unwrap();
$ctx.eval::<(),_>(r" $ctx.eval::<(),_>(r#"
globalThis.assert = (...arg) => { globalThis.assert = (...arg) => {
arg.forEach(x => { arg.forEach(x => {
if (!x) { if (!x) {
@ -90,7 +91,7 @@ mod test {
} }
throw new Error(`Code which should throw, didnt: \n${cb}`) throw new Error(`Code which should throw, didnt: \n${cb}`)
} }
").unwrap(); "#).unwrap();
$($t)* $($t)*
}).await; }).await;

View file

@ -1,5 +1,4 @@
//! stub implementations for the fetch API when `http` is not enabled. /// The stub implementations for the fetch API when `http` is not enabled.
use js::{bind, prelude::*, Ctx, Exception, Result}; use js::{bind, prelude::*, Ctx, Exception, Result};
#[cfg(test)] #[cfg(test)]

View file

@ -5,7 +5,6 @@ use crate::sql::object::Object;
use crate::sql::value::Value; use crate::sql::value::Value;
use crate::sql::Id; use crate::sql::Id;
use chrono::{TimeZone, Utc}; use chrono::{TimeZone, Utc};
use js::prelude::This;
use js::Ctx; use js::Ctx;
use js::Error; use js::Error;
use js::Exception; use js::Exception;
@ -21,7 +20,7 @@ fn check_nul(s: &str) -> Result<(), Error> {
} }
impl<'js> FromJs<'js> for Value { impl<'js> FromJs<'js> for Value {
fn from_js(ctx: &Ctx<'js>, val: js::Value<'js>) -> Result<Self, Error> { fn from_js(ctx: Ctx<'js>, val: js::Value<'js>) -> Result<Self, Error> {
match val { match val {
val if val.type_name() == "null" => Ok(Value::Null), val if val.type_name() == "null" => Ok(Value::Null),
val if val.type_name() == "undefined" => Ok(Value::None), val if val.type_name() == "undefined" => Ok(Value::None),
@ -50,15 +49,14 @@ impl<'js> FromJs<'js> for Value {
// Check to see if this object is an error // Check to see if this object is an error
if v.is_error() { if v.is_error() {
let e: String = v.get("message")?; let e: String = v.get("message")?;
let (Ok(e) | Err(e)) = let (Ok(e) | Err(e)) = Exception::from_message(ctx, &e).map(|x| x.throw());
Exception::from_message(ctx.clone(), &e).map(|x| x.throw());
return Err(e); return Err(e);
} }
// Check to see if this object is a record // Check to see if this object is a record
if (v).instance_of::<classes::record::Record>() { if (v).instance_of::<classes::record::record::Record>() {
let v = v.into_class::<classes::record::Record>().unwrap(); let v = v.into_instance::<classes::record::record::Record>().unwrap();
let borrow = v.borrow(); let borrow = v.borrow();
let v: &classes::record::Record = &borrow; let v: &classes::record::record::Record = &borrow;
check_nul(&v.value.tb)?; check_nul(&v.value.tb)?;
if let Id::String(s) = &v.value.id { if let Id::String(s) = &v.value.id {
check_nul(s)?; check_nul(s)?;
@ -66,20 +64,20 @@ impl<'js> FromJs<'js> for Value {
return Ok(v.value.clone().into()); return Ok(v.value.clone().into());
} }
// Check to see if this object is a duration // Check to see if this object is a duration
if (v).instance_of::<classes::duration::Duration>() { if (v).instance_of::<classes::duration::duration::Duration>() {
let v = v.into_class::<classes::duration::Duration>().unwrap(); let v = v.into_instance::<classes::duration::duration::Duration>().unwrap();
let borrow = v.borrow(); let borrow = v.borrow();
let v: &classes::duration::Duration = &borrow; let v: &classes::duration::duration::Duration = &borrow;
return match &v.value { return match &v.value {
Some(v) => Ok(v.clone().into()), Some(v) => Ok(v.clone().into()),
None => Ok(Value::None), None => Ok(Value::None),
}; };
} }
// Check to see if this object is a uuid // Check to see if this object is a uuid
if (v).instance_of::<classes::uuid::Uuid>() { if (v).instance_of::<classes::uuid::uuid::Uuid>() {
let v = v.into_class::<classes::uuid::Uuid>().unwrap(); let v = v.into_instance::<classes::uuid::uuid::Uuid>().unwrap();
let borrow = v.borrow(); let borrow = v.borrow();
let v: &classes::uuid::Uuid = &borrow; let v: &classes::uuid::uuid::Uuid = &borrow;
return match &v.value { return match &v.value {
Some(v) => Ok(v.clone().into()), Some(v) => Ok(v.clone().into()),
None => Ok(Value::None), None => Ok(Value::None),
@ -89,7 +87,7 @@ impl<'js> FromJs<'js> for Value {
let date: js::Object = ctx.globals().get("Date")?; let date: js::Object = ctx.globals().get("Date")?;
if (v).is_instance_of(&date) { if (v).is_instance_of(&date) {
let f: js::Function = v.get("getTime")?; let f: js::Function = v.get("getTime")?;
let m: i64 = f.call((This(v),))?; let m: i64 = f.call((js::prelude::This(v),))?;
let d = Utc.timestamp_millis_opt(m).unwrap(); let d = Utc.timestamp_millis_opt(m).unwrap();
return Ok(Datetime::from(d).into()); return Ok(Datetime::from(d).into());
} }

View file

@ -1,44 +1,32 @@
#[js::bind(object, public)]
#[quickjs(rename = "console")]
#[allow(clippy::module_inception)]
pub mod console {
// Specify the imports // Specify the imports
use crate::sql::value::Value; use crate::sql::value::Value;
use js::{prelude::Rest, Ctx, Object, Result}; use js::prelude::Rest;
/// Log the input values as INFO /// Log the input values as INFO
#[js::function]
pub fn log(args: Rest<Value>) { pub fn log(args: Rest<Value>) {
info!("{}", args.iter().map(|v| v.to_raw_string()).collect::<Vec<String>>().join(" ")); info!("{}", args.iter().map(|v| v.to_raw_string()).collect::<Vec<String>>().join(" "));
} }
/// Log the input values as INFO /// Log the input values as INFO
#[js::function]
pub fn info(args: Rest<Value>) { pub fn info(args: Rest<Value>) {
info!("{}", args.iter().map(|v| v.to_raw_string()).collect::<Vec<String>>().join(" ")); info!("{}", args.iter().map(|v| v.to_raw_string()).collect::<Vec<String>>().join(" "));
} }
/// Log the input values as WARN /// Log the input values as WARN
#[js::function]
pub fn warn(args: Rest<Value>) { pub fn warn(args: Rest<Value>) {
warn!("{}", args.iter().map(|v| v.to_raw_string()).collect::<Vec<String>>().join(" ")); warn!("{}", args.iter().map(|v| v.to_raw_string()).collect::<Vec<String>>().join(" "));
} }
/// Log the input values as ERROR /// Log the input values as ERROR
#[js::function]
pub fn error(args: Rest<Value>) { pub fn error(args: Rest<Value>) {
error!("{}", args.iter().map(|v| v.to_raw_string()).collect::<Vec<String>>().join(" ")); error!("{}", args.iter().map(|v| v.to_raw_string()).collect::<Vec<String>>().join(" "));
} }
/// Log the input values as DEBUG /// Log the input values as DEBUG
#[js::function]
pub fn debug(args: Rest<Value>) { pub fn debug(args: Rest<Value>) {
debug!("{}", args.iter().map(|v| v.to_raw_string()).collect::<Vec<String>>().join(" ")); debug!("{}", args.iter().map(|v| v.to_raw_string()).collect::<Vec<String>>().join(" "));
} }
/// Log the input values as TRACE /// Log the input values as TRACE
#[js::function]
pub fn trace(args: Rest<Value>) { pub fn trace(args: Rest<Value>) {
trace!("{}", args.iter().map(|v| v.to_raw_string()).collect::<Vec<String>>().join(" ")); trace!("{}", args.iter().map(|v| v.to_raw_string()).collect::<Vec<String>>().join(" "));
} }
pub fn console<'js>(ctx: &Ctx<'js>) -> Result<Object<'js>> {
let console = Object::new(ctx.clone())?;
console.set("log", js_log)?;
console.set("info", js_info)?;
console.set("warn", js_warn)?;
console.set("error", js_error)?;
console.set("debug", js_debug)?;
console.set("trace", js_trace)?;
Ok(console)
} }

View file

@ -12,58 +12,58 @@ use js::Object;
use js::Undefined; use js::Undefined;
impl<'js> IntoJs<'js> for Value { impl<'js> IntoJs<'js> for Value {
fn into_js(self, ctx: &Ctx<'js>) -> Result<js::Value<'js>, Error> { fn into_js(self, ctx: Ctx<'js>) -> Result<js::Value<'js>, Error> {
(&self).into_js(ctx) (&self).into_js(ctx)
} }
} }
impl<'js> IntoJs<'js> for &Value { impl<'js> IntoJs<'js> for &Value {
fn into_js(self, ctx: &Ctx<'js>) -> Result<js::Value<'js>, Error> { fn into_js(self, ctx: Ctx<'js>) -> Result<js::Value<'js>, Error> {
match self { match self {
Value::Null => Null.into_js(ctx), Value::Null => Null.into_js(ctx),
Value::None => Undefined.into_js(ctx), Value::None => Undefined.into_js(ctx),
Value::Bool(boolean) => Ok(js::Value::new_bool(ctx.clone(), *boolean)), Value::Bool(boolean) => Ok(js::Value::new_bool(ctx, *boolean)),
Value::Strand(v) => js::String::from_str(ctx.clone(), v)?.into_js(ctx), Value::Strand(v) => js::String::from_str(ctx, v)?.into_js(ctx),
Value::Number(Number::Int(v)) => Ok(js::Value::new_int(ctx.clone(), *v as i32)), Value::Number(Number::Int(v)) => Ok(js::Value::new_int(ctx, *v as i32)),
Value::Number(Number::Float(v)) => Ok(js::Value::new_float(ctx.clone(), *v)), Value::Number(Number::Float(v)) => Ok(js::Value::new_float(ctx, *v)),
&Value::Number(Number::Decimal(v)) => match decimal_is_integer(&v) { &Value::Number(Number::Decimal(v)) => match decimal_is_integer(&v) {
true => Ok(js::Value::new_int(ctx.clone(), v.try_into().unwrap_or_default())), true => Ok(js::Value::new_int(ctx, v.try_into().unwrap_or_default())),
false => Ok(js::Value::new_float(ctx.clone(), v.try_into().unwrap_or_default())), false => Ok(js::Value::new_float(ctx, v.try_into().unwrap_or_default())),
}, },
Value::Datetime(v) => { Value::Datetime(v) => {
let date: js::function::Constructor = ctx.globals().get("Date")?; let date: js::Function = ctx.globals().get("Date")?;
date.construct((v.0.timestamp_millis(),)) date.construct((v.0.timestamp_millis(),))
} }
Value::Thing(v) => Ok(Class::<classes::record::Record>::instance( Value::Thing(v) => Ok(Class::<classes::record::record::Record>::instance(
ctx.clone(), ctx,
classes::record::Record { classes::record::record::Record {
value: v.to_owned(), value: v.to_owned(),
}, },
)? )?
.into_value()), .into_value()),
Value::Duration(v) => Ok(Class::<classes::duration::Duration>::instance( Value::Duration(v) => Ok(Class::<classes::duration::duration::Duration>::instance(
ctx.clone(), ctx,
classes::duration::Duration { classes::duration::duration::Duration {
value: Some(v.to_owned()), value: Some(v.to_owned()),
}, },
)? )?
.into_value()), .into_value()),
Value::Uuid(v) => Ok(Class::<classes::uuid::Uuid>::instance( Value::Uuid(v) => Ok(Class::<classes::uuid::uuid::Uuid>::instance(
ctx.clone(), ctx,
classes::uuid::Uuid { classes::uuid::uuid::Uuid {
value: Some(v.to_owned()), value: Some(v.to_owned()),
}, },
)? )?
.into_value()), .into_value()),
Value::Array(v) => { Value::Array(v) => {
let x = Array::new(ctx.clone())?; let x = Array::new(ctx)?;
for (i, v) in v.iter().enumerate() { for (i, v) in v.iter().enumerate() {
x.set(i, v)?; x.set(i, v)?;
} }
x.into_js(ctx) x.into_js(ctx)
} }
Value::Object(v) => { Value::Object(v) => {
let x = Object::new(ctx.clone())?; let x = Object::new(ctx)?;
for (k, v) in v.iter() { for (k, v) in v.iter() {
x.set(k, v)?; x.set(k, v)?;
} }

View file

@ -50,24 +50,23 @@ pub async fn run(
// Attempt to execute the script // Attempt to execute the script
async_with!(ctx => |ctx|{ async_with!(ctx => |ctx|{
let res = async{ let res = async move {
// register all classes to the runtime. // register all classes to the runtime.
// Get the context global object // Get the context global object
let global = ctx.globals(); let global = ctx.globals();
// Register the surrealdb module as a global object // Register the surrealdb module as a global object
global.set( global.set(
"surrealdb", "surrealdb",
Module::evaluate_def::<modules::surrealdb::Package, _>(ctx.clone(), "surrealdb")? Module::evaluate_def::<modules::surrealdb::Package, _>(ctx, "surrealdb")?
.get::<_, js::Value>("default")?, .get::<_, js::Value>("default")?,
)?; )?;
fetch::register(&ctx)?; fetch::register(ctx)?;
let console = globals::console::console(&ctx)?;
// Register the console function to the globals // Register the console function to the globals
global.set("console",console)?; global.init_def::<globals::console::Console>()?;
// Register the special SurrealDB types as classes // Register the special SurrealDB types as classes
classes::init(&ctx)?; classes::init(ctx)?;
// Attempt to compile the script // Attempt to compile the script
let res = ctx.clone().compile("script", src)?; let res = ctx.compile("script", src)?;
// Attempt to fetch the main export // Attempt to fetch the main export
let fnc = res.get::<_, Function>("default")?; let fnc = res.get::<_, Function>("default")?;
// Extract the doc if any // Extract the doc if any
@ -77,7 +76,7 @@ pub async fn run(
promise.await promise.await
}.await; }.await;
res.catch(&ctx).map_err(Error::from) res.catch(ctx).map_err(Error::from)
}) })
.await .await
} }

View file

@ -54,8 +54,8 @@ macro_rules! impl_module_def {
Ok(()) Ok(())
} }
fn evaluate<'js>(ctx: &js::Ctx<'js>, exports: &mut js::module::Exports<'js>) -> js::Result<()> { fn evaluate<'js>(ctx: js::Ctx<'js>, exports: &mut js::module::Exports<'js>) -> js::Result<()> {
let default = js::Object::new(ctx.clone())?; let default = js::Object::new(ctx)?;
$( $(
exports.export($name, crate::fnc::script::modules::impl_module_def!(ctx, $path, $name, $action, $($wrapper)?))?; exports.export($name, crate::fnc::script::modules::impl_module_def!(ctx, $path, $name, $action, $($wrapper)?))?;
default.set($name, crate::fnc::script::modules::impl_module_def!(ctx, $path, $name, $action, $($wrapper)?))?; default.set($name, crate::fnc::script::modules::impl_module_def!(ctx, $path, $name, $action, $($wrapper)?))?;

View file

@ -1,31 +1,13 @@
use js::{ #[js::bind(module, public)]
module::{Declarations, Exports, ModuleDef}, #[quickjs(bare)]
Result, #[allow(non_upper_case_globals)]
}; pub mod package {
/// Get the target system architecture /// Get the target system architecture
#[js::function]
pub fn arch() -> &'static str { pub fn arch() -> &'static str {
crate::env::arch() crate::env::arch()
} }
/// Get the target operating system /// Get the target operating system
#[js::function]
pub fn platform() -> &'static str { pub fn platform() -> &'static str {
crate::env::os() crate::env::os()
} }
pub struct Package;
impl ModuleDef for Package {
fn declare(declare: &mut Declarations) -> Result<()> {
declare.declare("arch")?;
declare.declare("platform")?;
Ok(())
}
fn evaluate<'js>(_ctx: &js::Ctx<'js>, exports: &mut Exports<'js>) -> Result<()> {
exports.export("arch", js_arch)?;
exports.export("platform", js_platform)?;
Ok(())
}
} }

View file

@ -12,9 +12,9 @@ impl_module_def!(
"version" => (env!("CARGO_PKG_VERSION")) "version" => (env!("CARGO_PKG_VERSION"))
); );
fn pkg<'js, D>(ctx: &Ctx<'js>, name: &str) -> Result<Value<'js>> fn pkg<'js, D>(ctx: Ctx<'js>, name: &str) -> Result<Value<'js>>
where where
D: ModuleDef, D: ModuleDef,
{ {
Module::evaluate_def::<D, _>(ctx.clone(), name)?.get::<_, js::Value>("default") Module::evaluate_def::<D, _>(ctx, name)?.get::<_, js::Value>("default")
} }

View file

@ -2938,7 +2938,7 @@ mod tests {
assert_eq!(24, std::mem::size_of::<crate::sql::table::Table>()); assert_eq!(24, std::mem::size_of::<crate::sql::table::Table>());
assert_eq!(56, std::mem::size_of::<crate::sql::thing::Thing>()); assert_eq!(56, std::mem::size_of::<crate::sql::thing::Thing>());
assert_eq!(40, std::mem::size_of::<crate::sql::model::Model>()); assert_eq!(40, std::mem::size_of::<crate::sql::model::Model>());
assert_eq!(32, std::mem::size_of::<crate::sql::regex::Regex>()); assert_eq!(16, std::mem::size_of::<crate::sql::regex::Regex>());
assert_eq!(8, std::mem::size_of::<Box<crate::sql::range::Range>>()); assert_eq!(8, std::mem::size_of::<Box<crate::sql::range::Range>>());
assert_eq!(8, std::mem::size_of::<Box<crate::sql::edges::Edges>>()); assert_eq!(8, std::mem::size_of::<Box<crate::sql::edges::Edges>>());
assert_eq!(8, std::mem::size_of::<Box<crate::sql::function::Function>>()); assert_eq!(8, std::mem::size_of::<Box<crate::sql::function::Function>>());