cranpose_ui/

modifier_nodes.rs

use std::{
    cell::Cell,
    hash::{Hash, Hasher},
    rc::Rc,
};

use cranpose_core::NodeId;
use cranpose_foundation::{
    Constraints, DelegatableNode, DrawModifierNode, InvalidationKind, LayoutModifierNode,
    Measurable, ModifierNode, ModifierNodeContext, ModifierNodeElement, NodeCapabilities,
    NodeState, PointerEvent, PointerEventKind, PointerInputNode, Size,
};
use cranpose_ui_graphics::PointerIcon;
use cranpose_ui_layout::{Alignment, HorizontalAlignment, IntrinsicSize, VerticalAlignment};

use crate::{
    draw::DrawCommand,
    modifier::{
        BlendMode, Color, ColorFilter, CompositingStrategy, EdgeInsets, GraphicsLayer,
        LayoutWeight, Point, RoundedCornerShape,
    },
};

fn hash_f32_value<H: Hasher>(state: &mut H, value: f32) {
    state.write_u32(value.to_bits());
}

fn hash_option_f32<H: Hasher>(state: &mut H, value: Option<f32>) {
    match value {
        Some(v) => {
            state.write_u8(1);
            hash_f32_value(state, v);
        }
        None => state.write_u8(0),
    }
}

fn hash_graphics_layer<H: Hasher>(state: &mut H, layer: &GraphicsLayer) {
    hash_f32_value(state, layer.alpha);
    hash_f32_value(state, layer.scale);
    hash_f32_value(state, layer.scale_x);
    hash_f32_value(state, layer.scale_y);
    hash_f32_value(state, layer.rotation_x);
    hash_f32_value(state, layer.rotation_y);
    hash_f32_value(state, layer.rotation_z);
    hash_f32_value(state, layer.camera_distance);
    hash_f32_value(state, layer.transform_origin.pivot_fraction_x);
    hash_f32_value(state, layer.transform_origin.pivot_fraction_y);
    hash_f32_value(state, layer.translation_x);
    hash_f32_value(state, layer.translation_y);
    hash_f32_value(state, layer.shadow_elevation);
    hash_f32_value(state, layer.ambient_shadow_color.r());
    hash_f32_value(state, layer.ambient_shadow_color.g());
    hash_f32_value(state, layer.ambient_shadow_color.b());
    hash_f32_value(state, layer.ambient_shadow_color.a());
    hash_f32_value(state, layer.spot_shadow_color.r());
    hash_f32_value(state, layer.spot_shadow_color.g());
    hash_f32_value(state, layer.spot_shadow_color.b());
    hash_f32_value(state, layer.spot_shadow_color.a());
    match layer.shape {
        crate::modifier::LayerShape::Rectangle => {
            state.write_u8(0);
        }
        crate::modifier::LayerShape::Rounded(shape) => {
            state.write_u8(1);
            let radii = shape.radii();
            hash_f32_value(state, radii.top_left);
            hash_f32_value(state, radii.top_right);
            hash_f32_value(state, radii.bottom_right);
            hash_f32_value(state, radii.bottom_left);
        }
    }
    state.write_u8(layer.clip as u8);
    match layer.color_filter {
        Some(ColorFilter::Tint(color)) => {
            state.write_u8(1);
            hash_f32_value(state, color.r());
            hash_f32_value(state, color.g());
            hash_f32_value(state, color.b());
            hash_f32_value(state, color.a());
        }
        Some(ColorFilter::Modulate(color)) => {
            state.write_u8(2);
            hash_f32_value(state, color.r());
            hash_f32_value(state, color.g());
            hash_f32_value(state, color.b());
            hash_f32_value(state, color.a());
        }
        Some(ColorFilter::Matrix(matrix)) => {
            state.write_u8(3);
            for value in matrix {
                hash_f32_value(state, value);
            }
        }
        None => state.write_u8(0),
    }
    state.write_u8(layer.render_effect.is_some() as u8);
    state.write_u8(layer.backdrop_effect.is_some() as u8);
    let compositing_tag = match layer.compositing_strategy {
        CompositingStrategy::Auto => 0,
        CompositingStrategy::Offscreen => 1,
        CompositingStrategy::ModulateAlpha => 2,
    };
    state.write_u8(compositing_tag);
    let blend_tag = match layer.blend_mode {
        BlendMode::Clear => 0,
        BlendMode::Src => 1,
        BlendMode::Dst => 2,
        BlendMode::SrcOver => 3,
        BlendMode::DstOver => 4,
        BlendMode::SrcIn => 5,
        BlendMode::DstIn => 6,
        BlendMode::SrcOut => 7,
        BlendMode::DstOut => 8,
        BlendMode::SrcAtop => 9,
        BlendMode::DstAtop => 10,
        BlendMode::Xor => 11,
        BlendMode::Plus => 12,
        BlendMode::Modulate => 13,
        BlendMode::Screen => 14,
        BlendMode::Overlay => 15,
        BlendMode::Darken => 16,
        BlendMode::Lighten => 17,
        BlendMode::ColorDodge => 18,
        BlendMode::ColorBurn => 19,
        BlendMode::HardLight => 20,
        BlendMode::SoftLight => 21,
        BlendMode::Difference => 22,
        BlendMode::Exclusion => 23,
        BlendMode::Multiply => 24,
        BlendMode::Hue => 25,
        BlendMode::Saturation => 26,
        BlendMode::Color => 27,
        BlendMode::Luminosity => 28,
    };
    state.write_u8(blend_tag);
}

fn hash_horizontal_alignment<H: Hasher>(state: &mut H, alignment: HorizontalAlignment) {
    let tag = match alignment {
        HorizontalAlignment::Start => 0,
        HorizontalAlignment::CenterHorizontally => 1,
        HorizontalAlignment::End => 2,
    };
    state.write_u8(tag);
}

fn hash_vertical_alignment<H: Hasher>(state: &mut H, alignment: VerticalAlignment) {
    let tag = match alignment {
        VerticalAlignment::Top => 0,
        VerticalAlignment::CenterVertically => 1,
        VerticalAlignment::Bottom => 2,
    };
    state.write_u8(tag);
}

fn hash_alignment<H: Hasher>(state: &mut H, alignment: Alignment) {
    hash_horizontal_alignment(state, alignment.horizontal);
    hash_vertical_alignment(state, alignment.vertical);
}

macro_rules! impl_layout_modifier_node {
    ($ty:ty) => {
        impl ModifierNode for $ty {
            fn as_layout_node(&self) -> Option<&dyn LayoutModifierNode> {
                Some(self)
            }

            fn as_layout_node_mut(&mut self) -> Option<&mut dyn LayoutModifierNode> {
                Some(self)
            }
        }
    };
    ($ty:ty, invalidate = $invalidation:expr) => {
        impl ModifierNode for $ty {
            fn on_attach(&mut self, context: &mut dyn ModifierNodeContext) {
                context.invalidate($invalidation);
            }

            fn as_layout_node(&self) -> Option<&dyn LayoutModifierNode> {
                Some(self)
            }

            fn as_layout_node_mut(&mut self) -> Option<&mut dyn LayoutModifierNode> {
                Some(self)
            }
        }
    };
}
pub(crate) use impl_layout_modifier_node;

macro_rules! impl_draw_modifier_node {
    ($ty:ty) => {
        impl ModifierNode for $ty {
            fn on_attach(&mut self, context: &mut dyn ModifierNodeContext) {
                context.invalidate(InvalidationKind::Draw);
            }

            fn as_draw_node(&self) -> Option<&dyn DrawModifierNode> {
                Some(self)
            }

            fn as_draw_node_mut(&mut self) -> Option<&mut dyn DrawModifierNode> {
                Some(self)
            }
        }

        impl DrawModifierNode for $ty {}
    };
}

macro_rules! forward_intrinsics_to_child {
    () => {
        fn min_intrinsic_width(
            &self,
            measurable: &dyn Measurable,
            height: f32,
            _density: f32,
        ) -> f32 {
            measurable.min_intrinsic_width(height)
        }

        fn max_intrinsic_width(
            &self,
            measurable: &dyn Measurable,
            height: f32,
            _density: f32,
        ) -> f32 {
            measurable.max_intrinsic_width(height)
        }

        fn min_intrinsic_height(
            &self,
            measurable: &dyn Measurable,
            width: f32,
            _density: f32,
        ) -> f32 {
            measurable.min_intrinsic_height(width)
        }

        fn max_intrinsic_height(
            &self,
            measurable: &dyn Measurable,
            width: f32,
            _density: f32,
        ) -> f32 {
            measurable.max_intrinsic_height(width)
        }
    };
}

macro_rules! impl_sink_reporter_element {
    ($element:ty, $node:ty) => {
        impl ModifierNodeElement for $element {
            type Node = $node;

            fn create(&self) -> Self::Node {
                <$node>::new(self.sink.clone())
            }

            fn update(&self, node: &mut Self::Node) {
                node.sink = self.sink.clone();
            }

            fn capabilities(&self) -> NodeCapabilities {
                NodeCapabilities::LAYOUT
            }
        }
    };
}

