cranpose_ui/modifier/

slices.rs

use std::fmt;
use std::rc::Rc;

use cranpose_foundation::{ModifierNodeChain, NodeCapabilities, PointerEvent};
use cranpose_ui_graphics::GraphicsLayer;

use crate::draw::DrawCommand;
use crate::modifier::Modifier;
use crate::modifier_nodes::{
    BackgroundNode, ClipToBoundsNode, CornerShapeNode, DrawCommandNode, GraphicsLayerNode,
    PaddingNode,
};
use crate::text_field_modifier_node::TextFieldModifierNode;
use crate::text_modifier_node::TextModifierNode;
use cranpose_ui_graphics::EdgeInsets;
use std::cell::RefCell;

use super::{ModifierChainHandle, Point};

/// Snapshot of modifier node slices that impact draw and pointer subsystems.
#[derive(Default)]
pub struct ModifierNodeSlices {
    draw_commands: Vec<DrawCommand>,
    pointer_inputs: Vec<Rc<dyn Fn(PointerEvent)>>,
    click_handlers: Vec<Rc<dyn Fn(Point)>>,
    clip_to_bounds: bool,
    text_content: Option<Rc<str>>,
    graphics_layer: Option<GraphicsLayer>,
    chain_guard: Option<Rc<ChainGuard>>,
}

struct ChainGuard {
    _handle: ModifierChainHandle,
}

impl Clone for ModifierNodeSlices {
    fn clone(&self) -> Self {
        Self {
            draw_commands: self.draw_commands.clone(),
            pointer_inputs: self.pointer_inputs.clone(),
            click_handlers: self.click_handlers.clone(),
            clip_to_bounds: self.clip_to_bounds,
            text_content: self.text_content.clone(),
            graphics_layer: self.graphics_layer,
            chain_guard: self.chain_guard.clone(),
        }
    }
}

impl ModifierNodeSlices {
    pub fn draw_commands(&self) -> &[DrawCommand] {
        &self.draw_commands
    }

    pub fn pointer_inputs(&self) -> &[Rc<dyn Fn(PointerEvent)>] {
        &self.pointer_inputs
    }

    pub fn click_handlers(&self) -> &[Rc<dyn Fn(Point)>] {
        &self.click_handlers
    }

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

    pub fn text_content(&self) -> Option<&str> {
        self.text_content.as_deref()
    }

    pub fn text_content_rc(&self) -> Option<Rc<str>> {
        self.text_content.clone()
    }

    pub fn graphics_layer(&self) -> Option<GraphicsLayer> {
        self.graphics_layer
    }

    pub fn with_chain_guard(mut self, handle: ModifierChainHandle) -> Self {
        self.chain_guard = Some(Rc::new(ChainGuard { _handle: handle }));
        self
    }

    /// Resets the slice collection for reuse, retaining vector capacity.
    pub fn clear(&mut self) {
        self.draw_commands.clear();
        self.pointer_inputs.clear();
        self.click_handlers.clear();
        self.clip_to_bounds = false;
        self.text_content = None;
        self.graphics_layer = None;
        self.chain_guard = None;
    }
}

impl fmt::Debug for ModifierNodeSlices {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.debug_struct("ModifierNodeSlices")
            .field("draw_commands", &self.draw_commands.len())
            .field("pointer_inputs", &self.pointer_inputs.len())
            .field("click_handlers", &self.click_handlers.len())
            .field("clip_to_bounds", &self.clip_to_bounds)
            .field("text_content", &self.text_content)
            .field("graphics_layer", &self.graphics_layer)
            .finish()
    }
}

/// Collects modifier node slices directly from a reconciled [`ModifierNodeChain`].
pub fn collect_modifier_slices(chain: &ModifierNodeChain) -> ModifierNodeSlices {
    let mut slices = ModifierNodeSlices::default();
    collect_modifier_slices_into(chain, &mut slices);
    slices
}

