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

use yew::prelude::*;

const SLOT: &str = "onIcon";

/// Props for [`MatOnIconButtonToggle`]
#[derive(Properties, PartialEq, Clone)]
pub struct OnIconButtonToggleProps {
    pub children: Children,
}

/// Defines header for [`MatIconButtonToggle`][crate::MatIconButtonToggle].
///
/// If the child passed is an element (a `VTag`), then it is modified to include
/// the appropriate attributes. Otherwise, the child is wrapped in a `span`
/// containing said attributes.
pub struct MatOnIconButtonToggle {}

impl Component for MatOnIconButtonToggle {
    type Message = ();
    type Properties = OnIconButtonToggleProps;

    fn create(_: &Context<Self>) -> Self {
        Self {}
    }

    fn view(&self, ctx: &Context<Self>) -> Html {
        let props = ctx.props();
        let children = props
            .children
            .iter()
            .map(|child| {
                match child {
                    Html::VTag(mut vtag) => {
                        vtag.add_attribute("slot", SLOT);
                        Html::VTag(vtag)
                    }
                    _ => {
                        html! {
                             <span slot={SLOT}>
                                 {child}
                             </span>
                        }
                    }
                }
            })
            .collect::<Html>();

        html! {
             {children}
        }
    }
}