fn measure_pass_through(
    measurable: &dyn Measurable,
    constraints: Constraints,
    offset: impl FnOnce(Size) -> (f32, f32),
) -> cranpose_ui_layout::LayoutModifierMeasureResult {
    let placeable = measurable.measure(constraints);
    let size = Size {
        width: placeable.width(),
        height: placeable.height(),
    };
    let (x, y) = offset(size);
    cranpose_ui_layout::LayoutModifierMeasureResult::new(size, x, y)
}

fn attach_draw_observer(
    node_id_cell: &Cell<Option<NodeId>>,
    context: &mut dyn ModifierNodeContext,
) {
    node_id_cell.set(context.node_id());
    context.invalidate(InvalidationKind::Draw);
}

fn detach_draw_observer(node_id_cell: &Cell<Option<NodeId>>) {
    if let Some(node_id) = node_id_cell.replace(None) {
        crate::render_state::clear_draw_observations_for_node(node_id);
    }
}

enum SizeAxis {
    Width,
    Height,
}

fn size_intrinsic(
    target: Constraints,
    axis: SizeAxis,
    enforce_incoming: bool,
    cross: f32,
    intrinsic: impl FnOnce(f32) -> f32,
) -> f32 {
    let (target_min, target_max, cross_min, cross_max) = match axis {
        SizeAxis::Width => (
            target.min_width,
            target.max_width,
            target.min_height,
            target.max_height,
        ),
        SizeAxis::Height => (
            target.min_height,
            target.max_height,
            target.min_width,
            target.max_width,
        ),
    };
    if target_min == target_max && target_max != f32::INFINITY {
        target_max
    } else {
        let child_cross = if enforce_incoming {
            cross
        } else {
            cross.clamp(cross_min, cross_max)
        };
        intrinsic(child_cross).clamp(target_min, target_max)
    }
}

/// Node that adds padding around its content.
#[derive(Debug)]
pub struct PaddingNode {
    padding: EdgeInsets,
    state: NodeState,
}

impl PaddingNode {
    pub fn new(padding: EdgeInsets) -> Self {
        Self {
            padding,
            state: NodeState::new(),
        }
    }

    pub fn padding(&self) -> EdgeInsets {
        self.padding
    }
}

impl DelegatableNode for PaddingNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl_layout_modifier_node!(PaddingNode, invalidate = InvalidationKind::Layout);

impl LayoutModifierNode for PaddingNode {
    fn measure(
        &self,
        context: &mut dyn ModifierNodeContext,
        measurable: &dyn Measurable,
        constraints: Constraints,
    ) -> cranpose_ui_layout::LayoutModifierMeasureResult {
        let padding = device_padding(self.padding, context.density());
        let horizontal_padding = padding.horizontal_sum();
        let vertical_padding = padding.vertical_sum();

        let inner_constraints = Constraints {
            min_width: (constraints.min_width - horizontal_padding).max(0.0),
            max_width: (constraints.max_width - horizontal_padding).max(0.0),
            min_height: (constraints.min_height - vertical_padding).max(0.0),
            max_height: (constraints.max_height - vertical_padding).max(0.0),
        };

        let inner_placeable = measurable.measure(inner_constraints);
        let inner_width = inner_placeable.width();
        let inner_height = inner_placeable.height();

        let (width, height) = constraints.constrain(
            inner_width + horizontal_padding,
            inner_height + vertical_padding,
        );

        cranpose_ui_layout::LayoutModifierMeasureResult::new(
            Size { width, height },
            padding.left,
            padding.top,
        )
    }

    fn min_intrinsic_width(&self, measurable: &dyn Measurable, height: f32, density: f32) -> f32 {
        let padding = device_padding(self.padding, density);
        let inner_height = (height - padding.vertical_sum()).max(0.0);
        measurable.min_intrinsic_width(inner_height) + padding.horizontal_sum()
    }

    fn max_intrinsic_width(&self, measurable: &dyn Measurable, height: f32, density: f32) -> f32 {
        let padding = device_padding(self.padding, density);
        let inner_height = (height - padding.vertical_sum()).max(0.0);
        measurable.max_intrinsic_width(inner_height) + padding.horizontal_sum()
    }

    fn min_intrinsic_height(&self, measurable: &dyn Measurable, width: f32, density: f32) -> f32 {
        let padding = device_padding(self.padding, density);
        let inner_width = (width - padding.horizontal_sum()).max(0.0);
        measurable.min_intrinsic_height(inner_width) + padding.vertical_sum()
    }

    fn max_intrinsic_height(&self, measurable: &dyn Measurable, width: f32, density: f32) -> f32 {
        let padding = device_padding(self.padding, density);
        let inner_width = (width - padding.horizontal_sum()).max(0.0);
        measurable.max_intrinsic_height(inner_width) + padding.vertical_sum()
    }
}

/// `padding` with each side on a whole device pixel of `density`, as
/// Compose's `roundToPx` puts every padding side.
pub(crate) fn device_padding(padding: EdgeInsets, density: f32) -> EdgeInsets {
    use cranpose_ui_layout::round_to_px;
    EdgeInsets {
        left: round_to_px(padding.left, density),
        top: round_to_px(padding.top, density),
        right: round_to_px(padding.right, density),
        bottom: round_to_px(padding.bottom, density),
    }
}

/// Element that creates and updates padding nodes.
#[derive(Debug, Clone, PartialEq)]
pub struct PaddingElement {
    padding: EdgeInsets,
}

impl PaddingElement {
    pub fn new(padding: EdgeInsets) -> Self {
        Self { padding }
    }
}

impl Hash for PaddingElement {
    fn hash<H: Hasher>(&self, state: &mut H) {
        hash_f32_value(state, self.padding.left);
        hash_f32_value(state, self.padding.top);
        hash_f32_value(state, self.padding.right);
        hash_f32_value(state, self.padding.bottom);
    }
}

impl ModifierNodeElement for PaddingElement {
    type Node = PaddingNode;

    fn create(&self) -> Self::Node {
        PaddingNode::new(self.padding)
    }

    fn update(&self, node: &mut Self::Node) {
        if node.padding != self.padding {
            node.padding = self.padding;
        }
    }

    fn capabilities(&self) -> NodeCapabilities {
        NodeCapabilities::LAYOUT
    }
}

/// Node that draws a background behind its content.
#[derive(Debug)]
pub struct BackgroundNode {
    color: Color,
    shape: Option<RoundedCornerShape>,
    state: NodeState,
}

impl BackgroundNode {
    pub fn new(color: Color) -> Self {
        Self {
            color,
            shape: None,
            state: NodeState::new(),
        }
    }

    pub fn color(&self) -> Color {
        self.color
    }

    pub fn shape(&self) -> Option<RoundedCornerShape> {
        self.shape
    }
}

impl DelegatableNode for BackgroundNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl_draw_modifier_node!(BackgroundNode);

/// Element that creates and updates background nodes.
#[derive(Debug, Clone, PartialEq)]
pub struct BackgroundElement {
    color: Color,
}

impl BackgroundElement {
    pub fn new(color: Color) -> Self {
        Self { color }
    }
}

impl Hash for BackgroundElement {
    fn hash<H: Hasher>(&self, state: &mut H) {
        hash_f32_value(state, self.color.0);
        hash_f32_value(state, self.color.1);
        hash_f32_value(state, self.color.2);
        hash_f32_value(state, self.color.3);
    }
}

impl ModifierNodeElement for BackgroundElement {
    type Node = BackgroundNode;

    fn create(&self) -> Self::Node {
        BackgroundNode::new(self.color)
    }

    fn update(&self, node: &mut Self::Node) {
        if node.color != self.color {
            node.color = self.color;
        }
    }

    fn capabilities(&self) -> NodeCapabilities {
        NodeCapabilities::DRAW
    }
}

/// Node that tracks the latest rounded corner shape.
#[derive(Debug)]
pub struct CornerShapeNode {
    shape: RoundedCornerShape,
    state: NodeState,
}

impl CornerShapeNode {
    pub fn new(shape: RoundedCornerShape) -> Self {
        Self {
            shape,
            state: NodeState::new(),
        }
    }

    pub fn shape(&self) -> RoundedCornerShape {
        self.shape
    }
}

impl DelegatableNode for CornerShapeNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl_draw_modifier_node!(CornerShapeNode);

/// Element that creates and updates corner shape nodes.
#[derive(Debug, Clone, PartialEq)]
pub struct CornerShapeElement {
    shape: RoundedCornerShape,
}

impl CornerShapeElement {
    pub fn new(shape: RoundedCornerShape) -> Self {
        Self { shape }
    }
}

impl Hash for CornerShapeElement {
    fn hash<H: Hasher>(&self, state: &mut H) {
        let radii = self.shape.radii();
        hash_f32_value(state, radii.top_left);
        hash_f32_value(state, radii.top_right);
        hash_f32_value(state, radii.bottom_right);
        hash_f32_value(state, radii.bottom_left);
    }
}

impl ModifierNodeElement for CornerShapeElement {
    type Node = CornerShapeNode;

