context.rs

use floem_renderer::Renderer as FloemRenderer;
use peniko::kurbo::{Affine, Point, Rect, RoundedRect, Shape, Size, Vec2};
use std::{
    ops::{Deref, DerefMut},
    rc::Rc,
    sync::Arc,
    time::Instant,
};
use taffy::prelude::NodeId;

use crate::{
    action::{exec_after, show_context_menu},
    app_state::AppState,
    event::{Event, EventListener, EventPropagation},
    id::ViewId,
    inspector::CaptureState,
    menu::Menu,
    style::{Style, StyleProp, ZIndex},
    view::{paint_bg, paint_border, paint_outline, View},
    view_state::ChangeFlags,
};

pub type EventCallback = dyn Fn(&Event) -> EventPropagation;
pub type ResizeCallback = dyn Fn(Rect);
pub type MenuCallback = dyn Fn() -> Menu;

pub(crate) struct ResizeListener {
    pub(crate) rect: Rect,
    pub(crate) callback: Box<ResizeCallback>,
}

/// The listener when the view is got moved to a different position in the window
pub(crate) struct MoveListener {
    pub(crate) window_origin: Point,
    pub(crate) callback: Box<dyn Fn(Point)>,
}

pub struct DragState {
    pub(crate) id: ViewId,
    pub(crate) offset: Vec2,
    pub(crate) released_at: Option<std::time::Instant>,
}

pub(crate) enum FrameUpdate {
    Style(ViewId),
    Layout(ViewId),
    Paint(ViewId),
}

/// A bundle of helper methods to be used by `View::event` handlers
pub struct EventCx<'a> {
    pub(crate) app_state: &'a mut AppState,
}

impl<'a> EventCx<'a> {
    pub fn app_state_mut(&mut self) -> &mut AppState {
        self.app_state
    }

    pub fn app_state(&self) -> &AppState {
        self.app_state
    }

    pub fn update_active(&mut self, id: ViewId) {
        self.app_state.update_active(id);
    }

    pub fn is_active(&self, id: ViewId) -> bool {
        self.app_state.is_active(&id)
    }

    #[allow(unused)]
    pub(crate) fn update_focus(&mut self, id: ViewId, keyboard_navigation: bool) {
        self.app_state.update_focus(id, keyboard_navigation);
    }

    /// Internal method used by Floem. This can be called from parent `View`s to propagate an event to the child `View`.
    // pub fn view_event(
    //     &mut self,
    //     view: &mut dyn View,
    //     id_path: Option<&[Id]>,
    //     event: Event,
    // ) -> EventPropagation {
    //     if self.should_send(view.view_data().id(), &event) {
    //         self.unconditional_view_event(view, id_path, event)
    //     } else {
    //         EventPropagation::Continue
    //     }
    // }