/// Collects modifier node slices into an existing buffer to reuse allocations.
pub fn collect_modifier_slices_into(chain: &ModifierNodeChain, slices: &mut ModifierNodeSlices) {
    slices.clear();

    chain.for_each_node_with_capability(NodeCapabilities::POINTER_INPUT, |_ref, node| {
        let _any = node.as_any();

        // ClickableNode is now handled as a standard PointerInputNode
        // to support drag cancellation and proper click semantics (Up vs Down)

        // Collect general pointer input handlers (non-clickable)
        if let Some(handler) = node
            .as_pointer_input_node()
            .and_then(|n| n.pointer_input_handler())
        {
            slices.pointer_inputs.push(handler);
        }
    });

    // Track background and shape to combine them in draw commands
    let background_color = RefCell::new(None);
    let corner_shape = RefCell::new(None);

    chain.for_each_node_with_capability(NodeCapabilities::DRAW, |_ref, node| {
        let any = node.as_any();

        // Collect background color from BackgroundNode
        if let Some(bg_node) = any.downcast_ref::<BackgroundNode>() {
            *background_color.borrow_mut() = Some(bg_node.color());
            // Note: BackgroundNode can have an optional shape, but we primarily track
            // shape via CornerShapeNode for flexibility
            if bg_node.shape().is_some() {
                *corner_shape.borrow_mut() = bg_node.shape();
            }
        }

        // Collect corner shape from CornerShapeNode
        if let Some(shape_node) = any.downcast_ref::<CornerShapeNode>() {
            *corner_shape.borrow_mut() = Some(shape_node.shape());
        }

        // Collect draw commands from DrawCommandNode
        if let Some(commands) = any.downcast_ref::<DrawCommandNode>() {
            slices
                .draw_commands
                .extend(commands.commands().iter().cloned());
        }

        // Use create_draw_closure() for nodes with dynamic content (cursor blink, selection)
        // This defers evaluation to render time, enabling live updates.
        // Fallback to draw() for nodes with static content.
        if let Some(draw_node) = node.as_draw_node() {
            if let Some(closure) = draw_node.create_draw_closure() {
                // Deferred closure - evaluates at render time
                slices.draw_commands.push(DrawCommand::Overlay(closure));
            } else {
                // Static draw - evaluate now
                use cranpose_ui_graphics::{DrawScope as _, DrawScopeDefault};
                let mut scope = DrawScopeDefault::new(crate::modifier::Size {
                    width: 0.0,
                    height: 0.0,
                });
                draw_node.draw(&mut scope);
                let primitives = scope.into_primitives();
                if !primitives.is_empty() {
                    let draw_cmd = Rc::new(move |_size: crate::modifier::Size| primitives.clone());
                    slices.draw_commands.push(DrawCommand::Overlay(draw_cmd));
                }
            }
        }

        // Collect graphics layer from GraphicsLayerNode
        if let Some(layer_node) = any.downcast_ref::<GraphicsLayerNode>() {
            slices.graphics_layer = Some(layer_node.layer());
        }

        if any.is::<ClipToBoundsNode>() {
            slices.clip_to_bounds = true;
        }
    });

    // Collect padding from modifier chain for cursor positioning
    let mut padding = EdgeInsets::default();
    chain.for_each_node_with_capability(NodeCapabilities::LAYOUT, |_ref, node| {
        let any = node.as_any();
        if let Some(padding_node) = any.downcast_ref::<PaddingNode>() {
            let p = padding_node.padding();
            padding.left += p.left;
            padding.top += p.top;
            padding.right += p.right;
            padding.bottom += p.bottom;
        }
    });

    // Collect text content from TextModifierNode or TextFieldModifierNode (LAYOUT capability)
    chain.for_each_node_with_capability(NodeCapabilities::LAYOUT, |_ref, node| {
        let any = node.as_any();
        if let Some(text_node) = any.downcast_ref::<TextModifierNode>() {
            // Rightmost text modifier wins
            slices.text_content = Some(text_node.text_arc());
        }
        // Also check for TextFieldModifierNode (editable text fields)
        if let Some(text_field_node) = any.downcast_ref::<TextFieldModifierNode>() {
            let text = text_field_node.text();
            slices.text_content = Some(Rc::from(text));

            // Update content offsets for cursor positioning in collect_draw_primitives()
            text_field_node.set_content_offset(padding.left);
            text_field_node.set_content_y_offset(padding.top);

            // Cursor/selection rendering is now handled via DrawModifierNode::collect_draw_primitives()
            // in the DRAW capability loop above
        }
    });

    // Convert background + shape into a draw command
    if let Some(color) = background_color.into_inner() {
        let shape = corner_shape.into_inner();

        let draw_cmd = Rc::new(move |size: crate::modifier::Size| {
            use crate::modifier::{Brush, Rect};
            use cranpose_ui_graphics::DrawPrimitive;

            let brush = Brush::solid(color);
            let rect = Rect {
                x: 0.0,
                y: 0.0,
                width: size.width,
                height: size.height,
            };

            if let Some(shape) = shape {
                let radii = shape.resolve(size.width, size.height);
                vec![DrawPrimitive::RoundRect { rect, brush, radii }]
            } else {
                vec![DrawPrimitive::Rect { rect, brush }]
            }
        });

        slices
            .draw_commands
            .insert(0, DrawCommand::Behind(draw_cmd));
    }
}

/// Collects modifier node slices by instantiating a temporary node chain from a [`Modifier`].
pub fn collect_slices_from_modifier(modifier: &Modifier) -> ModifierNodeSlices {
    let mut handle = ModifierChainHandle::new();
    let _ = handle.update(modifier);
    collect_modifier_slices(handle.chain()).with_chain_guard(handle)
}