    fn create(&self) -> Self::Node {
        CornerShapeNode::new(self.shape)
    }

    fn update(&self, node: &mut Self::Node) {
        if node.shape != self.shape {
            node.shape = self.shape;
        }
    }

    fn capabilities(&self) -> NodeCapabilities {
        NodeCapabilities::DRAW
    }
}

pub struct GraphicsLayerNode {
    layer: GraphicsLayer,
    layer_resolver: Option<Rc<dyn Fn() -> GraphicsLayer>>,
    node_id: Rc<Cell<Option<NodeId>>>,
    state: NodeState,
}

impl GraphicsLayerNode {
    pub fn new(layer: GraphicsLayer) -> Self {
        Self {
            layer,
            layer_resolver: None,
            node_id: Rc::new(Cell::new(None)),
            state: NodeState::new(),
        }
    }

    pub fn new_lazy(layer_resolver: Rc<dyn Fn() -> GraphicsLayer>) -> Self {
        Self {
            layer: GraphicsLayer::default(),
            layer_resolver: Some(layer_resolver),
            node_id: Rc::new(Cell::new(None)),
            state: NodeState::new(),
        }
    }

    #[cfg(test)]
    pub fn layer(&self) -> GraphicsLayer {
        if let Some(resolve) = self.layer_resolver() {
            resolve()
        } else {
            self.layer.clone()
        }
    }

    pub fn layer_snapshot(&self) -> GraphicsLayer {
        self.layer.clone()
    }

    pub fn layer_resolver(&self) -> Option<Rc<dyn Fn() -> GraphicsLayer>> {
        self.layer_resolver.as_ref().map(|resolve| {
            let resolve = resolve.clone();
            let node_id = Rc::clone(&self.node_id);
            Rc::new(move || {
                if let Some(node_id) = node_id.get() {
                    let scope = crate::render_state::DrawObservationScope::new(node_id, usize::MAX);
                    crate::render_state::observe_draw_reads(scope, || resolve())
                } else {
                    resolve()
                }
            }) as Rc<dyn Fn() -> GraphicsLayer>
        })
    }

    fn set_static(&mut self, layer: GraphicsLayer) {
        let changed = self.layer != layer || self.layer_resolver.is_some();
        self.layer = layer;
        self.layer_resolver = None;
        if changed && let Some(node_id) = self.node_id.get() {
            crate::render_state::schedule_draw_repass(node_id);
        }
    }

    fn set_lazy(&mut self, layer_resolver: Rc<dyn Fn() -> GraphicsLayer>) {
        let changed = self
            .layer_resolver
            .as_ref()
            .is_none_or(|current| !Rc::ptr_eq(current, &layer_resolver));
        self.layer_resolver = Some(layer_resolver);
        if changed && let Some(node_id) = self.node_id.get() {
            crate::render_state::schedule_draw_repass(node_id);
        }
    }
}

impl DelegatableNode for GraphicsLayerNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl ModifierNode for GraphicsLayerNode {
    fn on_attach(&mut self, context: &mut dyn ModifierNodeContext) {
        attach_draw_observer(&self.node_id, context);
    }

    fn on_detach(&mut self) {
        detach_draw_observer(&self.node_id);
    }
}

impl std::fmt::Debug for GraphicsLayerNode {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("GraphicsLayerNode")
            .field("layer", &self.layer)
            .field("lazy", &self.layer_resolver.is_some())
            .finish()
    }
}

#[derive(Debug, Clone, PartialEq)]
pub struct GraphicsLayerElement {
    layer: GraphicsLayer,
}

impl GraphicsLayerElement {
    pub fn new(layer: GraphicsLayer) -> Self {
        Self { layer }
    }
}

impl Hash for GraphicsLayerElement {
    fn hash<H: Hasher>(&self, state: &mut H) {
        hash_graphics_layer(state, &self.layer);
    }
}

impl ModifierNodeElement for GraphicsLayerElement {
    type Node = GraphicsLayerNode;

    fn create(&self) -> Self::Node {
        GraphicsLayerNode::new(self.layer.clone())
    }

    fn update(&self, node: &mut Self::Node) {
        node.set_static(self.layer.clone());
    }

    fn capabilities(&self) -> NodeCapabilities {
        NodeCapabilities::DRAW
    }
}

#[derive(Clone)]
pub struct LazyGraphicsLayerElement {
    layer_resolver: Rc<dyn Fn() -> GraphicsLayer>,
}

impl LazyGraphicsLayerElement {
    pub fn new(layer_resolver: Rc<dyn Fn() -> GraphicsLayer>) -> Self {
        Self { layer_resolver }
    }
}

impl std::fmt::Debug for LazyGraphicsLayerElement {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("LazyGraphicsLayerElement")
            .field("resolver", &"<closure>")
            .finish()
    }
}

impl PartialEq for LazyGraphicsLayerElement {
    fn eq(&self, other: &Self) -> bool {
        Rc::ptr_eq(&self.layer_resolver, &other.layer_resolver)
    }
}

impl Eq for LazyGraphicsLayerElement {}

impl Hash for LazyGraphicsLayerElement {
    fn hash<H: Hasher>(&self, state: &mut H) {
        let ptr = Rc::as_ptr(&self.layer_resolver) as *const ();
        ptr.hash(state);
    }
}

impl ModifierNodeElement for LazyGraphicsLayerElement {
    type Node = GraphicsLayerNode;

    fn create(&self) -> Self::Node {
        GraphicsLayerNode::new_lazy(self.layer_resolver.clone())
    }

    fn update(&self, node: &mut Self::Node) {
        node.set_lazy(self.layer_resolver.clone());
    }

    fn capabilities(&self) -> NodeCapabilities {
        NodeCapabilities::DRAW
    }

    fn always_update(&self) -> bool {
        true
    }

    fn auto_invalidate_on_update(&self) -> bool {
        false
    }
}

/// Node that enforces size constraints on its content.
///
/// Matches Kotlin: `SizeNode` in foundation-layout/src/commonMain/kotlin/androidx/compose/foundation/layout/Size.kt
#[derive(Debug)]
pub struct SizeNode {
    min_width: Option<f32>,
    max_width: Option<f32>,
    min_height: Option<f32>,
    max_height: Option<f32>,
    enforce_incoming: bool,
    state: NodeState,
}

impl SizeNode {
    pub fn new(
        min_width: Option<f32>,
        max_width: Option<f32>,
        min_height: Option<f32>,
        max_height: Option<f32>,
        enforce_incoming: bool,
    ) -> Self {
        Self {
            min_width,
            max_width,
            min_height,
            max_height,
            enforce_incoming,
            state: NodeState::new(),
        }
    }

    /// The constraints the sizes ask for, each on a whole device pixel of
    /// `density` as Compose's `SizeNode` rounds them.
    fn target_constraints(&self, density: f32) -> Constraints {
        use cranpose_ui_layout::round_to_px;

        let max_width = self
            .max_width
            .map_or(f32::INFINITY, |v| round_to_px(v, density).max(0.0));
        let max_height = self
            .max_height
            .map_or(f32::INFINITY, |v| round_to_px(v, density).max(0.0));

        let min_width = self.min_width.map_or(0.0, |v| {
            let clamped = round_to_px(v, density).clamp(0.0, max_width);
            if clamped == f32::INFINITY {
                0.0
            } else {
                clamped
            }
        });

        let min_height = self.min_height.map_or(0.0, |v| {
            let clamped = round_to_px(v, density).clamp(0.0, max_height);
            if clamped == f32::INFINITY {
                0.0
            } else {
                clamped
            }
        });

        Constraints {
            min_width,
            max_width,
            min_height,
            max_height,
        }
    }

    pub fn min_width(&self) -> Option<f32> {
        self.min_width
    }

    pub fn max_width(&self) -> Option<f32> {
        self.max_width
    }

    pub fn min_height(&self) -> Option<f32> {
        self.min_height
    }

    pub fn max_height(&self) -> Option<f32> {
        self.max_height
    }

    pub fn enforce_incoming(&self) -> bool {
        self.enforce_incoming
    }
}

impl DelegatableNode for SizeNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl_layout_modifier_node!(SizeNode, invalidate = InvalidationKind::Layout);

