use std::cell::Cell;

use dioxus::core as dioxus_core;
use dioxus::core::{ElementId, VElement, VText};
use dioxus::prelude::*;
use dioxus_native_core::real_dom::RealDom;
use dioxus_native_core::state::State;
use dioxus_native_core_macro::State;

#[derive(Default, Clone, State)]
struct Empty {}

#[test]
fn initial_build_simple() {
    use std::cell::Cell;

    #[allow(non_snake_case)]
    fn Base(cx: Scope) -> Element {
        rsx!(cx, div {})
    }

    let vdom = VirtualDom::new(Base);

    let mutations = vdom.create_vnodes(rsx! {
        div{}
    });

    let mut dom: RealDom<Empty> = RealDom::new();

    let _to_update = dom.apply_mutations(vec![mutations]);
    let root_div = VElement {
        id: Cell::new(Some(ElementId(1))),
        key: None,
        tag: "div",
        namespace: None,
        parent: Cell::new(Some(ElementId(0))),
        listeners: &[],
        attributes: &[],
        children: &[],
    };
    assert_eq!(dom.size(), 1);
    assert!(&dom.contains_node(&VNode::Element(&root_div)));
    assert_eq!(dom[1].height, 1);
}

#[test]
fn initial_build_with_children() {
    #[allow(non_snake_case)]
    fn Base(cx: Scope) -> Element {
        rsx!(cx, div {})
    }

    let vdom = VirtualDom::new(Base);

    let mutations = vdom.create_vnodes(rsx! {
        div{
            div{
                "hello"
                p{
                    "world"
                }
                "hello world"
            }
        }
    });

    let mut dom: RealDom<Empty> = RealDom::new();

    let _to_update = dom.apply_mutations(vec![mutations]);
    let first_text = VText {
        id: Cell::new(Some(ElementId(3))),
        text: "hello",
        is_static: true,
    };
    let first_text_node = VNode::Text(&first_text);
    let child_text = VText {
        id: Cell::new(Some(ElementId(5))),
        text: "world",
        is_static: true,
    };
    let child_text_node = VNode::Text(&child_text);
    let child_p_el = VElement {
        id: Cell::new(Some(ElementId(4))),
        key: None,
        tag: "p",
        namespace: None,
        parent: Cell::new(Some(ElementId(2))),
        listeners: &[],
        attributes: &[],
        children: &[child_text_node],
    };
    let child_p_node = VNode::Element(&child_p_el);
    let second_text = VText {
        id: Cell::new(Some(ElementId(6))),
        text: "hello world",
        is_static: true,
    };
    let second_text_node = VNode::Text(&second_text);
    let child_div_el = VElement {
        id: Cell::new(Some(ElementId(2))),
        key: None,
        tag: "div",
        namespace: None,
        parent: Cell::new(Some(ElementId(1))),
        listeners: &[],
        attributes: &[],
        children: &[first_text_node, child_p_node, second_text_node],
    };
    let child_div_node = VNode::Element(&child_div_el);
    let root_div = VElement {
        id: Cell::new(Some(ElementId(1))),
        key: None,
        tag: "div",
        namespace: None,
        parent: Cell::new(Some(ElementId(0))),
        listeners: &[],
        attributes: &[],
        children: &[child_div_node],
    };
    assert_eq!(dom.size(), 6);
    assert!(&dom.contains_node(&VNode::Element(&root_div)));
    assert_eq!(dom[1].height, 1);
    assert_eq!(dom[2].height, 2);
    assert_eq!(dom[3].height, 3);
    assert_eq!(dom[4].height, 3);
    assert_eq!(dom[5].height, 4);
    assert_eq!(dom[6].height, 3);
}