use crate::html::{Component, ComponentLink, NodeRef, Scope};
use crate::utils::document;
use cfg_if::cfg_if;
cfg_if! {
if #[cfg(feature = "std_web")] {
use stdweb::web::{Element, INode, IParentNode};
} else if #[cfg(feature = "web_sys")] {
use web_sys::Element;
}
}
#[derive(Debug)]
pub struct App<COMP: Component> {
scope: Scope<COMP>,
}
impl<COMP> Default for App<COMP>
where
COMP: Component,
{
fn default() -> Self {
App::new()
}
}
impl<COMP> App<COMP>
where
COMP: Component,
COMP::Properties: Default,
{
pub fn mount(self, element: Element) -> ComponentLink<COMP> {
clear_element(&element);
self.scope.mount_in_place(
element,
NodeRef::default(),
None,
NodeRef::default(),
COMP::Properties::default(),
)
}
pub fn mount_to_body(self) -> ComponentLink<COMP> {
let element = document()
.query_selector("body")
.expect("can't get body node for rendering")
.expect("can't unwrap body node");
self.mount(element)
}
pub fn mount_as_body(self) -> ComponentLink<COMP> {
let html_element = document()
.query_selector("html")
.expect("can't get html node for rendering")
.expect("can't unwrap html node");
let body_element = document()
.query_selector("body")
.expect("can't get body node for rendering")
.expect("can't unwrap body node");
html_element
.remove_child(&body_element)
.expect("can't remove body child");
self.scope.mount_in_place(
html_element,
NodeRef::default(),
None,
NodeRef::default(),
COMP::Properties::default(),
)
}
}
impl<COMP> App<COMP>
where
COMP: Component,
{
pub fn new() -> Self {
let scope = Scope::new(None);
App { scope }
}
pub fn mount_with_props(
self,
element: Element,
props: COMP::Properties,
) -> ComponentLink<COMP> {
clear_element(&element);
self.scope
.mount_in_place(element, NodeRef::default(), None, NodeRef::default(), props)
}
pub fn mount_to_body_with_props(self, props: COMP::Properties) -> ComponentLink<COMP> {
let element = document()
.query_selector("body")
.expect("can't get body node for rendering")
.expect("can't unwrap body node");
self.mount_with_props(element, props)
}
pub fn mount_as_body_with_props(self, props: COMP::Properties) -> ComponentLink<COMP> {
let html_element = document()
.query_selector("html")
.expect("can't get html node for rendering")
.expect("can't unwrap html node");
let body_element = document()
.query_selector("body")
.expect("can't get body node for rendering")
.expect("can't unwrap body node");
html_element
.remove_child(&body_element)
.expect("can't remove body child");
self.scope.mount_in_place(
html_element,
NodeRef::default(),
None,
NodeRef::default(),
props,
)
}
}
fn clear_element(element: &Element) {
while let Some(child) = element.last_child() {
element.remove_child(&child).expect("can't remove a child");
}
}