impl LayoutModifierNode for SizeNode {
    fn measure(
        &self,
        context: &mut dyn ModifierNodeContext,
        measurable: &dyn Measurable,
        constraints: Constraints,
    ) -> cranpose_ui_layout::LayoutModifierMeasureResult {
        let target = self.target_constraints(context.density());

        let wrapped_constraints = if self.enforce_incoming {
            Constraints {
                min_width: target
                    .min_width
                    .max(constraints.min_width)
                    .min(constraints.max_width),
                max_width: target
                    .max_width
                    .min(constraints.max_width)
                    .max(constraints.min_width),
                min_height: target
                    .min_height
                    .max(constraints.min_height)
                    .min(constraints.max_height),
                max_height: target
                    .max_height
                    .min(constraints.max_height)
                    .max(constraints.min_height),
            }
        } else {
            let resolved_min_width = if self.min_width.is_some() {
                target.min_width
            } else {
                constraints.min_width.min(target.max_width)
            };
            let resolved_max_width = if self.max_width.is_some() {
                target.max_width
            } else {
                constraints.max_width.max(target.min_width)
            };
            let resolved_min_height = if self.min_height.is_some() {
                target.min_height
            } else {
                constraints.min_height.min(target.max_height)
            };
            let resolved_max_height = if self.max_height.is_some() {
                target.max_height
            } else {
                constraints.max_height.max(target.min_height)
            };

            Constraints {
                min_width: resolved_min_width,
                max_width: resolved_max_width,
                min_height: resolved_min_height,
                max_height: resolved_max_height,
            }
        };

        let placeable = measurable.measure(wrapped_constraints);
        let measured_width = placeable.width();
        let measured_height = placeable.height();

        let result_width = if self.min_width.is_some()
            && self.max_width.is_some()
            && self.min_width == self.max_width
            && target.min_width >= wrapped_constraints.min_width
            && target.min_width <= wrapped_constraints.max_width
        {
            target.min_width
        } else {
            measured_width
        };

        let result_height = if self.min_height.is_some()
            && self.max_height.is_some()
            && self.min_height == self.max_height
            && target.min_height >= wrapped_constraints.min_height
            && target.min_height <= wrapped_constraints.max_height
        {
            target.min_height
        } else {
            measured_height
        };

        cranpose_ui_layout::LayoutModifierMeasureResult::with_size(Size {
            width: result_width,
            height: result_height,
        })
    }

    fn min_intrinsic_width(&self, measurable: &dyn Measurable, height: f32, density: f32) -> f32 {
        size_intrinsic(
            self.target_constraints(density),
            SizeAxis::Width,
            self.enforce_incoming,
            height,
            |h| measurable.min_intrinsic_width(h),
        )
    }

    fn max_intrinsic_width(&self, measurable: &dyn Measurable, height: f32, density: f32) -> f32 {
        size_intrinsic(
            self.target_constraints(density),
            SizeAxis::Width,
            self.enforce_incoming,
            height,
            |h| measurable.max_intrinsic_width(h),
        )
    }

    fn min_intrinsic_height(&self, measurable: &dyn Measurable, width: f32, density: f32) -> f32 {
        size_intrinsic(
            self.target_constraints(density),
            SizeAxis::Height,
            self.enforce_incoming,
            width,
            |w| measurable.min_intrinsic_height(w),
        )
    }

    fn max_intrinsic_height(&self, measurable: &dyn Measurable, width: f32, density: f32) -> f32 {
        size_intrinsic(
            self.target_constraints(density),
            SizeAxis::Height,
            self.enforce_incoming,
            width,
            |w| measurable.max_intrinsic_height(w),
        )
    }
}

/// Element that creates and updates size nodes.
///
/// Matches Kotlin: `SizeElement` in foundation-layout/src/commonMain/kotlin/androidx/compose/foundation/layout/Size.kt
#[derive(Debug, Clone, PartialEq)]
pub struct SizeElement {
    min_width: Option<f32>,
    max_width: Option<f32>,
    min_height: Option<f32>,
    max_height: Option<f32>,
    enforce_incoming: bool,
}

impl SizeElement {
    pub fn new(width: Option<f32>, height: Option<f32>) -> Self {
        Self {
            min_width: width,
            max_width: width,
            min_height: height,
            max_height: height,
            enforce_incoming: true,
        }
    }

    pub fn with_constraints(
        min_width: Option<f32>,
        max_width: Option<f32>,
        min_height: Option<f32>,
        max_height: Option<f32>,
        enforce_incoming: bool,
    ) -> Self {
        Self {
            min_width,
            max_width,
            min_height,
            max_height,
            enforce_incoming,
        }
    }
}

impl Hash for SizeElement {
    fn hash<H: Hasher>(&self, state: &mut H) {
        hash_option_f32(state, self.min_width);
        hash_option_f32(state, self.max_width);
        hash_option_f32(state, self.min_height);
        hash_option_f32(state, self.max_height);
        self.enforce_incoming.hash(state);
    }
}

impl ModifierNodeElement for SizeElement {
    type Node = SizeNode;

    fn create(&self) -> Self::Node {
        SizeNode::new(
            self.min_width,
            self.max_width,
            self.min_height,
            self.max_height,
            self.enforce_incoming,
        )
    }

    fn update(&self, node: &mut Self::Node) {
        if node.min_width != self.min_width
            || node.max_width != self.max_width
            || node.min_height != self.min_height
            || node.max_height != self.max_height
            || node.enforce_incoming != self.enforce_incoming
        {
            node.min_width = self.min_width;
            node.max_width = self.max_width;
            node.min_height = self.min_height;
            node.max_height = self.max_height;
            node.enforce_incoming = self.enforce_incoming;
        }
    }

    fn capabilities(&self) -> NodeCapabilities {
        NodeCapabilities::LAYOUT
    }

    fn update_invalidation_kind(&self) -> Option<InvalidationKind> {
        Some(InvalidationKind::Layout)
    }
}

use std::cell::RefCell;

use cranpose_foundation::DRAG_THRESHOLD;

pub struct ClickableNode {
    on_press: Option<Rc<dyn Fn(Point)>>,
    on_click: Rc<dyn Fn(Point)>,
    state: NodeState,
    press_position: Rc<RefCell<Option<Point>>>,
    cached_handler: Rc<dyn Fn(PointerEvent)>,
}

impl std::fmt::Debug for ClickableNode {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("ClickableNode").finish()
    }
}

impl ClickableNode {
    pub fn new(on_click: impl Fn(Point) + 'static) -> Self {
        Self::with_handler(Rc::new(on_click))
    }

    pub fn with_handler(on_click: Rc<dyn Fn(Point)>) -> Self {
        Self::with_handlers(None, on_click)
    }

    pub fn with_handlers(on_press: Option<Rc<dyn Fn(Point)>>, on_click: Rc<dyn Fn(Point)>) -> Self {
        let press_position = Rc::new(RefCell::new(None));
        let cached_handler =
            Self::create_handler(on_press.clone(), on_click.clone(), press_position.clone());
        Self {
            on_press,
            on_click,
            state: NodeState::new(),
            press_position,
            cached_handler,
        }
    }

    fn create_handler(
        on_press: Option<Rc<dyn Fn(Point)>>,
        on_click: Rc<dyn Fn(Point)>,
        press_position: Rc<RefCell<Option<Point>>>,
    ) -> Rc<dyn Fn(PointerEvent)> {
        Rc::new(move |event: PointerEvent| {
            if event.id != 0 {
                return;
            }

            if event.is_consumed() {
                *press_position.borrow_mut() = None;
                return;
            }

            match event.kind {
                PointerEventKind::Down => {
                    *press_position.borrow_mut() = Some(event.travelled_to());
                    if let Some(on_press) = on_press.as_ref() {
                        on_press(event.position);
                    }
                }
                PointerEventKind::Move => {}
                PointerEventKind::Up => {
                    let press_pos_value = *press_position.borrow();

                    let should_click = if let Some(press_pos) = press_pos_value {
                        let travelled_to = event.travelled_to();
                        let dx = travelled_to.x - press_pos.x;
                        let dy = travelled_to.y - press_pos.y;
                        let distance = (dx * dx + dy * dy).sqrt();
                        distance <= DRAG_THRESHOLD
                    } else {
                        true
                    };

                    *press_position.borrow_mut() = None;

                    if should_click {
                        on_click(Point {
                            x: event.position.x,
                            y: event.position.y,
                        });
                        event.consume();
                    }
                }
                PointerEventKind::Cancel => {
                    *press_position.borrow_mut() = None;
                }
                PointerEventKind::Scroll
                | PointerEventKind::Zoom
                | PointerEventKind::RotaryScrollPre
                | PointerEventKind::RotaryScroll
                | PointerEventKind::Enter
                | PointerEventKind::Exit => {}
            }
        })
    }

    pub fn handler(&self) -> Rc<dyn Fn(Point)> {
        self.on_click.clone()
    }
}

impl DelegatableNode for ClickableNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl ModifierNode for ClickableNode {
    fn on_attach(&mut self, context: &mut dyn ModifierNodeContext) {
        context.invalidate(cranpose_foundation::InvalidationKind::PointerInput);
    }

    fn as_pointer_input_node(&self) -> Option<&dyn PointerInputNode> {
        Some(self)
    }

    fn as_pointer_input_node_mut(&mut self) -> Option<&mut dyn PointerInputNode> {
        Some(self)
    }
}

impl PointerInputNode for ClickableNode {
    fn on_pointer_event(
        &mut self,
        _context: &mut dyn ModifierNodeContext,
        event: &PointerEvent,
    ) -> bool {
        (self.cached_handler)(event.clone());
        event.is_consumed()
    }

    fn hit_test(&self, _x: f32, _y: f32) -> bool {
        true
    }

    fn pointer_input_handler(&self) -> Option<Rc<dyn Fn(PointerEvent)>> {
        Some(self.cached_handler.clone())
    }
}

