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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

use std::sync::{
    Arc,
    Mutex,
};

use freya_common::Layers;
use freya_native_core::{
    attributes::AttributeName,
    exports::shipyard::Component,
    node_ref::NodeView,
    prelude::{
        AttributeMaskBuilder,
        Dependancy,
        NodeMaskBuilder,
        State,
    },
    NodeId,
    SendAnyMap,
};
use freya_native_core_macro::partial_derive_state;

use crate::{
    CustomAttributeValues,
    ParseAttribute,
    ParseError,
};

#[derive(Default, PartialEq, Clone, Debug, Component)]
pub struct LayerState {
    pub layer: i16,
    pub layer_for_children: i16,
}

impl ParseAttribute for LayerState {
    fn parse_attribute(
        &mut self,
        attr: freya_native_core::prelude::OwnedAttributeView<CustomAttributeValues>,
    ) -> Result<(), crate::ParseError> {
        #[allow(clippy::single_match)]
        match attr.attribute {
            AttributeName::Layer => {
                if let Some(value) = attr.value.as_text() {
                    let layer = value.parse::<i16>().map_err(|_| ParseError)?;
                    self.layer -= layer;
                    self.layer_for_children += layer;
                }
            }
            _ => {}
        }

        Ok(())
    }
}

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

    type ChildDependencies = ();

    type NodeDependencies = ();

    const NODE_MASK: NodeMaskBuilder<'static> = NodeMaskBuilder::new()
        .with_attrs(AttributeMaskBuilder::Some(&[AttributeName::Layer]))
        .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 root_id = context.get::<NodeId>().unwrap();
        let layers = context.get::<Arc<Mutex<Layers>>>().unwrap();
        let inherited_layer = parent.map(|(p,)| p.layer_for_children).unwrap_or(0i16);

        let mut layer_state = LayerState {
            layer: node_view.height() as i16 - inherited_layer,
            layer_for_children: inherited_layer,
        };

        if let Some(attributes) = node_view.attributes() {
            for attr in attributes {
                layer_state.parse_safe(attr);
            }
        }

        let changed = &layer_state != self;

        let is_orphan = node_view.height() == 0 && node_view.node_id() != *root_id;

        if changed && !is_orphan {
            layers
                .lock()
                .unwrap()
                .insert_node_in_layer(node_view.node_id(), layer_state.layer);
        }

        *self = layer_state;
        changed
    }
}