    /// Internal method used by Floem. This can be called from parent `View`s to propagate an event to the child `View`.
    pub(crate) fn unconditional_view_event(
        &mut self,
        view_id: ViewId,
        event: Event,
        directed: bool,
    ) -> EventPropagation {
        if view_id.is_hidden() {
            // we don't process events for hidden view
            return EventPropagation::Continue;
        }
        if self.app_state.is_disabled(&view_id) && !event.allow_disabled() {
            // if the view is disabled and the event is not processed
            // for disabled views
            return EventPropagation::Continue;
        }

        // offset the event positions if the event has positions
        // e.g. pointer events, so that the position is relative
        // to the view, taking into account of the layout location
        // of the view and the viewport of the view if it's in a scroll.
        let event = self.offset_event(view_id, event);

        // if there's id_path, it's an event only for a view.
        // if let Some(id_path) = id_path {
        //     if id_path.is_empty() {
        //         // this happens when the parent is the destination,
        //         // but the parent just passed the event on,
        //         // so it's not really for this view and we stop
        //         // the event propagation.
        //         return EventPropagation::Continue;
        //     }

        //     let id = id_path[0];
        //     let id_path = &id_path[1..];

        //     if id != view.view_data().id() {
        //         // This shouldn't happen
        //         return EventPropagation::Continue;
        //     }

        //     // we're the parent of the event destination, so pass it on to the child
        //     if !id_path.is_empty() {
        //         if let Some(child) = view.child_mut(id_path[0]) {
        //             return self.unconditional_view_event(child, Some(id_path), event.clone());
        //         } else {
        //             // we don't have the child, stop the event propagation
        //             return EventPropagation::Continue;
        //         }
        //     }
        // }

        let view = view_id.view();
        if view
            .borrow_mut()
            .event_before_children(self, &event)
            .is_processed()
        {
            return EventPropagation::Stop;
        }

        if !directed {
            let children = view_id.children();
            for child in children.into_iter().rev() {
                if self.should_send(child, &event)
                    && self
                        .unconditional_view_event(child, event.clone(), false)
                        .is_processed()
                {
                    return EventPropagation::Stop;
                }
            }
        }

        // if the event was dispatched to an id_path, the event is supposed to be only
        // handled by this view only, so we pass an empty id_path
        // and the event propagation would be stopped at this view
        // if view
        //     .event(
        //         self,
        //         if id_path.is_some() { Some(&[]) } else { None },
        //         event.clone(),
        //     )
        //     .is_processed()
        // {
        //     return EventPropagation::Stop;
        // }

        if view
            .borrow_mut()
            .event_after_children(self, &event)
            .is_processed()
        {
            return EventPropagation::Stop;
        }

        let view_state = view_id.state();

        match &event {
            Event::PointerDown(event) => {
                self.app_state.clicking.insert(view_id);
                if event.button.is_primary() {
                    let rect = view_id.get_size().unwrap_or_default().to_rect();
                    let now_focused = rect.contains(event.pos);

                    if now_focused {
                        if self.app_state.keyboard_navigable.contains(&view_id) {
                            // if the view can be focused, we update the focus
                            self.app_state.update_focus(view_id, false);
                        }
                        if event.count == 2
                            && view_state
                                .borrow()
                                .event_listeners
                                .contains_key(&EventListener::DoubleClick)
                        {
                            view_state.borrow_mut().last_pointer_down = Some(event.clone());
                        }
                        if view_state
                            .borrow()
                            .event_listeners
                            .contains_key(&EventListener::Click)
                        {
                            view_state.borrow_mut().last_pointer_down = Some(event.clone());
                        }

                        let bottom_left = {
                            let layout = view_state.borrow().layout_rect;
                            Point::new(layout.x0, layout.y1)
                        };
                        let popout_menu = view_state.borrow().popout_menu.clone();
                        if let Some(menu) = popout_menu {
                            show_context_menu(menu(), Some(bottom_left));
                            return EventPropagation::Stop;
                        }
                        if self.app_state.draggable.contains(&view_id)
                            && self.app_state.drag_start.is_none()
                        {
                            self.app_state.drag_start = Some((view_id, event.pos));
                        }
                    }
                } else if event.button.is_secondary() {
                    let rect = view_id.get_size().unwrap_or_default().to_rect();
                    let now_focused = rect.contains(event.pos);

                    if now_focused {
                        if self.app_state.keyboard_navigable.contains(&view_id) {
                            // if the view can be focused, we update the focus
                            self.app_state.update_focus(view_id, false);
                        }
                        if view_state
                            .borrow()
                            .event_listeners
                            .contains_key(&EventListener::SecondaryClick)
                        {
                            view_state.borrow_mut().last_pointer_down = Some(event.clone());
                        }
                    }
                }
            }
            Event::PointerMove(pointer_event) => {
                let rect = view_id.get_size().unwrap_or_default().to_rect();
                if rect.contains(pointer_event.pos) {
                    if self.app_state.is_dragging() {
                        self.app_state.dragging_over.insert(view_id);
                        view_id.apply_event(&EventListener::DragOver, &event);
                    } else {
                        self.app_state.hovered.insert(view_id);
                        let view_state = view_state.borrow();
                        let style = view_state.combined_style.builtin();
                        if let Some(cursor) = style.cursor() {
                            if self.app_state.cursor.is_none() {
                                self.app_state.cursor = Some(cursor);
                            }
                        }
                    }
                }
                if self.app_state.draggable.contains(&view_id) {
                    if let Some((_, drag_start)) = self
                        .app_state
                        .drag_start
                        .as_ref()
                        .filter(|(drag_id, _)| drag_id == &view_id)
                    {
                        let vec2 = pointer_event.pos - *drag_start;

                        if let Some(dragging) = self
                            .app_state
                            .dragging
                            .as_mut()
                            .filter(|d| d.id == view_id && d.released_at.is_none())
                        {
                            // update the dragging offset if the view is dragging and not released
                            dragging.offset = vec2;
                            self.app_state.request_paint(view_id);
                        } else if vec2.x.abs() + vec2.y.abs() > 1.0 {
                            // start dragging when moved 1 px
                            self.app_state.active = None;
                            self.update_active(view_id);
                            self.app_state.dragging = Some(DragState {
                                id: view_id,
                                offset: vec2,
                                released_at: None,
                            });
                            self.app_state.request_paint(view_id);
                            view_id.apply_event(&EventListener::DragStart, &event);
                        }
                    }
                }
                if view_id
                    .apply_event(&EventListener::PointerMove, &event)
                    .is_some_and(|prop| prop.is_processed())
                {
                    return EventPropagation::Stop;
                }
            }
            Event::PointerUp(pointer_event) => {
                if pointer_event.button.is_primary() {
                    let rect = view_id.get_size().unwrap_or_default().to_rect();
                    let on_view = rect.contains(pointer_event.pos);

                    // if id_path.is_none() {
                    if !directed {
                        if on_view {
                            if let Some(dragging) = self.app_state.dragging.as_mut() {
                                let dragging_id = dragging.id;
                                if view_id
                                    .apply_event(&EventListener::Drop, &event)
                                    .is_some_and(|prop| prop.is_processed())
                                {
                                    // if the drop is processed, we set dragging to none so that the animation
                                    // for the dragged view back to its original position isn't played.
                                    self.app_state.dragging = None;
                                    self.app_state.request_paint(view_id);
                                    dragging_id.apply_event(&EventListener::DragEnd, &event);
                                }
                            }
                        }
                    } else if let Some(dragging) =
                        self.app_state.dragging.as_mut().filter(|d| d.id == view_id)
                    {
                        let dragging_id = dragging.id;
                        dragging.released_at = Some(std::time::Instant::now());
                        self.app_state.request_paint(view_id);
                        dragging_id.apply_event(&EventListener::DragEnd, &event);
                    }

                    let last_pointer_down = view_state.borrow_mut().last_pointer_down.take();

                    let event_listeners = view_state.borrow().event_listeners.clone();
                    if let Some(handlers) = event_listeners.get(&EventListener::DoubleClick) {
                        view_state.borrow_mut();
                        if on_view
                            && self.app_state.is_clicking(&view_id)
                            && last_pointer_down
                                .as_ref()
                                .map(|e| e.count == 2)
                                .unwrap_or(false)
                            && handlers.iter().fold(false, |handled, handler| {
                                handled | handler(&event).is_processed()
                            })
                        {
                            return EventPropagation::Stop;
                        }
                    }

                    if let Some(handlers) = event_listeners.get(&EventListener::Click) {
                        if on_view
                            && self.app_state.is_clicking(&view_id)
                            && last_pointer_down.is_some()
                            && handlers.iter().fold(false, |handled, handler| {
                                handled | handler(&event).is_processed()
                            })
                        {
                            return EventPropagation::Stop;
                        }
                    }

                    if view_id
                        .apply_event(&EventListener::PointerUp, &event)
                        .is_some_and(|prop| prop.is_processed())
                    {
                        return EventPropagation::Stop;
                    }
                } else if pointer_event.button.is_secondary() {
                    let rect = view_id.get_size().unwrap_or_default().to_rect();
                    let on_view = rect.contains(pointer_event.pos);

                    let last_pointer_down = view_state.borrow_mut().last_pointer_down.take();
                    let event_listeners = view_state.borrow().event_listeners.clone();
                    if let Some(handlers) = event_listeners.get(&EventListener::SecondaryClick) {
                        if on_view
                            && last_pointer_down.is_some()
                            && handlers.iter().fold(false, |handled, handler| {
                                handled | handler(&event).is_processed()
                            })
                        {
                            return EventPropagation::Stop;
                        }
                    }

                    let viewport_event_position = {
                        let layout = view_state.borrow().layout_rect;
                        Point::new(
                            layout.x0 + pointer_event.pos.x,
                            layout.y0 + pointer_event.pos.y,
                        )
                    };
                    let context_menu = view_state.borrow().context_menu.clone();
                    if let Some(menu) = context_menu {
                        show_context_menu(menu(), Some(viewport_event_position));
                        return EventPropagation::Stop;
                    }
                }
            }
            Event::KeyDown(_) => {
                if self.app_state.is_focused(&view_id) && event.is_keyboard_trigger() {
                    view_id.apply_event(&EventListener::Click, &event);
                }
            }
            Event::WindowResized(_) => {
                if view_state.borrow().has_style_selectors.has_responsive() {
                    view_id.request_style();
                }
            }
            _ => (),
        }

        if let Some(listener) = event.listener() {
            let event_listeners = view_state.borrow().event_listeners.clone();
            if let Some(handlers) = event_listeners.get(&listener).cloned() {
                let should_run = if let Some(pos) = event.point() {
                    let rect = view_id.get_size().unwrap_or_default().to_rect();
                    rect.contains(pos)
                } else {
                    true
                };
                if should_run
                    && handlers.iter().fold(false, |handled, handler| {
                        handled | handler(&event).is_processed()
                    })
                {
                    return EventPropagation::Stop;
                }
            }
        }

        EventPropagation::Continue
    }