/// Element that creates and updates clickable nodes.
#[derive(Clone)]
pub struct ClickableElement {
    on_press: Option<Rc<dyn Fn(Point)>>,
    on_click: Rc<dyn Fn(Point)>,
}

impl ClickableElement {
    pub fn new(on_click: impl Fn(Point) + 'static) -> Self {
        Self {
            on_press: None,
            on_click: Rc::new(on_click),
        }
    }

    pub fn with_handler(on_click: Rc<dyn Fn(Point)>) -> Self {
        Self {
            on_press: None,
            on_click,
        }
    }

    pub fn with_handlers(on_press: Rc<dyn Fn(Point)>, on_click: Rc<dyn Fn(Point)>) -> Self {
        Self {
            on_press: Some(on_press),
            on_click,
        }
    }
}

impl std::fmt::Debug for ClickableElement {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("ClickableElement").finish()
    }
}

impl PartialEq for ClickableElement {
    fn eq(&self, _other: &Self) -> bool {
        true
    }
}

impl Eq for ClickableElement {}

impl Hash for ClickableElement {
    fn hash<H: Hasher>(&self, state: &mut H) {
        "clickable".hash(state);
    }
}

impl ModifierNodeElement for ClickableElement {
    type Node = ClickableNode;

    fn create(&self) -> Self::Node {
        ClickableNode::with_handlers(self.on_press.clone(), self.on_click.clone())
    }

    fn update(&self, node: &mut Self::Node) {
        node.on_press.clone_from(&self.on_press);
        node.on_click.clone_from(&self.on_click);
        node.cached_handler = ClickableNode::create_handler(
            node.on_press.clone(),
            node.on_click.clone(),
            node.press_position.clone(),
        );
    }

    fn capabilities(&self) -> NodeCapabilities {
        NodeCapabilities::POINTER_INPUT
    }

    fn always_update(&self) -> bool {
        true
    }
}

/// Node that names the pointer's appearance over its layout node.
///
/// It handles no pointer events; it carries [`PointerIcon`] as pointer-input
/// data so the node becomes a hit target and the shell can read the icon off
/// the topmost region under the pointer.
#[derive(Debug)]
pub struct PointerIconNode {
    icon: PointerIcon,
    state: NodeState,
}

impl PointerIconNode {
    /// Creates a node requesting `icon` while the pointer is over it.
    pub fn new(icon: PointerIcon) -> Self {
        Self {
            icon,
            state: NodeState::new(),
        }
    }

    /// The icon this node requests.
    pub fn icon(&self) -> &PointerIcon {
        &self.icon
    }
}

impl DelegatableNode for PointerIconNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl ModifierNode for PointerIconNode {
    fn on_attach(&mut self, context: &mut dyn ModifierNodeContext) {
        context.invalidate(cranpose_foundation::InvalidationKind::PointerInput);
    }

    fn as_pointer_input_node(&self) -> Option<&dyn PointerInputNode> {
        Some(self)
    }

    fn as_pointer_input_node_mut(&mut self) -> Option<&mut dyn PointerInputNode> {
        Some(self)
    }
}

impl PointerInputNode for PointerIconNode {
    fn on_pointer_event(
        &mut self,
        _context: &mut dyn ModifierNodeContext,
        _event: &PointerEvent,
    ) -> bool {
        false
    }

    fn hit_test(&self, _x: f32, _y: f32) -> bool {
        true
    }

    fn pointer_input_handler(&self) -> Option<Rc<dyn Fn(PointerEvent)>> {
        None
    }
}

/// Element that creates and updates [`PointerIconNode`]s.
#[derive(Clone, Debug, PartialEq, Eq, Hash)]
pub struct PointerIconElement {
    icon: PointerIcon,
}

impl PointerIconElement {
    /// Creates an element requesting `icon`.
    pub fn new(icon: PointerIcon) -> Self {
        Self { icon }
    }
}

impl ModifierNodeElement for PointerIconElement {
    type Node = PointerIconNode;

    fn create(&self) -> Self::Node {
        PointerIconNode::new(self.icon.clone())
    }

    fn update(&self, node: &mut Self::Node) {
        node.icon = self.icon.clone();
    }

    fn capabilities(&self) -> NodeCapabilities {
        NodeCapabilities::POINTER_INPUT
    }
}

/// Node that applies alpha transparency to its content.
#[derive(Debug)]
pub struct AlphaNode {
    alpha: f32,
    state: NodeState,
}

impl AlphaNode {
    pub fn new(alpha: f32) -> Self {
        Self {
            alpha: alpha.clamp(0.0, 1.0),
            state: NodeState::new(),
        }
    }
}

impl DelegatableNode for AlphaNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl_draw_modifier_node!(AlphaNode);

/// Element that creates and updates alpha nodes.
#[derive(Debug, Clone, PartialEq)]
pub struct AlphaElement {
    alpha: f32,
}

impl AlphaElement {
    pub fn new(alpha: f32) -> Self {
        Self {
            alpha: alpha.clamp(0.0, 1.0),
        }
    }
}

impl Hash for AlphaElement {
    fn hash<H: Hasher>(&self, state: &mut H) {
        hash_f32_value(state, self.alpha);
    }
}

impl ModifierNodeElement for AlphaElement {
    type Node = AlphaNode;

    fn create(&self) -> Self::Node {
        AlphaNode::new(self.alpha)
    }

    fn update(&self, node: &mut Self::Node) {
        let new_alpha = self.alpha.clamp(0.0, 1.0);
        if (node.alpha - new_alpha).abs() > f32::EPSILON {
            node.alpha = new_alpha;
        }
    }

    fn capabilities(&self) -> NodeCapabilities {
        NodeCapabilities::DRAW
    }
}

#[derive(Debug)]
pub struct ClipToBoundsNode {
    state: NodeState,
}

impl ClipToBoundsNode {
    pub fn new() -> Self {
        Self {
            state: NodeState::new(),
        }
    }
}

impl DelegatableNode for ClipToBoundsNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl_draw_modifier_node!(ClipToBoundsNode);

#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub struct ClipToBoundsElement;

impl ClipToBoundsElement {
    pub fn new() -> Self {
        Self
    }
}

impl ModifierNodeElement for ClipToBoundsElement {
    type Node = ClipToBoundsNode;

    fn create(&self) -> Self::Node {
        ClipToBoundsNode::new()
    }

    fn update(&self, _node: &mut Self::Node) {}

    fn capabilities(&self) -> NodeCapabilities {
        NodeCapabilities::DRAW
    }
}

pub trait WindowRectSink {
    fn set(&self, rect: cranpose_ui_graphics::Rect);
}

impl WindowRectSink for Cell<cranpose_ui_graphics::Rect> {
    fn set(&self, rect: cranpose_ui_graphics::Rect) {
        Cell::set(self, rect);
    }
}

struct StateWindowRectSink(cranpose_core::MutableState<cranpose_ui_graphics::Rect>);

impl WindowRectSink for StateWindowRectSink {
    fn set(&self, rect: cranpose_ui_graphics::Rect) {
        self.0.set(rect);
    }
}

pub struct WindowRectReporterNode {
    sink: Rc<dyn WindowRectSink>,
    state: NodeState,
}

impl WindowRectReporterNode {
    pub(crate) fn new(sink: Rc<dyn WindowRectSink>) -> Self {
        Self {
            sink,
            state: NodeState::new(),
        }
    }

    pub(crate) fn window_rect_sink(&self) -> Rc<dyn WindowRectSink> {
        self.sink.clone()
    }
}

impl DelegatableNode for WindowRectReporterNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl_layout_modifier_node!(WindowRectReporterNode);

impl LayoutModifierNode for WindowRectReporterNode {
    fn measure(
        &self,
        _context: &mut dyn ModifierNodeContext,
        measurable: &dyn Measurable,
        constraints: Constraints,
    ) -> cranpose_ui_layout::LayoutModifierMeasureResult {
        measure_pass_through(measurable, constraints, |_| (0.0, 0.0))
    }
}

#[derive(Clone)]
pub struct WindowRectReporterElement {
    sink: Rc<dyn WindowRectSink>,
}

impl WindowRectReporterElement {
    pub fn new(sink: Rc<Cell<cranpose_ui_graphics::Rect>>) -> Self {
        Self { sink }
    }

    pub fn from_state(sink: cranpose_core::MutableState<cranpose_ui_graphics::Rect>) -> Self {
        Self {
            sink: Rc::new(StateWindowRectSink(sink)),
        }
    }
}

impl std::fmt::Debug for WindowRectReporterElement {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("WindowRectReporterElement").finish()
    }
}

impl PartialEq for WindowRectReporterElement {
    fn eq(&self, other: &Self) -> bool {
        Rc::ptr_eq(&self.sink, &other.sink)
    }
}

impl Eq for WindowRectReporterElement {}

impl Hash for WindowRectReporterElement {
    fn hash<H: Hasher>(&self, state: &mut H) {
        std::ptr::hash(Rc::as_ptr(&self.sink).cast::<()>(), state);
    }
}

