86 lines
2.2 KiB
Rust
86 lines
2.2 KiB
Rust
|
use std::convert::TryInto;
|
||
|
|
use std::time::Duration;
|
||
|
|
|
||
|
|
use js_sys::Function;
|
||
|
|
use wasm_bindgen::prelude::{wasm_bindgen, Closure};
|
||
|
|
use wasm_bindgen::{JsCast, JsValue};
|
||
|
|
use web_sys::{AbortController, AbortSignal};
|
||
|
|
|
||
|
|
mod body;
|
||
|
|
mod client;
|
||
|
|
/// TODO
|
||
|
|
#[cfg(feature = "multipart")]
|
||
|
|
pub mod multipart;
|
||
|
|
mod request;
|
||
|
|
mod response;
|
||
|
|
|
||
|
|
pub use self::body::Body;
|
||
|
|
pub use self::client::{Client, ClientBuilder};
|
||
|
|
pub use self::request::{Request, RequestBuilder};
|
||
|
|
pub use self::response::Response;
|
||
|
|
|
||
|
|
#[wasm_bindgen]
|
||
|
|
extern "C" {
|
||
|
|
#[wasm_bindgen(js_name = "setTimeout")]
|
||
|
|
fn set_timeout(handler: &Function, timeout: i32) -> JsValue;
|
||
|
|
|
||
|
|
#[wasm_bindgen(js_name = "clearTimeout")]
|
||
|
|
fn clear_timeout(handle: JsValue) -> JsValue;
|
||
|
|
}
|
||
|
|
|
||
|
|
async fn promise<T>(promise: js_sys::Promise) -> Result<T, crate::error::BoxError>
|
||
|
|
where
|
||
|
|
T: JsCast,
|
||
|
|
{
|
||
|
|
use wasm_bindgen_futures::JsFuture;
|
||
|
|
|
||
|
|
let js_val = JsFuture::from(promise).await.map_err(crate::error::wasm)?;
|
||
|
|
|
||
|
|
js_val
|
||
|
|
.dyn_into::<T>()
|
||
|
|
.map_err(|_js_val| "promise resolved to unexpected type".into())
|
||
|
|
}
|
||
|
|
|
||
|
|
/// A guard that cancels a fetch request when dropped.
|
||
|
|
struct AbortGuard {
|
||
|
|
ctrl: AbortController,
|
||
|
|
timeout: Option<(JsValue, Closure<dyn FnMut()>)>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl AbortGuard {
|
||
|
|
fn new() -> crate::Result<Self> {
|
||
|
|
Ok(AbortGuard {
|
||
|
|
ctrl: AbortController::new()
|
||
|
|
.map_err(crate::error::wasm)
|
||
|
|
.map_err(crate::error::builder)?,
|
||
|
|
timeout: None,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
fn signal(&self) -> AbortSignal {
|
||
|
|
self.ctrl.signal()
|
||
|
|
}
|
||
|
|
|
||
|
|
fn timeout(&mut self, timeout: Duration) {
|
||
|
|
let ctrl = self.ctrl.clone();
|
||
|
|
let abort =
|
||
|
|
Closure::once(move || ctrl.abort_with_reason(&"reqwest::errors::TimedOut".into()));
|
||
|
|
let timeout = set_timeout(
|
||
|
|
abort.as_ref().unchecked_ref::<js_sys::Function>(),
|
||
|
|
timeout.as_millis().try_into().expect("timeout"),
|
||
|
|
);
|
||
|
|
if let Some((id, _)) = self.timeout.replace((timeout, abort)) {
|
||
|
|
clear_timeout(id);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Drop for AbortGuard {
|
||
|
|
fn drop(&mut self) {
|
||
|
|
self.ctrl.abort();
|
||
|
|
if let Some((id, _)) = self.timeout.take() {
|
||
|
|
clear_timeout(id);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|