    /// translate a window-positioned event to the local coordinate system of a view
    pub(crate) fn offset_event(&self, id: ViewId, event: Event) -> Event {
        let state = id.state();
        let viewport = state.borrow().viewport;

        if let Some(layout) = id.get_layout() {
            event.offset((
                layout.location.x as f64 - viewport.map(|rect| rect.x0).unwrap_or(0.0),
                layout.location.y as f64 - viewport.map(|rect| rect.y0).unwrap_or(0.0),
            ))
        } else {
            event
        }
    }

    /// Used to determine if you should send an event to another view. This is basically a check for pointer events to see if the pointer is inside a child view and to make sure the current view isn't hidden or disabled.
    /// Usually this is used if you want to propagate an event to a child view
    pub fn should_send(&mut self, id: ViewId, event: &Event) -> bool {
        if id.is_hidden() || (self.app_state.is_disabled(&id) && !event.allow_disabled()) {
            return false;
        }
        if let Some(point) = event.point() {
            let layout_rect = id.layout_rect();
            if let Some(layout) = id.get_layout() {
                if layout_rect
                    .with_origin(Point::new(
                        layout.location.x as f64,
                        layout.location.y as f64,
                    ))
                    .contains(point)
                {
                    return true;
                }
            }
            false
        } else {
            true
        }
    }
}