impl_sink_reporter_element!(WindowRectReporterElement, WindowRectReporterNode);

/// Tells a selectable text's geometry where its content sits: slice
/// collection hands it the padding in front of the text and layout the
/// node's window origin, which is what a container needs to find the text
/// under a pointer.
pub(crate) struct SelectableTextNode {
    sink: Rc<crate::selection_container::SelectableGeometry>,
    state: NodeState,
}

impl SelectableTextNode {
    pub(crate) fn new(sink: Rc<crate::selection_container::SelectableGeometry>) -> Self {
        Self {
            sink,
            state: NodeState::new(),
        }
    }

    pub(crate) fn geometry(&self) -> &crate::selection_container::SelectableGeometry {
        &self.sink
    }
}

impl DelegatableNode for SelectableTextNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl_layout_modifier_node!(SelectableTextNode);

impl LayoutModifierNode for SelectableTextNode {
    fn measure(
        &self,
        _context: &mut dyn ModifierNodeContext,
        measurable: &dyn Measurable,
        constraints: Constraints,
    ) -> cranpose_ui_layout::LayoutModifierMeasureResult {
        measure_pass_through(measurable, constraints, |_| (0.0, 0.0))
    }
}

#[derive(Clone)]
pub(crate) struct SelectableTextElement {
    sink: Rc<crate::selection_container::SelectableGeometry>,
}

impl SelectableTextElement {
    pub(crate) fn new(sink: Rc<crate::selection_container::SelectableGeometry>) -> Self {
        Self { sink }
    }
}

impl std::fmt::Debug for SelectableTextElement {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("SelectableTextElement").finish()
    }
}

impl PartialEq for SelectableTextElement {
    fn eq(&self, other: &Self) -> bool {
        Rc::ptr_eq(&self.sink, &other.sink)
    }
}

impl Eq for SelectableTextElement {}

impl Hash for SelectableTextElement {
    fn hash<H: Hasher>(&self, state: &mut H) {
        std::ptr::hash(Rc::as_ptr(&self.sink), state);
    }
}

impl_sink_reporter_element!(SelectableTextElement, SelectableTextNode);

pub trait SizeSink {
    fn set(&self, size: Size);
}

impl SizeSink for Cell<Size> {
    fn set(&self, size: Size) {
        Cell::set(self, size);
    }
}

struct StateSizeSink(cranpose_core::MutableState<Size>);

impl SizeSink for StateSizeSink {
    fn set(&self, size: Size) {
        self.0.set(size);
    }
}

pub struct SizeReporterNode {
    sink: Rc<dyn SizeSink>,
    state: NodeState,
    #[cfg(debug_assertions)]
    oscillation: Cell<(Size, Size, u32)>,
}

impl SizeReporterNode {
    pub fn new(sink: Rc<dyn SizeSink>) -> Self {
        Self {
            sink,
            state: NodeState::new(),
            #[cfg(debug_assertions)]
            oscillation: Cell::new((Size::default(), Size::default(), 0)),
        }
    }

    #[cfg(debug_assertions)]
    fn check_oscillation(&self, size: Size) {
        const ALTERNATION_CEILING: u32 = 64;
        let (last, second_last, count) = self.oscillation.get();
        let count = if size == second_last && size != last {
            count + 1
        } else if size == last {
            count
        } else {
            0
        };
        assert!(
            count <= ALTERNATION_CEILING,
            "size-reactive feedback loop: this node's measured size has \
             alternated between {last:?} and {size:?} for {count} passes — \
             its content's size depends on the size it reports (the \
             onSizeChanged self-reference hazard). Break the cycle by making \
             the reported size feed only content that does not change this \
             node's own measured size."
        );
        self.oscillation.set((size, last, count));
    }
}

impl DelegatableNode for SizeReporterNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl_layout_modifier_node!(SizeReporterNode);

impl LayoutModifierNode for SizeReporterNode {
    fn measure(
        &self,
        _context: &mut dyn ModifierNodeContext,
        measurable: &dyn Measurable,
        constraints: Constraints,
    ) -> cranpose_ui_layout::LayoutModifierMeasureResult {
        measure_pass_through(measurable, constraints, |size| {
            #[cfg(debug_assertions)]
            self.check_oscillation(size);
            self.sink.set(size);
            (0.0, 0.0)
        })
    }
}

#[derive(Clone)]
pub struct SizeReporterElement {
    sink: Rc<dyn SizeSink>,
}

impl SizeReporterElement {
    pub fn new(sink: Rc<Cell<Size>>) -> Self {
        Self { sink }
    }

    pub fn from_state(sink: cranpose_core::MutableState<Size>) -> Self {
        Self {
            sink: Rc::new(StateSizeSink(sink)),
        }
    }
}

impl std::fmt::Debug for SizeReporterElement {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("SizeReporterElement").finish()
    }
}

impl PartialEq for SizeReporterElement {
    fn eq(&self, other: &Self) -> bool {
        std::ptr::addr_eq(Rc::as_ptr(&self.sink), Rc::as_ptr(&other.sink))
    }
}

impl Hash for SizeReporterElement {
    fn hash<H: Hasher>(&self, state: &mut H) {
        (Rc::as_ptr(&self.sink) as *const () as usize).hash(state);
    }
}

impl_sink_reporter_element!(SizeReporterElement, SizeReporterNode);

pub struct DrawCommandNode {
    commands: Vec<DrawCommand>,
    node_id: Cell<Option<NodeId>>,
    state: NodeState,
}

impl DrawCommandNode {
    pub fn new(commands: Vec<DrawCommand>) -> Self {
        Self {
            commands,
            node_id: Cell::new(None),
            state: NodeState::new(),
        }
    }

    #[cfg(test)]
    pub fn commands(&self) -> &[DrawCommand] {
        &self.commands
    }

    pub(crate) fn observed_commands(&self) -> Vec<DrawCommand> {
        let node_id = self.node_id.get();
        self.commands
            .iter()
            .cloned()
            .enumerate()
            .map(|(index, command)| observe_draw_command(command, node_id, index))
            .collect()
    }
}

impl DelegatableNode for DrawCommandNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl ModifierNode for DrawCommandNode {
    fn on_attach(&mut self, context: &mut dyn ModifierNodeContext) {
        attach_draw_observer(&self.node_id, context);
    }

    fn on_detach(&mut self) {
        detach_draw_observer(&self.node_id);
    }

    fn as_draw_node(&self) -> Option<&dyn DrawModifierNode> {
        Some(self)
    }

    fn as_draw_node_mut(&mut self) -> Option<&mut dyn DrawModifierNode> {
        Some(self)
    }
}

impl DrawModifierNode for DrawCommandNode {}

fn observe_draw_command(
    command: DrawCommand,
    node_id: Option<NodeId>,
    command_index: usize,
) -> DrawCommand {
    let Some(node_id) = node_id else {
        return command;
    };
    let observation = crate::render_state::DrawObservationScope::new(node_id, command_index);
    match command {
        DrawCommand::Behind(draw) => DrawCommand::Behind(Rc::new(move |scope| {
            crate::render_state::observe_draw_reads(observation, || draw(scope));
        })),
        DrawCommand::WithContent(draw) => DrawCommand::WithContent(Rc::new(move |scope| {
            crate::render_state::observe_draw_reads(observation, || draw(scope));
        })),
        DrawCommand::Overlay(draw) => DrawCommand::Overlay(Rc::new(move |scope| {
            crate::render_state::observe_draw_reads(observation, || draw(scope));
        })),
    }
}

fn draw_command_tag(cmd: &DrawCommand) -> u8 {
    match cmd {
        DrawCommand::Behind(_) => 0,
        DrawCommand::WithContent(_) => 1,
        DrawCommand::Overlay(_) => 2,
    }
}

fn draw_command_closure_identity(cmd: &DrawCommand) -> *const () {
    match cmd {
        DrawCommand::Behind(f) | DrawCommand::WithContent(f) | DrawCommand::Overlay(f) => {
            Rc::as_ptr(f) as *const ()
        }
    }
}

#[derive(Clone)]
pub struct DrawCommandElement {
    commands: Vec<DrawCommand>,
}

impl DrawCommandElement {
    pub fn new(command: DrawCommand) -> Self {
        Self {
            commands: vec![command],
        }
    }

    pub fn from_commands(commands: Vec<DrawCommand>) -> Self {
        Self { commands }
    }
}

impl std::fmt::Debug for DrawCommandElement {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("DrawCommandElement")
            .field("commands", &self.commands.len())
            .finish()
    }
}

impl PartialEq for DrawCommandElement {
    fn eq(&self, other: &Self) -> bool {
        if self.commands.len() != other.commands.len() {
            return false;
        }
        self.commands
            .iter()
            .zip(other.commands.iter())
            .all(|(a, b)| {
                draw_command_tag(a) == draw_command_tag(b)
                    && draw_command_closure_identity(a) == draw_command_closure_identity(b)
            })
    }
}

impl Eq for DrawCommandElement {}

