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

use dioxus_native_core::{
    attributes::AttributeName,
    exports::shipyard::Component,
    node_ref::NodeView,
    prelude::{AttributeMaskBuilder, Dependancy, NodeMaskBuilder, State},
    NodeId, SendAnyMap,
};
use dioxus_native_core_macro::partial_derive_state;

use crate::{CustomAttributeValues, OverflowMode, Parse};

#[derive(Default, PartialEq, Clone, Debug, Component)]
pub struct ViewportState {
    pub viewports: Vec<NodeId>,
    pub node_id: NodeId,
    pub overflow: OverflowMode,
}

#[partial_derive_state]
impl State<CustomAttributeValues> for ViewportState {
    type ParentDependencies = (Self,);

    type ChildDependencies = ();

    type NodeDependencies = ();

    const NODE_MASK: NodeMaskBuilder<'static> = NodeMaskBuilder::new()
        .with_attrs(AttributeMaskBuilder::Some(&[AttributeName::Overflow]))
        .with_tag();

    fn update<'a>(
        &mut self,
        node_view: NodeView<CustomAttributeValues>,
        _node: <Self::NodeDependencies as Dependancy>::ElementBorrowed<'a>,
        parent: Option<<Self::ParentDependencies as Dependancy>::ElementBorrowed<'a>>,
        _children: Vec<<Self::ChildDependencies as Dependancy>::ElementBorrowed<'a>>,
        _context: &SendAnyMap,
    ) -> bool {
        if !node_view.node_type().is_visible_element() {
            return false;
        }

        let mut viewports_state = ViewportState {
            node_id: node_view.node_id(),
            ..Default::default()
        };

        if let Some(attributes) = node_view.attributes() {
            for attr in attributes {
                #[allow(clippy::single_match)]
                match attr.attribute {
                    AttributeName::Overflow => {
                        if let Some(value) = attr.value.as_text() {
                            if let Ok(overflow) = OverflowMode::parse(value) {
                                viewports_state.overflow = overflow;
                            }
                        }
                    }
                    _ => {}
                }
            }
        }

        if let Some((parent,)) = parent {
            viewports_state.viewports.extend(parent.viewports.clone());
            if parent.overflow == OverflowMode::Clip {
                viewports_state.viewports.push(parent.node_id);
            }
        }

        let changed = &viewports_state != self;
        *self = viewports_state;
        changed
    }
}