#[derive(Default)]
pub struct InteractionState {
    pub(crate) is_hovered: bool,
    pub(crate) is_selected: bool,
    pub(crate) is_disabled: bool,
    pub(crate) is_focused: bool,
    pub(crate) is_clicking: bool,
    pub(crate) using_keyboard_navigation: bool,
}

pub struct StyleCx<'a> {
    pub(crate) app_state: &'a mut AppState,
    pub(crate) current_view: ViewId,
    pub(crate) current: Rc<Style>,
    pub(crate) direct: Style,
    saved: Vec<Rc<Style>>,
    pub(crate) now: Instant,
    saved_disabled: Vec<bool>,
    saved_selected: Vec<bool>,
    disabled: bool,
    selected: bool,
}

impl<'a> StyleCx<'a> {
    pub(crate) fn new(app_state: &'a mut AppState, root: ViewId) -> Self {
        Self {
            app_state,
            current_view: root,
            current: Default::default(),
            direct: Default::default(),
            saved: Default::default(),
            now: Instant::now(),
            saved_disabled: Default::default(),
            saved_selected: Default::default(),
            disabled: false,
            selected: false,
        }
    }

    /// Marks the current context as selected.
    pub fn selected(&mut self) {
        self.selected = true;
    }

    fn get_interact_state(&self, id: &ViewId) -> InteractionState {
        InteractionState {
            is_selected: self.selected,
            is_hovered: self.app_state.is_hovered(id),
            is_disabled: self.app_state.is_disabled(id),
            is_focused: self.app_state.is_focused(id),
            is_clicking: self.app_state.is_clicking(id),
            using_keyboard_navigation: self.app_state.keyboard_navigation,
        }
    }