impl std::hash::Hash for DrawCommandElement {
    fn hash<H: Hasher>(&self, state: &mut H) {
        "draw_commands".hash(state);
        self.commands.len().hash(state);
        for command in &self.commands {
            draw_command_tag(command).hash(state);
            (draw_command_closure_identity(command) as usize).hash(state);
        }
    }
}

impl ModifierNodeElement for DrawCommandElement {
    type Node = DrawCommandNode;

    fn create(&self) -> Self::Node {
        DrawCommandNode::new(self.commands.clone())
    }

    fn update(&self, node: &mut Self::Node) {
        node.commands.clone_from(&self.commands);
    }

    fn capabilities(&self) -> NodeCapabilities {
        NodeCapabilities::DRAW
    }
}

/// Node that offsets its content by a fixed (x, y) amount.
///
/// Matches Kotlin: `OffsetNode` in foundation-layout/src/commonMain/kotlin/androidx/compose/foundation/layout/Offset.kt
#[derive(Debug)]
pub struct OffsetNode {
    x: f32,
    y: f32,
    rtl_aware: bool,
    state: NodeState,
}

impl OffsetNode {
    pub fn new(x: f32, y: f32, rtl_aware: bool) -> Self {
        Self {
            x,
            y,
            rtl_aware,
            state: NodeState::new(),
        }
    }

    pub fn offset(&self) -> Point {
        Point {
            x: self.x,
            y: self.y,
        }
    }

    /// The offset on a whole device pixel of `density`, where Compose's
    /// `OffsetNode` places its content.
    pub fn device_offset(&self, density: f32) -> Point {
        use cranpose_ui_layout::round_to_px;
        Point {
            x: round_to_px(self.x, density),
            y: round_to_px(self.y, density),
        }
    }

    pub fn rtl_aware(&self) -> bool {
        self.rtl_aware
    }
}

impl DelegatableNode for OffsetNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl_layout_modifier_node!(OffsetNode, invalidate = InvalidationKind::Layout);

impl LayoutModifierNode for OffsetNode {
    fn measure(
        &self,
        context: &mut dyn ModifierNodeContext,
        measurable: &dyn Measurable,
        constraints: Constraints,
    ) -> cranpose_ui_layout::LayoutModifierMeasureResult {
        let offset = self.device_offset(context.density());
        measure_pass_through(measurable, constraints, |_| (offset.x, offset.y))
    }

    forward_intrinsics_to_child!();
}

/// Element that creates and updates offset nodes.
///
/// Matches Kotlin: `OffsetElement` in foundation-layout/src/commonMain/kotlin/androidx/compose/foundation/layout/Offset.kt
#[derive(Debug, Clone, PartialEq)]
pub struct OffsetElement {
    x: f32,
    y: f32,
    rtl_aware: bool,
}

impl OffsetElement {
    pub fn new(x: f32, y: f32, rtl_aware: bool) -> Self {
        Self { x, y, rtl_aware }
    }
}

impl Hash for OffsetElement {
    fn hash<H: Hasher>(&self, state: &mut H) {
        hash_f32_value(state, self.x);
        hash_f32_value(state, self.y);
        self.rtl_aware.hash(state);
    }
}

impl ModifierNodeElement for OffsetElement {
    type Node = OffsetNode;

    fn create(&self) -> Self::Node {
        OffsetNode::new(self.x, self.y, self.rtl_aware)
    }

    fn update(&self, node: &mut Self::Node) {
        if node.x != self.x || node.y != self.y || node.rtl_aware != self.rtl_aware {
            node.x = self.x;
            node.y = self.y;
            node.rtl_aware = self.rtl_aware;
        }
    }

    fn capabilities(&self) -> NodeCapabilities {
        NodeCapabilities::LAYOUT
    }

    fn update_invalidation_kind(&self) -> Option<InvalidationKind> {
        Some(InvalidationKind::Layout)
    }
}

/// Node that offsets its content by a fraction of its own measured size.
///
/// There is no direct Jetpack Compose modifier equivalent; Compose's slide
/// transitions receive the measured size through a lambda instead. This node
/// backs `slide_in_vertically` / `slide_out_vertically` in
/// `AnimatedVisibility`, where the offset is expressed as a fraction of the
/// content height.
#[derive(Debug)]
pub struct FractionalOffsetNode {
    x_fraction: f32,
    y_fraction: f32,
    state: NodeState,
}

impl FractionalOffsetNode {
    pub fn new(x_fraction: f32, y_fraction: f32) -> Self {
        Self {
            x_fraction,
            y_fraction,
            state: NodeState::new(),
        }
    }

    pub fn fractions(&self) -> Point {
        Point {
            x: self.x_fraction,
            y: self.y_fraction,
        }
    }
}

impl DelegatableNode for FractionalOffsetNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl_layout_modifier_node!(FractionalOffsetNode, invalidate = InvalidationKind::Layout);

impl LayoutModifierNode for FractionalOffsetNode {
    fn measure(
        &self,
        _context: &mut dyn ModifierNodeContext,
        measurable: &dyn Measurable,
        constraints: Constraints,
    ) -> cranpose_ui_layout::LayoutModifierMeasureResult {
        measure_pass_through(measurable, constraints, |size| {
            (self.x_fraction * size.width, self.y_fraction * size.height)
        })
    }

    forward_intrinsics_to_child!();
}

/// Element that creates and updates fractional offset nodes.
#[derive(Debug, Clone, PartialEq)]
pub struct FractionalOffsetElement {
    x_fraction: f32,
    y_fraction: f32,
}

impl FractionalOffsetElement {
    pub fn new(x_fraction: f32, y_fraction: f32) -> Self {
        Self {
            x_fraction,
            y_fraction,
        }
    }
}

impl Hash for FractionalOffsetElement {
    fn hash<H: Hasher>(&self, state: &mut H) {
        "fractional_offset".hash(state);
        hash_f32_value(state, self.x_fraction);
        hash_f32_value(state, self.y_fraction);
    }
}

impl ModifierNodeElement for FractionalOffsetElement {
    type Node = FractionalOffsetNode;

    fn create(&self) -> Self::Node {
        FractionalOffsetNode::new(self.x_fraction, self.y_fraction)
    }

    fn update(&self, node: &mut Self::Node) {
        if node.x_fraction != self.x_fraction || node.y_fraction != self.y_fraction {
            node.x_fraction = self.x_fraction;
            node.y_fraction = self.y_fraction;
        }
    }

    fn capabilities(&self) -> NodeCapabilities {
        NodeCapabilities::LAYOUT
    }

    fn update_invalidation_kind(&self) -> Option<InvalidationKind> {
        Some(InvalidationKind::Layout)
    }
}

/// Direction for fill modifiers (horizontal, vertical, or both).
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum FillDirection {
    Horizontal,
    Vertical,
    Both,
}

/// Node that fills the maximum available space in one or both dimensions.
///
/// Matches Kotlin: `FillNode` in foundation-layout/src/commonMain/kotlin/androidx/compose/foundation/layout/Size.kt
#[derive(Debug)]
pub struct FillNode {
    direction: FillDirection,
    fraction: f32,
    state: NodeState,
}

impl FillNode {
    pub fn new(direction: FillDirection, fraction: f32) -> Self {
        Self {
            direction,
            fraction,
            state: NodeState::new(),
        }
    }

    pub fn direction(&self) -> FillDirection {
        self.direction
    }

    pub fn fraction(&self) -> f32 {
        self.fraction
    }
}

impl DelegatableNode for FillNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl_layout_modifier_node!(FillNode, invalidate = InvalidationKind::Layout);

impl LayoutModifierNode for FillNode {
    fn measure(
        &self,
        context: &mut dyn ModifierNodeContext,
        measurable: &dyn Measurable,
        constraints: Constraints,
    ) -> cranpose_ui_layout::LayoutModifierMeasureResult {
        let density = context.density();
        let (fill_width, child_min_width, child_max_width) = if self.direction
            != FillDirection::Vertical
            && constraints.max_width != f32::INFINITY
        {
            let width =
                cranpose_ui_layout::round_to_px(constraints.max_width * self.fraction, density)
                    .clamp(constraints.min_width, constraints.max_width);
            (width, width, width)
        } else {
            (
                constraints.max_width,
                constraints.min_width,
                constraints.max_width,
            )
        };

        let (fill_height, child_min_height, child_max_height) = if self.direction
            != FillDirection::Horizontal
            && constraints.max_height != f32::INFINITY
        {
            let height =
                cranpose_ui_layout::round_to_px(constraints.max_height * self.fraction, density)
                    .clamp(constraints.min_height, constraints.max_height);
            (height, height, height)
        } else {
            (
                constraints.max_height,
                constraints.min_height,
                constraints.max_height,
            )
        };

        let fill_constraints = Constraints {
            min_width: child_min_width,
            max_width: child_max_width,
            min_height: child_min_height,
            max_height: child_max_height,
        };

        let placeable = measurable.measure(fill_constraints);

        let result_width = if self.direction != FillDirection::Vertical
            && constraints.max_width != f32::INFINITY
        {
            fill_width
        } else {
            placeable.width()
        };

        let result_height = if self.direction != FillDirection::Horizontal
            && constraints.max_height != f32::INFINITY
        {
            fill_height
        } else {
            placeable.height()
        };

        cranpose_ui_layout::LayoutModifierMeasureResult::with_size(Size {
            width: result_width,
            height: result_height,
        })
    }

