ratatui-toolkit 0.2.6

//! TreeView::render_fn method.

use ratatui::text::Line;

use crate::primitives::tree_view::node_state::NodeState;
use crate::primitives::tree_view::widget::TreeView;

impl<'a, T> TreeView<'a, T> {
    /// Sets the render function for nodes.
    ///
    /// # Arguments
    ///
    /// * `f` - A function that takes node data and state, returns a Line.
    ///
    /// # Returns
    ///
    /// Self for method chaining.
    ///
    /// # Example
    ///
    /// ```rust
    /// use ratatui::text::Line;
    /// use ratatui_toolkit::tree_view::{TreeNode, TreeView};
    ///
    /// let nodes = vec![TreeNode::new("Item")];
    /// let tree = TreeView::new(nodes)
    ///     .render_fn(|data, state| {
    ///         if state.is_selected {
    ///             Line::from(format!("> {}", data))
    ///         } else {
    ///             Line::from(*data)
    ///         }
    ///     });
    /// ```
    pub fn render_fn<F>(mut self, f: F) -> Self
    where
        F: Fn(&T, &NodeState) -> Line<'a> + 'a,
    {
        self.render_fn = Box::new(f);
        self
    }
}