    /// Internal method used by Floem to compute the styles for the view.
    pub fn style_view(&mut self, view_id: ViewId) {
        self.save();
        let view = view_id.view();
        let view_state = view_id.state();
        {
            let mut view_state = view_state.borrow_mut();
            if !view_state.requested_changes.contains(ChangeFlags::STYLE) {
                return;
            }
            view_state.requested_changes.remove(ChangeFlags::STYLE);
        }

        let view_style = view.borrow().view_style();
        let view_class = view.borrow().view_class();
        {
            let mut view_state = view_state.borrow_mut();

            // Propagate style requests to children if needed.
            if view_state.request_style_recursive {
                view_state.request_style_recursive = false;
                let children = view_id.children();
                for child in children {
                    let view_state = child.state();
                    let mut state = view_state.borrow_mut();
                    state.request_style_recursive = true;
                    state.requested_changes.insert(ChangeFlags::STYLE);
                }
            }
        }

        let mut view_interact_state = self.get_interact_state(&view_id);
        view_interact_state.is_disabled |= self.disabled;
        self.disabled = view_interact_state.is_disabled;
        let mut new_frame = self.app_state.compute_style(
            view_id,
            view_style,
            view_interact_state,
            view_class,
            &self.current,
        );

        let style = view_state.borrow().combined_style.clone();
        self.direct = style;
        Style::apply_only_inherited(&mut self.current, &self.direct);
        CaptureState::capture_style(view_id, self);

        // If there's any changes to the Taffy style, request layout.
        let taffy_style = self.direct.to_taffy_style();
        if taffy_style != view_state.borrow().taffy_style {
            view_state.borrow_mut().taffy_style = taffy_style;
            view_id.request_layout();
        }

        // This is used by the `request_transition` and `style` methods below.
        self.current_view = view_id;

        {
            let mut view_state = view_state.borrow_mut();
            // Extract the relevant layout properties so the content rect can be calculated
            // when painting.
            view_state.layout_props.read_explicit(
                &self.direct,
                &self.current,
                &self.now,
                &mut new_frame,
            );

            view_state.view_style_props.read_explicit(
                &self.direct,
                &self.current,
                &self.now,
                &mut new_frame,
            );
            if new_frame {
                self.app_state.schedule_style(view_id);
            }
        }

        view.borrow_mut().style(self);

        self.restore();
    }

    pub fn now(&self) -> Instant {
        self.now
    }

    pub fn save(&mut self) {
        self.saved.push(self.current.clone());
        self.saved_disabled.push(self.disabled);
        self.saved_selected.push(self.selected);
    }

    pub fn restore(&mut self) {
        self.current = self.saved.pop().unwrap_or_default();
        self.disabled = self.saved_disabled.pop().unwrap_or_default();
        self.selected = self.saved_selected.pop().unwrap_or_default();
    }

    pub fn get_prop<P: StyleProp>(&self, _prop: P) -> Option<P::Type> {
        self.direct
            .get_prop::<P>()
            .or_else(|| self.current.get_prop::<P>())
    }

    pub fn style(&self) -> Style {
        (*self.current).clone().apply(self.direct.clone())
    }

    pub fn direct_style(&self) -> &Style {
        &self.direct
    }

    pub fn indirect_style(&self) -> &Style {
        &self.current
    }

    pub fn request_transition(&mut self) {
        let id = self.current_view;
        self.app_state_mut().schedule_style(id);
    }

    pub fn app_state_mut(&mut self) -> &mut AppState {
        self.app_state
    }

    pub fn app_state(&self) -> &AppState {
        self.app_state
    }
}

pub struct ComputeLayoutCx<'a> {
    pub(crate) app_state: &'a mut AppState,
    pub(crate) viewport: Rect,
    pub(crate) window_origin: Point,
    pub(crate) saved_viewports: Vec<Rect>,
    pub(crate) saved_window_origins: Vec<Point>,
}

impl<'a> ComputeLayoutCx<'a> {
    pub(crate) fn new(app_state: &'a mut AppState, viewport: Rect) -> Self {
        Self {
            app_state,
            viewport,
            window_origin: Point::ZERO,
            saved_viewports: Vec::new(),
            saved_window_origins: Vec::new(),
        }
    }

    pub fn app_state_mut(&mut self) -> &mut AppState {
        self.app_state
    }

    pub fn app_state(&self) -> &AppState {
        self.app_state
    }

    pub fn save(&mut self) {
        self.saved_viewports.push(self.viewport);
        self.saved_window_origins.push(self.window_origin);
    }

    pub fn restore(&mut self) {
        self.viewport = self.saved_viewports.pop().unwrap_or_default();
        self.window_origin = self.saved_window_origins.pop().unwrap_or_default();
    }

