utils: Add body() and document_element() getters (#161)

rustwasm · Oct 26, 2021 · 5b67085 · 5b67085
1 parent 53c6425
commit 5b67085
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/crates/utils/Cargo.toml b/crates/utils/Cargo.toml
@@ -20,6 +20,7 @@ version = "0.3"
 features = [
     "Document",
     "History",
+    "HtmlElement",
     "Location",
     "Window"
 ]
diff --git a/crates/utils/src/lib.rs b/crates/utils/src/lib.rs
@@ -11,6 +11,18 @@ pub fn document() -> web_sys::Document {
     window().document().expect_throw("Can't find document")
 }
 
+/// Convenience function to access `document.body`.
+pub fn body() -> web_sys::HtmlElement {
+    document().body().expect_throw("Can't find document body")
+}
+
+/// Convenience function to access `document.documentElement`.
+pub fn document_element() -> web_sys::Element {
+    document()
+        .document_element()
+        .expect_throw("Can't find document element")
+}
+
 /// Convenience function to access the web_sys history.
 pub fn history() -> web_sys::History {
     window().history().expect_throw("Can't find history")