    forward_intrinsics_to_child!();
}

/// Element that creates and updates fill nodes.
///
/// Matches Kotlin: `FillElement` in foundation-layout/src/commonMain/kotlin/androidx/compose/foundation/layout/Size.kt
#[derive(Debug, Clone, PartialEq)]
pub struct FillElement {
    direction: FillDirection,
    fraction: f32,
}

impl FillElement {
    pub fn width(fraction: f32) -> Self {
        Self {
            direction: FillDirection::Horizontal,
            fraction,
        }
    }

    pub fn height(fraction: f32) -> Self {
        Self {
            direction: FillDirection::Vertical,
            fraction,
        }
    }

    pub fn size(fraction: f32) -> Self {
        Self {
            direction: FillDirection::Both,
            fraction,
        }
    }
}

impl Hash for FillElement {
    fn hash<H: Hasher>(&self, state: &mut H) {
        self.direction.hash(state);
        hash_f32_value(state, self.fraction);
    }
}

impl ModifierNodeElement for FillElement {
    type Node = FillNode;

    fn create(&self) -> Self::Node {
        FillNode::new(self.direction, self.fraction)
    }

    fn update(&self, node: &mut Self::Node) {
        if node.direction != self.direction || node.fraction != self.fraction {
            node.direction = self.direction;
            node.fraction = self.fraction;
        }
    }

    fn capabilities(&self) -> NodeCapabilities {
        NodeCapabilities::LAYOUT
    }
}

#[derive(Debug)]
pub struct WeightNode {
    weight: f32,
    fill: bool,
    state: NodeState,
}

impl WeightNode {
    pub fn new(weight: f32, fill: bool) -> Self {
        Self {
            weight,
            fill,
            state: NodeState::new(),
        }
    }

    pub fn layout_weight(&self) -> LayoutWeight {
        LayoutWeight {
            weight: self.weight,
            fill: self.fill,
        }
    }
}

impl DelegatableNode for WeightNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl ModifierNode for WeightNode {
    fn on_attach(&mut self, context: &mut dyn ModifierNodeContext) {
        context.invalidate(cranpose_foundation::InvalidationKind::Layout);
    }
}

#[derive(Debug, Clone, PartialEq)]
pub struct WeightElement {
    weight: f32,
    fill: bool,
}

impl WeightElement {
    pub fn new(weight: f32, fill: bool) -> Self {
        Self { weight, fill }
    }
}

impl Hash for WeightElement {
    fn hash<H: Hasher>(&self, state: &mut H) {
        hash_f32_value(state, self.weight);
        self.fill.hash(state);
    }
}

impl ModifierNodeElement for WeightElement {
    type Node = WeightNode;

    fn create(&self) -> Self::Node {
        WeightNode::new(self.weight, self.fill)
    }

    fn update(&self, node: &mut Self::Node) {
        if node.weight != self.weight || node.fill != self.fill {
            node.weight = self.weight;
            node.fill = self.fill;
        }
    }

    fn capabilities(&self) -> NodeCapabilities {
        NodeCapabilities::LAYOUT
    }
}

#[derive(Debug)]
pub struct AlignmentNode {
    box_alignment: Option<Alignment>,
    column_alignment: Option<HorizontalAlignment>,
    row_alignment: Option<VerticalAlignment>,
    state: NodeState,
}

impl AlignmentNode {
    pub fn new(
        box_alignment: Option<Alignment>,
        column_alignment: Option<HorizontalAlignment>,
        row_alignment: Option<VerticalAlignment>,
    ) -> Self {
        Self {
            box_alignment,
            column_alignment,
            row_alignment,
            state: NodeState::new(),
        }
    }

    pub fn box_alignment(&self) -> Option<Alignment> {
        self.box_alignment
    }

    pub fn column_alignment(&self) -> Option<HorizontalAlignment> {
        self.column_alignment
    }

    pub fn row_alignment(&self) -> Option<VerticalAlignment> {
        self.row_alignment
    }
}

impl DelegatableNode for AlignmentNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl ModifierNode for AlignmentNode {
    fn on_attach(&mut self, context: &mut dyn ModifierNodeContext) {
        context.invalidate(cranpose_foundation::InvalidationKind::Layout);
    }
}

#[derive(Debug, Clone, PartialEq)]
pub struct AlignmentElement {
    box_alignment: Option<Alignment>,
    column_alignment: Option<HorizontalAlignment>,
    row_alignment: Option<VerticalAlignment>,
}

impl AlignmentElement {
    pub fn box_alignment(alignment: Alignment) -> Self {
        Self {
            box_alignment: Some(alignment),
            column_alignment: None,
            row_alignment: None,
        }
    }

    pub fn column_alignment(alignment: HorizontalAlignment) -> Self {
        Self {
            box_alignment: None,
            column_alignment: Some(alignment),
            row_alignment: None,
        }
    }

    pub fn row_alignment(alignment: VerticalAlignment) -> Self {
        Self {
            box_alignment: None,
            column_alignment: None,
            row_alignment: Some(alignment),
        }
    }
}

impl Hash for AlignmentElement {
    fn hash<H: Hasher>(&self, state: &mut H) {
        if let Some(alignment) = self.box_alignment {
            state.write_u8(1);
            hash_alignment(state, alignment);
        } else {
            state.write_u8(0);
        }
        if let Some(alignment) = self.column_alignment {
            state.write_u8(1);
            hash_horizontal_alignment(state, alignment);
        } else {
            state.write_u8(0);
        }
        if let Some(alignment) = self.row_alignment {
            state.write_u8(1);
            hash_vertical_alignment(state, alignment);
        } else {
            state.write_u8(0);
        }
    }
}

impl ModifierNodeElement for AlignmentElement {
    type Node = AlignmentNode;

    fn create(&self) -> Self::Node {
        AlignmentNode::new(
            self.box_alignment,
            self.column_alignment,
            self.row_alignment,
        )
    }

    fn update(&self, node: &mut Self::Node) {
        if node.box_alignment != self.box_alignment {
            node.box_alignment = self.box_alignment;
        }
        if node.column_alignment != self.column_alignment {
            node.column_alignment = self.column_alignment;
        }
        if node.row_alignment != self.row_alignment {
            node.row_alignment = self.row_alignment;
        }
    }

    fn capabilities(&self) -> NodeCapabilities {
        NodeCapabilities::LAYOUT
    }
}

#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
pub enum IntrinsicAxis {
    Width,
    Height,
}

#[derive(Debug)]
pub struct IntrinsicSizeNode {
    axis: IntrinsicAxis,
    size: IntrinsicSize,
    state: NodeState,
}

impl IntrinsicSizeNode {
    pub fn new(axis: IntrinsicAxis, size: IntrinsicSize) -> Self {
        Self {
            axis,
            size,
            state: NodeState::new(),
        }
    }

    pub fn axis(&self) -> IntrinsicAxis {
        self.axis
    }

    pub fn intrinsic_size(&self) -> IntrinsicSize {
        self.size
    }
}

impl DelegatableNode for IntrinsicSizeNode {
    fn node_state(&self) -> &NodeState {
        &self.state
    }
}

impl ModifierNode for IntrinsicSizeNode {
    fn on_attach(&mut self, context: &mut dyn ModifierNodeContext) {
        context.invalidate(cranpose_foundation::InvalidationKind::Layout);
    }
}

#[derive(Debug, Clone, PartialEq)]
pub struct IntrinsicSizeElement {
    axis: IntrinsicAxis,
    size: IntrinsicSize,
}

impl IntrinsicSizeElement {
    pub fn width(size: IntrinsicSize) -> Self {
        Self {
            axis: IntrinsicAxis::Width,
            size,
        }
    }

    pub fn height(size: IntrinsicSize) -> Self {
        Self {
            axis: IntrinsicAxis::Height,
            size,
        }
    }
}

impl Hash for IntrinsicSizeElement {
    fn hash<H: Hasher>(&self, state: &mut H) {
        state.write_u8(match self.axis {
            IntrinsicAxis::Width => 0,
            IntrinsicAxis::Height => 1,
        });
        state.write_u8(match self.size {
            IntrinsicSize::Min => 0,
            IntrinsicSize::Max => 1,
        });
    }
}

impl ModifierNodeElement for IntrinsicSizeElement {
    type Node = IntrinsicSizeNode;

    fn create(&self) -> Self::Node {
        IntrinsicSizeNode::new(self.axis, self.size)
    }

    fn update(&self, node: &mut Self::Node) {
        if node.axis != self.axis {
            node.axis = self.axis;
        }
        if node.size != self.size {
            node.size = self.size;
        }
    }

    fn capabilities(&self) -> NodeCapabilities {
        NodeCapabilities::LAYOUT
    }
}

#[cfg(test)]
#[path = "tests/modifier_nodes_tests.rs"]
mod tests;