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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

use yew::prelude::*;

use crate::root_state::BounceRootState;

/// Properties for [`BounceRoot`].
#[derive(Properties, Debug, PartialEq)]
pub struct BounceRootProps {
    /// Children of a Bounce Root.
    #[prop_or_default]
    pub children: Children,
}

/// A `<BounceRoot />`.
///
/// For bounce states to function, A `<BounceRoot />` must present and registered as a context
/// provider.
///
/// # Example
///
/// ```
/// # use yew::prelude::*;
/// # use bounce::prelude::*;
/// # use bounce::BounceRoot;
/// #[function_component(App)]
/// fn app() -> Html {
///     html! {
///         <BounceRoot>
///             // children...
///         </BounceRoot>
///     }
/// }
///
/// ```
#[function_component(BounceRoot)]
pub fn bounce_root(props: &BounceRootProps) -> Html {
    let children = props.children.clone();

    let root_state = (*use_state(BounceRootState::new)).clone();

    {
        let root_state = root_state.clone();
        use_effect_with_deps(
            move |_| {
                // We clear all states manually.
                move || {
                    root_state.clear();
                }
            },
            (),
        );
    }

    #[allow(clippy::unused_unit)]
    {
        let _root_state = root_state.clone();
        let _ = use_transitive_state!(
            move |_| -> () {
                #[cfg(feature = "ssr")]
                #[cfg(feature = "helmet")]
                {
                    // Workaround to send helmet states back to static writer
                    use crate::helmet::StaticWriterState;

                    let states = _root_state.states();
                    let writer_state = states.get_atom_value::<StaticWriterState>();

                    if let Some(ref w) = writer_state.writer {
                        w.send_helmet(
                            states,
                            writer_state.format_title.clone(),
                            writer_state.default_title.clone(),
                        );
                    }
                }

                // We drop the root state on SSR as well.
                _root_state.clear();
            },
            ()
        );
    }

    html! {
        <ContextProvider<BounceRootState> context={root_state}>{children}</ContextProvider<BounceRootState>>
    }
}