    pub fn current_viewport(&self) -> Rect {
        self.viewport
    }

    /// Internal method used by Floem. This method derives its calculations based on the [Taffy Node](taffy::tree::NodeId) returned by the `View::layout` method.
    ///
    /// It's responsible for:
    /// - calculating and setting the view's origin (local coordinates and window coordinates)
    /// - calculating and setting the view's viewport
    /// - invoking any attached context::ResizeListeners
    ///
    /// Returns the bounding rect that encompasses this view and its children
    pub fn compute_view_layout(&mut self, id: ViewId) -> Option<Rect> {
        let view_state = id.state();
        if id.is_hidden() {
            view_state.borrow_mut().layout_rect = Rect::ZERO;
            return None;
        }

        self.save();

        let layout = id.get_layout().unwrap_or_default();
        let origin = Point::new(layout.location.x as f64, layout.location.y as f64);
        let this_viewport = view_state.borrow().viewport;
        let this_viewport_origin = this_viewport.unwrap_or_default().origin().to_vec2();
        let size = Size::new(layout.size.width as f64, layout.size.height as f64);
        let parent_viewport = self.viewport.with_origin(
            Point::new(
                self.viewport.x0 - layout.location.x as f64,
                self.viewport.y0 - layout.location.y as f64,
            ) + this_viewport_origin,
        );
        self.viewport = parent_viewport.intersect(size.to_rect());
        if let Some(this_viewport) = this_viewport {
            self.viewport = self.viewport.intersect(this_viewport);
        }

        let window_origin = origin + self.window_origin.to_vec2() - this_viewport_origin;
        self.window_origin = window_origin;
        {
            view_state.borrow_mut().window_origin = window_origin;
        }

        let resize_listener = view_state.borrow().resize_listener.clone();
        if let Some(resize) = resize_listener.as_ref() {
            let mut resize = resize.borrow_mut();
            let new_rect = size.to_rect().with_origin(origin);
            if new_rect != resize.rect {
                resize.rect = new_rect;
                (*resize.callback)(new_rect);
            }
        }

        let move_listener = view_state.borrow().move_listener.clone();
        if let Some(listener) = move_listener {
            let mut listener = listener.borrow_mut();
            if window_origin != listener.window_origin {
                listener.window_origin = window_origin;
                (*listener.callback)(window_origin);
            }
        }

        let view = id.view();
        let child_layout_rect = view.borrow_mut().compute_layout(self);

        let layout_rect = size.to_rect().with_origin(self.window_origin);
        let layout_rect = if let Some(child_layout_rect) = child_layout_rect {
            layout_rect.union(child_layout_rect)
        } else {
            layout_rect
        };
        view_state.borrow_mut().layout_rect = layout_rect;

        self.restore();

        Some(layout_rect)
    }
}

/// Holds current layout state for given position in the tree.
/// You'll use this in the `View::layout` implementation to call `layout_node` on children and to access any font
pub struct LayoutCx<'a> {
    pub(crate) app_state: &'a mut AppState,
}

impl<'a> LayoutCx<'a> {
    pub(crate) fn new(app_state: &'a mut AppState) -> Self {
        Self { app_state }
    }

    pub fn app_state_mut(&mut self) -> &mut AppState {
        self.app_state
    }

    pub fn app_state(&self) -> &AppState {
        self.app_state
    }

    /// Responsible for invoking the recalculation of style and thus the layout and
    /// creating or updating the layout of child nodes within the closure.
    ///
    /// You should ensure that all children are laid out within the closure and/or whatever
    /// other work you need to do to ensure that the layout for the returned nodes is correct.
    pub fn layout_node(
        &mut self,
        id: ViewId,
        has_children: bool,
        mut children: impl FnMut(&mut LayoutCx) -> Vec<NodeId>,
    ) -> NodeId {
        let view_state = id.state();
        let node = view_state.borrow().node;
        if !view_state
            .borrow()
            .requested_changes
            .contains(ChangeFlags::LAYOUT)
        {
            return node;
        }
        view_state
            .borrow_mut()
            .requested_changes
            .remove(ChangeFlags::LAYOUT);
        let style = view_state.borrow().combined_style.to_taffy_style();
        let _ = id.taffy().borrow_mut().set_style(node, style);

        if has_children {
            let nodes = children(self);
            let _ = id.taffy().borrow_mut().set_children(node, &nodes);
        }

        node
    }

