Allow use in a web worker (#2)

* fix: allow use in web worker

* Add tests in worker scope

* Add common module for browser and worker tests

* js_panic changes from main

Author: Fedeparma74

Cargo.toml

@@ -35,6 +35,7 @@ web-sys = { version = "0.3.66", features = [
     "IdbTransactionMode",
     "IdbVersionChangeEvent",
     "Window",
+    "WorkerGlobalScope",
 ] }
 
 [dev-dependencies]

src/factory.rs

@@ -6,9 +6,9 @@ use futures_util::{
 };
 use std::{future::Future, marker::PhantomData, sync::atomic::Ordering};
 use web_sys::{
-    js_sys::Function,
+    js_sys::{self, Function},
     wasm_bindgen::{closure::Closure, JsCast, JsValue},
-    IdbDatabase, IdbFactory, IdbOpenDbRequest, IdbVersionChangeEvent,
+    IdbDatabase, IdbFactory, IdbOpenDbRequest, IdbVersionChangeEvent, WorkerGlobalScope,
 };
 
 /// Wrapper for [`IDBFactory`](https://developer.mozilla.org/en-US/docs/Web/API/IDBFactory)
@@ -28,11 +28,18 @@ impl<Err: 'static> Factory<Err> {
     ///
     /// This internally uses [`indexedDB`](https://developer.mozilla.org/en-US/docs/Web/API/indexedDB).
     pub fn get() -> crate::Result<Factory<Err>, Err> {
-        let window = web_sys::window().ok_or(crate::Error::NotInBrowser)?;
-        let sys = window
-            .indexed_db()
+        let indexed_db = if let Some(window) = web_sys::window() {
+            window.indexed_db()
+        } else if let Ok(worker_scope) = js_sys::global().dyn_into::<WorkerGlobalScope>() {
+            worker_scope.indexed_db()
+        } else {
+            return Err(crate::Error::NotInBrowser);
+        };
+
+        let sys = indexed_db
             .map_err(|_| crate::Error::IndexedDbDisabled)?
             .ok_or(crate::Error::IndexedDbDisabled)?;
+
         Ok(Factory {
             sys,
             _phantom: PhantomData,

tests/browser.rs

@@ -0,0 +1,3 @@
+wasm_bindgen_test::wasm_bindgen_test_configure!(run_in_browser);
+
+mod common;

tests/js.rs renamed to tests/common/js.rs

@@ -1,12 +1,10 @@
 use indexed_db::{Error, Factory};
-use wasm_bindgen_test::{wasm_bindgen_test, wasm_bindgen_test_configure};
+use wasm_bindgen_test::wasm_bindgen_test;
 use web_sys::{
     js_sys::{JsString, Number},
     wasm_bindgen::JsValue,
 };
 
-wasm_bindgen_test_configure!(run_in_browser);
-
 #[wasm_bindgen_test]
 async fn smoke_test() {
     // tracing_wasm::set_as_global_default();

tests/js-panic.rs renamed to tests/common/js_panic.rs

@@ -1,10 +1,8 @@
 use anyhow::Context;
 use indexed_db::Factory;
-use wasm_bindgen_test::{wasm_bindgen_test, wasm_bindgen_test_configure};
+use wasm_bindgen_test::wasm_bindgen_test;
 use web_sys::js_sys::JsString;
 
-wasm_bindgen_test_configure!(run_in_browser);
-
 // Currently, panic=abort, so the sender is never Drop'd in the callback handlers.
 // As such, despite the proper error message being displayed by the console error
 // panic hook, `should_panic` never detects it due to the deadlock happening first.

tests/common/mod.rs

@@ -0,0 +1,2 @@
+mod js;
+mod js_panic;

tests/worker.rs

@@ -0,0 +1,3 @@
+wasm_bindgen_test::wasm_bindgen_test_configure!(run_in_worker);
+
+mod common;