use super::utils; use crate::xx::*; use crate::*; pub use async_executors::JoinHandle; use async_executors::{Bindgen, LocalSpawnHandleExt /*, SpawnHandleExt*/}; use core::fmt; use futures_util::future::{select, Either}; use js_sys::*; use wasm_bindgen_futures::*; use web_sys::*; #[wasm_bindgen] extern "C" { #[wasm_bindgen(catch, structural, js_namespace = global, js_name = setTimeout)] fn nodejs_global_set_timeout_with_callback_and_timeout_and_arguments_0( handler: &::js_sys::Function, timeout: u32, ) -> Result; } pub fn get_timestamp() -> u64 { if utils::is_browser() { return (Date::now() * 1000.0f64) as u64; } else if utils::is_nodejs() { return (Date::now() * 1000.0f64) as u64; } else { panic!("WASM requires browser or nodejs environment"); } } pub fn random_bytes(dest: &mut [u8]) -> Result<(), String> { let len = dest.len(); let u32len = len / 4; let remlen = len % 4; for n in 0..u32len { let r = (Math::random() * (u32::max_value() as f64)) as u32; dest[n * 4 + 0] = (r & 0xFF) as u8; dest[n * 4 + 1] = ((r >> 8) & 0xFF) as u8; dest[n * 4 + 2] = ((r >> 16) & 0xFF) as u8; dest[n * 4 + 3] = ((r >> 24) & 0xFF) as u8; } if remlen > 0 { let r = (Math::random() * (u32::max_value() as f64)) as u32; for n in 0..remlen { dest[u32len * 4 + n] = ((r >> (n * 8)) & 0xFF) as u8; } } Ok(()) } pub fn get_random_u32() -> u32 { (Math::random() * (u32::max_value() as f64)) as u32 } pub fn get_random_u64() -> u64 { let v1: u32 = get_random_u32(); let v2: u32 = get_random_u32(); ((v1 as u64) << 32) | ((v2 as u32) as u64) } pub async fn sleep(millis: u32) { if utils::is_browser() { let wait_millis = if millis > u32::MAX { i32::MAX } else { millis as i32 }; let promise = Promise::new(&mut |yes, _| { let win = window().unwrap(); win.set_timeout_with_callback_and_timeout_and_arguments_0(&yes, wait_millis) .unwrap(); }); JsFuture::from(promise).await.unwrap(); } else if utils::is_nodejs() { let promise = Promise::new(&mut |yes, _| { nodejs_global_set_timeout_with_callback_and_timeout_and_arguments_0(&yes, millis) .unwrap(); }); JsFuture::from(promise).await.unwrap(); } else { panic!("WASM requires browser or nodejs environment"); } } pub fn spawn(future: impl Future + 'static) -> JoinHandle where Out: Send + 'static, { Bindgen .spawn_handle_local(future) .expect("wasm-bindgen-futures spawn should never error out") } pub fn spawn_local(future: impl Future + 'static) -> JoinHandle where Out: 'static, { Bindgen .spawn_handle_local(future) .expect("wasm-bindgen-futures spawn_local should never error out") } pub fn interval(freq_ms: u32, callback: F) -> SystemPinBoxFuture<()> where F: Fn() -> FUT + 'static, FUT: Future, { let e = Eventual::new(); let ie = e.clone(); let jh = spawn_local(Box::pin(async move { while timeout(freq_ms, ie.instance_clone(())).await.is_err() { callback().await; } })); Box::pin(async move { e.resolve().await; jh.await; }) } #[derive(Clone, Copy, Debug, Eq, PartialEq)] pub struct TimeoutError { _private: (), } //impl Error for TimeoutError {} impl fmt::Display for TimeoutError { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { "future has timed out".fmt(f) } } pub async fn timeout(dur_ms: u32, f: F) -> Result where F: Future, { match select(Box::pin(intf::sleep(dur_ms)), Box::pin(f)).await { Either::Left((_x, _b)) => Err(TimeoutError { _private: () }), Either::Right((y, _a)) => Ok(y), } } // xxx: for now until wasm threads are more stable, and/or we bother with web workers pub fn get_concurrency() -> u32 { 1 }