    /// Internal method used by Floem to invoke the user-defined `View::layout` method.
    pub fn layout_view(&mut self, view: &mut dyn View) -> NodeId {
        view.layout(self)
    }
}

pub struct PaintCx<'a> {
    pub(crate) app_state: &'a mut AppState,
    pub(crate) paint_state: &'a mut PaintState,
    pub(crate) transform: Affine,
    pub(crate) clip: Option<RoundedRect>,
    pub(crate) z_index: Option<i32>,
    pub(crate) saved_transforms: Vec<Affine>,
    pub(crate) saved_clips: Vec<Option<RoundedRect>>,
    pub(crate) saved_z_indexes: Vec<Option<i32>>,
}

impl<'a> PaintCx<'a> {
    pub fn save(&mut self) {
        self.saved_transforms.push(self.transform);
        self.saved_clips.push(self.clip);
        self.saved_z_indexes.push(self.z_index);
    }

    pub fn restore(&mut self) {
        self.transform = self.saved_transforms.pop().unwrap_or_default();
        self.clip = self.saved_clips.pop().unwrap_or_default();
        self.z_index = self.saved_z_indexes.pop().unwrap_or_default();
        self.paint_state.renderer.transform(self.transform);
        if let Some(z_index) = self.z_index {
            self.paint_state.renderer.set_z_index(z_index);
        } else {
            self.paint_state.renderer.set_z_index(0);
        }
        if let Some(rect) = self.clip {
            self.paint_state.renderer.clip(&rect);
        } else {
            self.paint_state.renderer.clear_clip();
        }
    }

    /// paint the children of this view
    pub fn paint_children(&mut self, id: ViewId) {
        let children = id.children();
        for child in children {
            self.paint_view(child);
        }
    }

    /// The entry point for painting a view. You shouldn't need to implement this yourself. Instead, implement [`Widget::paint`].
    /// It handles the internal work before and after painting [`Widget::paint`] implementations.
    /// It is responsible for
    /// - managing hidden status
    /// - clipping
    /// - painting computed styles like background color, border, font-styles, and z-index and handling painting requirements of drag and drop
    pub fn paint_view(&mut self, id: ViewId) {
        if id.is_hidden() {
            return;
        }
        let view = id.view();
        let view_state = id.state();

        self.save();
        let size = self.transform(id);
        let is_empty = self
            .clip
            .map(|rect| rect.rect().intersect(size.to_rect()).is_empty())
            .unwrap_or(false);
        if !is_empty {
            let style = view_state.borrow().combined_style.clone();
            let view_style_props = view_state.borrow().view_style_props.clone();

            if let Some(z_index) = style.get(ZIndex) {
                self.set_z_index(z_index);
            }

            paint_bg(self, &style, &view_style_props, size);

            view.borrow_mut().paint(self);
            paint_border(self, &view_style_props, size);
            paint_outline(self, &view_style_props, size)
        }

        let mut drag_set_to_none = false;
        if let Some(dragging) = self.app_state.dragging.as_ref() {
            if dragging.id == id {
                let dragging_offset = dragging.offset;
                let mut offset_scale = None;
                if let Some(released_at) = dragging.released_at {
                    const LIMIT: f64 = 300.0;
                    let elapsed = released_at.elapsed().as_millis() as f64;
                    if elapsed < LIMIT {
                        offset_scale = Some(1.0 - elapsed / LIMIT);
                        exec_after(std::time::Duration::from_millis(8), move |_| {
                            id.request_paint();
                        });
                    } else {
                        drag_set_to_none = true;
                    }
                } else {
                    offset_scale = Some(1.0);
                }

                if let Some(offset_scale) = offset_scale {
                    let offset = dragging_offset * offset_scale;
                    self.save();

                    let mut new = self.transform.as_coeffs();
                    new[4] += offset.x;
                    new[5] += offset.y;
                    self.transform = Affine::new(new);
                    self.paint_state.renderer.transform(self.transform);
                    self.set_z_index(1000);
                    self.clear_clip();

                    let style = view_state.borrow().combined_style.clone();
                    let mut view_style_props = view_state.borrow().view_style_props.clone();
                    let style =
                        if let Some(dragging_style) = view_state.borrow().dragging_style.clone() {
                            let style = style.apply(dragging_style);
                            let mut _new_frame = false;
                            view_style_props.read_explicit(
                                &style,
                                &style,
                                &Instant::now(),
                                &mut _new_frame,
                            );
                            style
                        } else {
                            style
                        };
                    paint_bg(self, &style, &view_style_props, size);

                    view.borrow_mut().paint(self);
                    paint_border(self, &view_style_props, size);
                    paint_outline(self, &view_style_props, size);

                    self.restore();
                }
            }
        }
        if drag_set_to_none {
            self.app_state.dragging = None;
        }

        self.restore();
    }

