1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

//! Helper functions for accessing DOM environment
use wasm_bindgen::JsCast;
use web_sys::{Document, Element, Window};

/// Return the current browser [`web_sys::Window`] element
pub fn window() -> Window {
    web_sys::window().unwrap()
}

/// Return the current browser [`web_sys::Document`] element
pub fn document() -> Document {
    web_sys::window().unwrap().document().unwrap()
}

/// Return the `body` element of the current document
pub fn body() -> std::result::Result<Element, String> {
    let b = document()
        .query_selector("body")
        .unwrap()
        .ok_or_else(|| "Unable to get body element".to_string())?;
    Ok(b)
}

pub fn location() -> Result<web_sys::Location, wasm_bindgen::JsValue> {
    let location = js_sys::Reflect::get(&js_sys::global(), &"location".into())?;
    location.dyn_into()
}