    /// Clip the drawing area to the given shape.
    pub fn clip(&mut self, shape: &impl Shape) {
        let rect = if let Some(rect) = shape.as_rect() {
            rect.to_rounded_rect(0.0)
        } else if let Some(rect) = shape.as_rounded_rect() {
            rect
        } else {
            let rect = shape.bounding_box();
            rect.to_rounded_rect(0.0)
        };

        let rect = if let Some(existing) = self.clip {
            let rect = existing.rect().intersect(rect.rect());
            self.paint_state.renderer.clip(&rect);
            rect.to_rounded_rect(0.0)
        } else {
            self.paint_state.renderer.clip(&shape);
            rect
        };
        self.clip = Some(rect);
    }

    /// Remove clipping so the entire window can be rendered to.
    pub fn clear_clip(&mut self) {
        self.clip = None;
        self.paint_state.renderer.clear_clip();
    }

    pub fn offset(&mut self, offset: (f64, f64)) {
        let mut new = self.transform.as_coeffs();
        new[4] += offset.0;
        new[5] += offset.1;
        self.transform = Affine::new(new);
        self.paint_state.renderer.transform(self.transform);
        if let Some(rect) = self.clip.as_mut() {
            let raidus = rect.radii();
            *rect = rect
                .rect()
                .with_origin(rect.origin() - Vec2::new(offset.0, offset.1))
                .to_rounded_rect(raidus);
        }
    }

    pub fn transform(&mut self, id: ViewId) -> Size {
        if let Some(layout) = id.get_layout() {
            let offset = layout.location;
            let mut new = self.transform.as_coeffs();
            new[4] += offset.x as f64;
            new[5] += offset.y as f64;
            self.transform = Affine::new(new);
            self.paint_state.renderer.transform(self.transform);

            if let Some(rect) = self.clip.as_mut() {
                let raidus = rect.radii();
                *rect = rect
                    .rect()
                    .with_origin(rect.origin() - Vec2::new(offset.x as f64, offset.y as f64))
                    .to_rounded_rect(raidus);
            }

            Size::new(layout.size.width as f64, layout.size.height as f64)
        } else {
            Size::ZERO
        }
    }

    pub(crate) fn set_z_index(&mut self, z_index: i32) {
        self.z_index = Some(z_index);
        self.paint_state.renderer.set_z_index(z_index);
    }

    pub fn is_focused(&self, id: ViewId) -> bool {
        self.app_state.is_focused(&id)
    }
}

// TODO: should this be private?
pub struct PaintState {
    pub(crate) renderer: crate::renderer::Renderer<Arc<dyn wgpu::WindowHandle>>,
}

impl PaintState {
    pub fn new(window: Arc<dyn wgpu::WindowHandle>, scale: f64, size: Size) -> Self {
        Self {
            renderer: crate::renderer::Renderer::new(Arc::new(window), scale, size),
        }
    }

    pub(crate) fn resize(&mut self, scale: f64, size: Size) {
        self.renderer.resize(scale, size);
    }

    pub(crate) fn set_scale(&mut self, scale: f64) {
        self.renderer.set_scale(scale);
    }
}

pub struct UpdateCx<'a> {
    pub(crate) app_state: &'a mut AppState,
}

impl<'a> UpdateCx<'a> {
    pub fn app_state_mut(&mut self) -> &mut AppState {
        self.app_state
    }

    pub fn app_state(&self) -> &AppState {
        self.app_state
    }
}

impl Deref for PaintCx<'_> {
    type Target = crate::renderer::Renderer<Arc<dyn wgpu::WindowHandle>>;

    fn deref(&self) -> &Self::Target {
        &self.paint_state.renderer
    }
}

impl DerefMut for PaintCx<'_> {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.paint_state.renderer
    }
}