div.rs

   1use crate::{
   2    point, px, Action, AnyDrag, AnyElement, AnyTooltip, AnyView, AppContext, BorrowAppContext,
   3    BorrowWindow, Bounds, ClickEvent, DispatchPhase, Element, ElementId, FocusEvent, FocusHandle,
   4    IntoElement, KeyContext, KeyDownEvent, KeyUpEvent, LayoutId, MouseButton, MouseDownEvent,
   5    MouseMoveEvent, MouseUpEvent, ParentElement, Pixels, Point, Render, ScrollWheelEvent,
   6    SharedString, Size, StackingOrder, Style, StyleRefinement, Styled, Task, View, Visibility,
   7    WindowContext,
   8};
   9use collections::HashMap;
  10use refineable::Refineable;
  11use smallvec::SmallVec;
  12use std::{
  13    any::{Any, TypeId},
  14    cell::RefCell,
  15    cmp::Ordering,
  16    fmt::Debug,
  17    mem,
  18    rc::Rc,
  19    time::Duration,
  20};
  21use taffy::style::Overflow;
  22use util::ResultExt;
  23
  24const DRAG_THRESHOLD: f64 = 2.;
  25const TOOLTIP_DELAY: Duration = Duration::from_millis(500);
  26
  27pub struct GroupStyle {
  28    pub group: SharedString,
  29    pub style: Box<StyleRefinement>,
  30}
  31
  32pub trait InteractiveElement: Sized + Element {
  33    fn interactivity(&mut self) -> &mut Interactivity;
  34
  35    fn group(mut self, group: impl Into<SharedString>) -> Self {
  36        self.interactivity().group = Some(group.into());
  37        self
  38    }
  39
  40    fn id(mut self, id: impl Into<ElementId>) -> Stateful<Self> {
  41        self.interactivity().element_id = Some(id.into());
  42
  43        Stateful { element: self }
  44    }
  45
  46    fn track_focus(mut self, focus_handle: &FocusHandle) -> Focusable<Self> {
  47        self.interactivity().focusable = true;
  48        self.interactivity().tracked_focus_handle = Some(focus_handle.clone());
  49        Focusable { element: self }
  50    }
  51
  52    fn key_context<C, E>(mut self, key_context: C) -> Self
  53    where
  54        C: TryInto<KeyContext, Error = E>,
  55        E: Debug,
  56    {
  57        if let Some(key_context) = key_context.try_into().log_err() {
  58            self.interactivity().key_context = Some(key_context);
  59        }
  60        self
  61    }
  62
  63    fn hover(mut self, f: impl FnOnce(StyleRefinement) -> StyleRefinement) -> Self {
  64        self.interactivity().hover_style = Some(Box::new(f(StyleRefinement::default())));
  65        self
  66    }
  67
  68    fn group_hover(
  69        mut self,
  70        group_name: impl Into<SharedString>,
  71        f: impl FnOnce(StyleRefinement) -> StyleRefinement,
  72    ) -> Self {
  73        self.interactivity().group_hover_style = Some(GroupStyle {
  74            group: group_name.into(),
  75            style: Box::new(f(StyleRefinement::default())),
  76        });
  77        self
  78    }
  79
  80    fn on_mouse_down(
  81        mut self,
  82        button: MouseButton,
  83        listener: impl Fn(&MouseDownEvent, &mut WindowContext) + 'static,
  84    ) -> Self {
  85        self.interactivity().mouse_down_listeners.push(Box::new(
  86            move |event, bounds, phase, cx| {
  87                if phase == DispatchPhase::Bubble
  88                    && event.button == button
  89                    && bounds.visibly_contains(&event.position, cx)
  90                {
  91                    (listener)(event, cx)
  92                }
  93            },
  94        ));
  95        self
  96    }
  97
  98    fn on_any_mouse_down(
  99        mut self,
 100        listener: impl Fn(&MouseDownEvent, &mut WindowContext) + 'static,
 101    ) -> Self {
 102        self.interactivity().mouse_down_listeners.push(Box::new(
 103            move |event, bounds, phase, cx| {
 104                if phase == DispatchPhase::Bubble && bounds.visibly_contains(&event.position, cx) {
 105                    (listener)(event, cx)
 106                }
 107            },
 108        ));
 109        self
 110    }
 111
 112    fn on_mouse_up(
 113        mut self,
 114        button: MouseButton,
 115        listener: impl Fn(&MouseUpEvent, &mut WindowContext) + 'static,
 116    ) -> Self {
 117        self.interactivity()
 118            .mouse_up_listeners
 119            .push(Box::new(move |event, bounds, phase, cx| {
 120                if phase == DispatchPhase::Bubble
 121                    && event.button == button
 122                    && bounds.visibly_contains(&event.position, cx)
 123                {
 124                    (listener)(event, cx)
 125                }
 126            }));
 127        self
 128    }
 129
 130    fn on_any_mouse_up(
 131        mut self,
 132        listener: impl Fn(&MouseUpEvent, &mut WindowContext) + 'static,
 133    ) -> Self {
 134        self.interactivity()
 135            .mouse_up_listeners
 136            .push(Box::new(move |event, bounds, phase, cx| {
 137                if phase == DispatchPhase::Bubble && bounds.visibly_contains(&event.position, cx) {
 138                    (listener)(event, cx)
 139                }
 140            }));
 141        self
 142    }
 143
 144    fn on_mouse_down_out(
 145        mut self,
 146        listener: impl Fn(&MouseDownEvent, &mut WindowContext) + 'static,
 147    ) -> Self {
 148        self.interactivity().mouse_down_listeners.push(Box::new(
 149            move |event, bounds, phase, cx| {
 150                if phase == DispatchPhase::Capture && !bounds.visibly_contains(&event.position, cx)
 151                {
 152                    (listener)(event, cx)
 153                }
 154            },
 155        ));
 156        self
 157    }
 158
 159    fn on_mouse_up_out(
 160        mut self,
 161        button: MouseButton,
 162        listener: impl Fn(&MouseUpEvent, &mut WindowContext) + 'static,
 163    ) -> Self {
 164        self.interactivity()
 165            .mouse_up_listeners
 166            .push(Box::new(move |event, bounds, phase, cx| {
 167                if phase == DispatchPhase::Capture
 168                    && event.button == button
 169                    && !bounds.visibly_contains(&event.position, cx)
 170                {
 171                    (listener)(event, cx);
 172                }
 173            }));
 174        self
 175    }
 176
 177    fn on_mouse_move(
 178        mut self,
 179        listener: impl Fn(&MouseMoveEvent, &mut WindowContext) + 'static,
 180    ) -> Self {
 181        self.interactivity().mouse_move_listeners.push(Box::new(
 182            move |event, bounds, phase, cx| {
 183                if phase == DispatchPhase::Bubble && bounds.visibly_contains(&event.position, cx) {
 184                    (listener)(event, cx);
 185                }
 186            },
 187        ));
 188        self
 189    }
 190
 191    fn on_scroll_wheel(
 192        mut self,
 193        listener: impl Fn(&ScrollWheelEvent, &mut WindowContext) + 'static,
 194    ) -> Self {
 195        self.interactivity().scroll_wheel_listeners.push(Box::new(
 196            move |event, bounds, phase, cx| {
 197                if phase == DispatchPhase::Bubble && bounds.visibly_contains(&event.position, cx) {
 198                    (listener)(event, cx);
 199                }
 200            },
 201        ));
 202        self
 203    }
 204
 205    /// Capture the given action, before normal action dispatch can fire
 206    fn capture_action<A: Action>(
 207        mut self,
 208        listener: impl Fn(&A, &mut WindowContext) + 'static,
 209    ) -> Self {
 210        self.interactivity().action_listeners.push((
 211            TypeId::of::<A>(),
 212            Box::new(move |action, phase, cx| {
 213                let action = action.downcast_ref().unwrap();
 214                if phase == DispatchPhase::Capture {
 215                    (listener)(action, cx)
 216                }
 217            }),
 218        ));
 219        self
 220    }
 221
 222    /// Add a listener for the given action, fires during the bubble event phase
 223    fn on_action<A: Action>(mut self, listener: impl Fn(&A, &mut WindowContext) + 'static) -> Self {
 224        self.interactivity().action_listeners.push((
 225            TypeId::of::<A>(),
 226            Box::new(move |action, phase, cx| {
 227                let action = action.downcast_ref().unwrap();
 228                if phase == DispatchPhase::Bubble {
 229                    (listener)(action, cx)
 230                }
 231            }),
 232        ));
 233        self
 234    }
 235
 236    fn on_boxed_action(
 237        mut self,
 238        action: &Box<dyn Action>,
 239        listener: impl Fn(&Box<dyn Action>, &mut WindowContext) + 'static,
 240    ) -> Self {
 241        let action = action.boxed_clone();
 242        self.interactivity().action_listeners.push((
 243            (*action).type_id(),
 244            Box::new(move |_, phase, cx| {
 245                if phase == DispatchPhase::Bubble {
 246                    (listener)(&action, cx)
 247                }
 248            }),
 249        ));
 250        self
 251    }
 252
 253    fn on_key_down(
 254        mut self,
 255        listener: impl Fn(&KeyDownEvent, &mut WindowContext) + 'static,
 256    ) -> Self {
 257        self.interactivity()
 258            .key_down_listeners
 259            .push(Box::new(move |event, phase, cx| {
 260                if phase == DispatchPhase::Bubble {
 261                    (listener)(event, cx)
 262                }
 263            }));
 264        self
 265    }
 266
 267    fn capture_key_down(
 268        mut self,
 269        listener: impl Fn(&KeyDownEvent, &mut WindowContext) + 'static,
 270    ) -> Self {
 271        self.interactivity()
 272            .key_down_listeners
 273            .push(Box::new(move |event, phase, cx| {
 274                if phase == DispatchPhase::Capture {
 275                    listener(event, cx)
 276                }
 277            }));
 278        self
 279    }
 280
 281    fn on_key_up(mut self, listener: impl Fn(&KeyUpEvent, &mut WindowContext) + 'static) -> Self {
 282        self.interactivity()
 283            .key_up_listeners
 284            .push(Box::new(move |event, phase, cx| {
 285                if phase == DispatchPhase::Bubble {
 286                    listener(event, cx)
 287                }
 288            }));
 289        self
 290    }
 291
 292    fn capture_key_up(
 293        mut self,
 294        listener: impl Fn(&KeyUpEvent, &mut WindowContext) + 'static,
 295    ) -> Self {
 296        self.interactivity()
 297            .key_up_listeners
 298            .push(Box::new(move |event, phase, cx| {
 299                if phase == DispatchPhase::Capture {
 300                    listener(event, cx)
 301                }
 302            }));
 303        self
 304    }
 305
 306    fn drag_over<S: 'static>(mut self, f: impl FnOnce(StyleRefinement) -> StyleRefinement) -> Self {
 307        self.interactivity()
 308            .drag_over_styles
 309            .push((TypeId::of::<S>(), f(StyleRefinement::default())));
 310        self
 311    }
 312
 313    fn group_drag_over<S: 'static>(
 314        mut self,
 315        group_name: impl Into<SharedString>,
 316        f: impl FnOnce(StyleRefinement) -> StyleRefinement,
 317    ) -> Self {
 318        self.interactivity().group_drag_over_styles.push((
 319            TypeId::of::<S>(),
 320            GroupStyle {
 321                group: group_name.into(),
 322                style: Box::new(f(StyleRefinement::default())),
 323            },
 324        ));
 325        self
 326    }
 327
 328    fn on_drop<W: 'static>(
 329        mut self,
 330        listener: impl Fn(&View<W>, &mut WindowContext) + 'static,
 331    ) -> Self {
 332        self.interactivity().drop_listeners.push((
 333            TypeId::of::<W>(),
 334            Box::new(move |dragged_view, cx| {
 335                listener(&dragged_view.downcast().unwrap(), cx);
 336            }),
 337        ));
 338        self
 339    }
 340}
 341
 342pub trait StatefulInteractiveElement: InteractiveElement {
 343    fn focusable(mut self) -> Focusable<Self> {
 344        self.interactivity().focusable = true;
 345        Focusable { element: self }
 346    }
 347
 348    fn overflow_scroll(mut self) -> Self {
 349        self.interactivity().base_style.overflow.x = Some(Overflow::Scroll);
 350        self.interactivity().base_style.overflow.y = Some(Overflow::Scroll);
 351        self
 352    }
 353
 354    fn overflow_x_scroll(mut self) -> Self {
 355        self.interactivity().base_style.overflow.x = Some(Overflow::Scroll);
 356        self
 357    }
 358
 359    fn overflow_y_scroll(mut self) -> Self {
 360        self.interactivity().base_style.overflow.y = Some(Overflow::Scroll);
 361        self
 362    }
 363
 364    fn track_scroll(mut self, scroll_handle: &ScrollHandle) -> Self {
 365        self.interactivity().scroll_handle = Some(scroll_handle.clone());
 366        self
 367    }
 368
 369    fn active(mut self, f: impl FnOnce(StyleRefinement) -> StyleRefinement) -> Self
 370    where
 371        Self: Sized,
 372    {
 373        self.interactivity().active_style = Some(Box::new(f(StyleRefinement::default())));
 374        self
 375    }
 376
 377    fn group_active(
 378        mut self,
 379        group_name: impl Into<SharedString>,
 380        f: impl FnOnce(StyleRefinement) -> StyleRefinement,
 381    ) -> Self
 382    where
 383        Self: Sized,
 384    {
 385        self.interactivity().group_active_style = Some(GroupStyle {
 386            group: group_name.into(),
 387            style: Box::new(f(StyleRefinement::default())),
 388        });
 389        self
 390    }
 391
 392    fn on_click(mut self, listener: impl Fn(&ClickEvent, &mut WindowContext) + 'static) -> Self
 393    where
 394        Self: Sized,
 395    {
 396        self.interactivity()
 397            .click_listeners
 398            .push(Box::new(move |event, cx| listener(event, cx)));
 399        self
 400    }
 401
 402    fn on_drag<W>(mut self, listener: impl Fn(&mut WindowContext) -> View<W> + 'static) -> Self
 403    where
 404        Self: Sized,
 405        W: 'static + Render,
 406    {
 407        debug_assert!(
 408            self.interactivity().drag_listener.is_none(),
 409            "calling on_drag more than once on the same element is not supported"
 410        );
 411        self.interactivity().drag_listener = Some(Box::new(move |cursor_offset, cx| AnyDrag {
 412            view: listener(cx).into(),
 413            cursor_offset,
 414        }));
 415        self
 416    }
 417
 418    fn on_hover(mut self, listener: impl Fn(&bool, &mut WindowContext) + 'static) -> Self
 419    where
 420        Self: Sized,
 421    {
 422        debug_assert!(
 423            self.interactivity().hover_listener.is_none(),
 424            "calling on_hover more than once on the same element is not supported"
 425        );
 426        self.interactivity().hover_listener = Some(Box::new(listener));
 427        self
 428    }
 429
 430    fn tooltip(mut self, build_tooltip: impl Fn(&mut WindowContext) -> AnyView + 'static) -> Self
 431    where
 432        Self: Sized,
 433    {
 434        debug_assert!(
 435            self.interactivity().tooltip_builder.is_none(),
 436            "calling tooltip more than once on the same element is not supported"
 437        );
 438        self.interactivity().tooltip_builder = Some(Rc::new(build_tooltip));
 439
 440        self
 441    }
 442}
 443
 444pub trait FocusableElement: InteractiveElement {
 445    fn focus(mut self, f: impl FnOnce(StyleRefinement) -> StyleRefinement) -> Self
 446    where
 447        Self: Sized,
 448    {
 449        self.interactivity().focus_style = Some(Box::new(f(StyleRefinement::default())));
 450        self
 451    }
 452
 453    fn in_focus(mut self, f: impl FnOnce(StyleRefinement) -> StyleRefinement) -> Self
 454    where
 455        Self: Sized,
 456    {
 457        self.interactivity().in_focus_style = Some(Box::new(f(StyleRefinement::default())));
 458        self
 459    }
 460}
 461
 462pub type FocusListeners = Vec<FocusListener>;
 463
 464pub type FocusListener = Box<dyn Fn(&FocusHandle, &FocusEvent, &mut WindowContext) + 'static>;
 465
 466pub type MouseDownListener =
 467    Box<dyn Fn(&MouseDownEvent, &InteractiveBounds, DispatchPhase, &mut WindowContext) + 'static>;
 468pub type MouseUpListener =
 469    Box<dyn Fn(&MouseUpEvent, &InteractiveBounds, DispatchPhase, &mut WindowContext) + 'static>;
 470
 471pub type MouseMoveListener =
 472    Box<dyn Fn(&MouseMoveEvent, &InteractiveBounds, DispatchPhase, &mut WindowContext) + 'static>;
 473
 474pub type ScrollWheelListener =
 475    Box<dyn Fn(&ScrollWheelEvent, &InteractiveBounds, DispatchPhase, &mut WindowContext) + 'static>;
 476
 477pub type ClickListener = Box<dyn Fn(&ClickEvent, &mut WindowContext) + 'static>;
 478
 479pub type DragListener = Box<dyn Fn(Point<Pixels>, &mut WindowContext) -> AnyDrag + 'static>;
 480
 481type DropListener = dyn Fn(AnyView, &mut WindowContext) + 'static;
 482
 483pub type TooltipBuilder = Rc<dyn Fn(&mut WindowContext) -> AnyView + 'static>;
 484
 485pub type KeyDownListener = Box<dyn Fn(&KeyDownEvent, DispatchPhase, &mut WindowContext) + 'static>;
 486
 487pub type KeyUpListener = Box<dyn Fn(&KeyUpEvent, DispatchPhase, &mut WindowContext) + 'static>;
 488
 489pub type DragEventListener = Box<dyn Fn(&MouseMoveEvent, &mut WindowContext) + 'static>;
 490
 491pub type ActionListener = Box<dyn Fn(&dyn Any, DispatchPhase, &mut WindowContext) + 'static>;
 492
 493pub fn div() -> Div {
 494    Div {
 495        interactivity: Interactivity::default(),
 496        children: SmallVec::default(),
 497    }
 498}
 499
 500pub struct Div {
 501    interactivity: Interactivity,
 502    children: SmallVec<[AnyElement; 2]>,
 503}
 504
 505impl Styled for Div {
 506    fn style(&mut self) -> &mut StyleRefinement {
 507        &mut self.interactivity.base_style
 508    }
 509}
 510
 511impl InteractiveElement for Div {
 512    fn interactivity(&mut self) -> &mut Interactivity {
 513        &mut self.interactivity
 514    }
 515}
 516
 517impl ParentElement for Div {
 518    fn children_mut(&mut self) -> &mut SmallVec<[AnyElement; 2]> {
 519        &mut self.children
 520    }
 521}
 522
 523impl Element for Div {
 524    type State = DivState;
 525
 526    fn layout(
 527        &mut self,
 528        element_state: Option<Self::State>,
 529        cx: &mut WindowContext,
 530    ) -> (LayoutId, Self::State) {
 531        let mut child_layout_ids = SmallVec::new();
 532        let (layout_id, interactive_state) = self.interactivity.layout(
 533            element_state.map(|s| s.interactive_state),
 534            cx,
 535            |style, cx| {
 536                cx.with_text_style(style.text_style().cloned(), |cx| {
 537                    child_layout_ids = self
 538                        .children
 539                        .iter_mut()
 540                        .map(|child| child.layout(cx))
 541                        .collect::<SmallVec<_>>();
 542                    cx.request_layout(&style, child_layout_ids.iter().copied())
 543                })
 544            },
 545        );
 546        (
 547            layout_id,
 548            DivState {
 549                interactive_state,
 550                child_layout_ids,
 551            },
 552        )
 553    }
 554
 555    fn paint(
 556        self,
 557        bounds: Bounds<Pixels>,
 558        element_state: &mut Self::State,
 559        cx: &mut WindowContext,
 560    ) {
 561        let mut child_min = point(Pixels::MAX, Pixels::MAX);
 562        let mut child_max = Point::default();
 563        let content_size = if element_state.child_layout_ids.is_empty() {
 564            bounds.size
 565        } else if let Some(scroll_handle) = self.interactivity.scroll_handle.as_ref() {
 566            let mut state = scroll_handle.0.borrow_mut();
 567            state.child_bounds = Vec::with_capacity(element_state.child_layout_ids.len());
 568            state.bounds = bounds;
 569            let requested = state.requested_scroll_top.take();
 570
 571            for (ix, child_layout_id) in element_state.child_layout_ids.iter().enumerate() {
 572                let child_bounds = cx.layout_bounds(*child_layout_id);
 573                child_min = child_min.min(&child_bounds.origin);
 574                child_max = child_max.max(&child_bounds.lower_right());
 575                state.child_bounds.push(child_bounds);
 576
 577                if let Some(requested) = requested.as_ref() {
 578                    if requested.0 == ix {
 579                        *state.offset.borrow_mut() =
 580                            bounds.origin - (child_bounds.origin - point(px(0.), requested.1));
 581                    }
 582                }
 583            }
 584            (child_max - child_min).into()
 585        } else {
 586            for child_layout_id in &element_state.child_layout_ids {
 587                let child_bounds = cx.layout_bounds(*child_layout_id);
 588                child_min = child_min.min(&child_bounds.origin);
 589                child_max = child_max.max(&child_bounds.lower_right());
 590            }
 591            (child_max - child_min).into()
 592        };
 593
 594        self.interactivity.paint(
 595            bounds,
 596            content_size,
 597            &mut element_state.interactive_state,
 598            cx,
 599            |style, scroll_offset, cx| {
 600                if style.visibility == Visibility::Hidden {
 601                    return;
 602                }
 603
 604                let z_index = style.z_index.unwrap_or(0);
 605
 606                cx.with_z_index(z_index, |cx| {
 607                    cx.with_z_index(0, |cx| {
 608                        style.paint(bounds, cx);
 609                    });
 610                    cx.with_z_index(1, |cx| {
 611                        cx.with_text_style(style.text_style().cloned(), |cx| {
 612                            cx.with_content_mask(style.overflow_mask(bounds), |cx| {
 613                                cx.with_element_offset(scroll_offset, |cx| {
 614                                    for child in self.children {
 615                                        child.paint(cx);
 616                                    }
 617                                })
 618                            })
 619                        })
 620                    })
 621                })
 622            },
 623        );
 624    }
 625}
 626
 627impl IntoElement for Div {
 628    type Element = Self;
 629
 630    fn element_id(&self) -> Option<ElementId> {
 631        self.interactivity.element_id.clone()
 632    }
 633
 634    fn into_element(self) -> Self::Element {
 635        self
 636    }
 637}
 638
 639pub struct DivState {
 640    child_layout_ids: SmallVec<[LayoutId; 2]>,
 641    interactive_state: InteractiveElementState,
 642}
 643
 644impl DivState {
 645    pub fn is_active(&self) -> bool {
 646        self.interactive_state.pending_mouse_down.borrow().is_some()
 647    }
 648}
 649
 650pub struct Interactivity {
 651    pub element_id: Option<ElementId>,
 652    pub key_context: Option<KeyContext>,
 653    pub focusable: bool,
 654    pub tracked_focus_handle: Option<FocusHandle>,
 655    pub scroll_handle: Option<ScrollHandle>,
 656    pub group: Option<SharedString>,
 657    pub base_style: Box<StyleRefinement>,
 658    pub focus_style: Option<Box<StyleRefinement>>,
 659    pub in_focus_style: Option<Box<StyleRefinement>>,
 660    pub hover_style: Option<Box<StyleRefinement>>,
 661    pub group_hover_style: Option<GroupStyle>,
 662    pub active_style: Option<Box<StyleRefinement>>,
 663    pub group_active_style: Option<GroupStyle>,
 664    pub drag_over_styles: Vec<(TypeId, StyleRefinement)>,
 665    pub group_drag_over_styles: Vec<(TypeId, GroupStyle)>,
 666    pub mouse_down_listeners: Vec<MouseDownListener>,
 667    pub mouse_up_listeners: Vec<MouseUpListener>,
 668    pub mouse_move_listeners: Vec<MouseMoveListener>,
 669    pub scroll_wheel_listeners: Vec<ScrollWheelListener>,
 670    pub key_down_listeners: Vec<KeyDownListener>,
 671    pub key_up_listeners: Vec<KeyUpListener>,
 672    pub action_listeners: Vec<(TypeId, ActionListener)>,
 673    pub drop_listeners: Vec<(TypeId, Box<DropListener>)>,
 674    pub click_listeners: Vec<ClickListener>,
 675    pub drag_listener: Option<DragListener>,
 676    pub hover_listener: Option<Box<dyn Fn(&bool, &mut WindowContext)>>,
 677    pub tooltip_builder: Option<TooltipBuilder>,
 678}
 679
 680#[derive(Clone, Debug)]
 681pub struct InteractiveBounds {
 682    pub bounds: Bounds<Pixels>,
 683    pub stacking_order: StackingOrder,
 684}
 685
 686impl InteractiveBounds {
 687    pub fn visibly_contains(&self, point: &Point<Pixels>, cx: &WindowContext) -> bool {
 688        self.bounds.contains(point) && cx.was_top_layer(&point, &self.stacking_order)
 689    }
 690
 691    pub fn drag_target_contains(&self, point: &Point<Pixels>, cx: &WindowContext) -> bool {
 692        self.bounds.contains(point)
 693            && cx.was_top_layer_under_active_drag(&point, &self.stacking_order)
 694    }
 695}
 696
 697impl Interactivity {
 698    pub fn layout(
 699        &mut self,
 700        element_state: Option<InteractiveElementState>,
 701        cx: &mut WindowContext,
 702        f: impl FnOnce(Style, &mut WindowContext) -> LayoutId,
 703    ) -> (LayoutId, InteractiveElementState) {
 704        let mut element_state = element_state.unwrap_or_default();
 705
 706        // Ensure we store a focus handle in our element state if we're focusable.
 707        // If there's an explicit focus handle we're tracking, use that. Otherwise
 708        // create a new handle and store it in the element state, which lives for as
 709        // as frames contain an element with this id.
 710        if self.focusable {
 711            element_state.focus_handle.get_or_insert_with(|| {
 712                self.tracked_focus_handle
 713                    .clone()
 714                    .unwrap_or_else(|| cx.focus_handle())
 715            });
 716        }
 717
 718        if let Some(scroll_handle) = self.scroll_handle.as_ref() {
 719            element_state.scroll_offset = Some(scroll_handle.0.borrow().offset.clone());
 720        }
 721
 722        let style = self.compute_style(None, &mut element_state, cx);
 723        let layout_id = f(style, cx);
 724        (layout_id, element_state)
 725    }
 726
 727    pub fn paint(
 728        mut self,
 729        bounds: Bounds<Pixels>,
 730        content_size: Size<Pixels>,
 731        element_state: &mut InteractiveElementState,
 732        cx: &mut WindowContext,
 733        f: impl FnOnce(Style, Point<Pixels>, &mut WindowContext),
 734    ) {
 735        let style = self.compute_style(Some(bounds), element_state, cx);
 736
 737        if style
 738            .background
 739            .as_ref()
 740            .is_some_and(|fill| fill.color().is_some_and(|color| !color.is_transparent()))
 741        {
 742            cx.with_z_index(style.z_index.unwrap_or(0), |cx| cx.add_opaque_layer(bounds))
 743        }
 744
 745        let interactive_bounds = Rc::new(InteractiveBounds {
 746            bounds: bounds.intersect(&cx.content_mask().bounds),
 747            stacking_order: cx.stacking_order().clone(),
 748        });
 749
 750        if let Some(mouse_cursor) = style.mouse_cursor {
 751            let mouse_position = &cx.mouse_position();
 752            let hovered = interactive_bounds.visibly_contains(mouse_position, cx);
 753            if hovered {
 754                cx.set_cursor_style(mouse_cursor);
 755            }
 756        }
 757
 758        for listener in self.mouse_down_listeners.drain(..) {
 759            let interactive_bounds = interactive_bounds.clone();
 760            cx.on_mouse_event(move |event: &MouseDownEvent, phase, cx| {
 761                listener(event, &*interactive_bounds, phase, cx);
 762            })
 763        }
 764
 765        for listener in self.mouse_up_listeners.drain(..) {
 766            let interactive_bounds = interactive_bounds.clone();
 767            cx.on_mouse_event(move |event: &MouseUpEvent, phase, cx| {
 768                listener(event, &*interactive_bounds, phase, cx);
 769            })
 770        }
 771
 772        for listener in self.mouse_move_listeners.drain(..) {
 773            let interactive_bounds = interactive_bounds.clone();
 774            cx.on_mouse_event(move |event: &MouseMoveEvent, phase, cx| {
 775                listener(event, &*interactive_bounds, phase, cx);
 776            })
 777        }
 778
 779        for listener in self.scroll_wheel_listeners.drain(..) {
 780            let interactive_bounds = interactive_bounds.clone();
 781            cx.on_mouse_event(move |event: &ScrollWheelEvent, phase, cx| {
 782                listener(event, &*interactive_bounds, phase, cx);
 783            })
 784        }
 785
 786        let hover_group_bounds = self
 787            .group_hover_style
 788            .as_ref()
 789            .and_then(|group_hover| GroupBounds::get(&group_hover.group, cx));
 790
 791        if let Some(group_bounds) = hover_group_bounds {
 792            let hovered = group_bounds.contains(&cx.mouse_position());
 793            cx.on_mouse_event(move |event: &MouseMoveEvent, phase, cx| {
 794                if phase == DispatchPhase::Capture {
 795                    if group_bounds.contains(&event.position) != hovered {
 796                        cx.notify();
 797                    }
 798                }
 799            });
 800        }
 801
 802        if self.hover_style.is_some()
 803            || self.base_style.mouse_cursor.is_some()
 804            || cx.active_drag.is_some() && !self.drag_over_styles.is_empty()
 805        {
 806            let bounds = bounds.intersect(&cx.content_mask().bounds);
 807            let hovered = bounds.contains(&cx.mouse_position());
 808            cx.on_mouse_event(move |event: &MouseMoveEvent, phase, cx| {
 809                if phase == DispatchPhase::Capture {
 810                    if bounds.contains(&event.position) != hovered {
 811                        cx.notify();
 812                    }
 813                }
 814            });
 815        }
 816
 817        if cx.active_drag.is_some() {
 818            let drop_listeners = mem::take(&mut self.drop_listeners);
 819            let interactive_bounds = interactive_bounds.clone();
 820            if !drop_listeners.is_empty() {
 821                cx.on_mouse_event(move |event: &MouseUpEvent, phase, cx| {
 822                    if phase == DispatchPhase::Bubble
 823                        && interactive_bounds.drag_target_contains(&event.position, cx)
 824                    {
 825                        if let Some(drag_state_type) =
 826                            cx.active_drag.as_ref().map(|drag| drag.view.entity_type())
 827                        {
 828                            for (drop_state_type, listener) in &drop_listeners {
 829                                if *drop_state_type == drag_state_type {
 830                                    let drag = cx
 831                                        .active_drag
 832                                        .take()
 833                                        .expect("checked for type drag state type above");
 834
 835                                    listener(drag.view.clone(), cx);
 836                                    cx.notify();
 837                                    cx.stop_propagation();
 838                                }
 839                            }
 840                        } else {
 841                            cx.active_drag = None;
 842                        }
 843                    }
 844                });
 845            }
 846        }
 847
 848        let click_listeners = mem::take(&mut self.click_listeners);
 849        let drag_listener = mem::take(&mut self.drag_listener);
 850
 851        if !click_listeners.is_empty() || drag_listener.is_some() {
 852            let pending_mouse_down = element_state.pending_mouse_down.clone();
 853            let mouse_down = pending_mouse_down.borrow().clone();
 854            if let Some(mouse_down) = mouse_down {
 855                if let Some(drag_listener) = drag_listener {
 856                    let active_state = element_state.clicked_state.clone();
 857                    let interactive_bounds = interactive_bounds.clone();
 858
 859                    cx.on_mouse_event(move |event: &MouseMoveEvent, phase, cx| {
 860                        if cx.active_drag.is_some() {
 861                            if phase == DispatchPhase::Capture {
 862                                cx.notify();
 863                            }
 864                        } else if phase == DispatchPhase::Bubble
 865                            && interactive_bounds.visibly_contains(&event.position, cx)
 866                            && (event.position - mouse_down.position).magnitude() > DRAG_THRESHOLD
 867                        {
 868                            *active_state.borrow_mut() = ElementClickedState::default();
 869                            let cursor_offset = event.position - bounds.origin;
 870                            let drag = drag_listener(cursor_offset, cx);
 871                            cx.active_drag = Some(drag);
 872                            cx.notify();
 873                            cx.stop_propagation();
 874                        }
 875                    });
 876                }
 877
 878                let interactive_bounds = interactive_bounds.clone();
 879                cx.on_mouse_event(move |event: &MouseUpEvent, phase, cx| {
 880                    if phase == DispatchPhase::Bubble
 881                        && interactive_bounds.visibly_contains(&event.position, cx)
 882                    {
 883                        let mouse_click = ClickEvent {
 884                            down: mouse_down.clone(),
 885                            up: event.clone(),
 886                        };
 887                        for listener in &click_listeners {
 888                            listener(&mouse_click, cx);
 889                        }
 890                    }
 891                    *pending_mouse_down.borrow_mut() = None;
 892                    cx.notify();
 893                });
 894            } else {
 895                let interactive_bounds = interactive_bounds.clone();
 896                cx.on_mouse_event(move |event: &MouseDownEvent, phase, cx| {
 897                    if phase == DispatchPhase::Bubble
 898                        && event.button == MouseButton::Left
 899                        && interactive_bounds.visibly_contains(&event.position, cx)
 900                    {
 901                        *pending_mouse_down.borrow_mut() = Some(event.clone());
 902                        cx.notify();
 903                    }
 904                });
 905            }
 906        }
 907
 908        if let Some(hover_listener) = self.hover_listener.take() {
 909            let was_hovered = element_state.hover_state.clone();
 910            let has_mouse_down = element_state.pending_mouse_down.clone();
 911            let interactive_bounds = interactive_bounds.clone();
 912
 913            cx.on_mouse_event(move |event: &MouseMoveEvent, phase, cx| {
 914                if phase != DispatchPhase::Bubble {
 915                    return;
 916                }
 917                let is_hovered = interactive_bounds.visibly_contains(&event.position, cx)
 918                    && has_mouse_down.borrow().is_none();
 919                let mut was_hovered = was_hovered.borrow_mut();
 920
 921                if is_hovered != was_hovered.clone() {
 922                    *was_hovered = is_hovered;
 923                    drop(was_hovered);
 924
 925                    hover_listener(&is_hovered, cx);
 926                }
 927            });
 928        }
 929
 930        if let Some(tooltip_builder) = self.tooltip_builder.take() {
 931            let active_tooltip = element_state.active_tooltip.clone();
 932            let pending_mouse_down = element_state.pending_mouse_down.clone();
 933            let interactive_bounds = interactive_bounds.clone();
 934
 935            cx.on_mouse_event(move |event: &MouseMoveEvent, phase, cx| {
 936                let is_hovered = interactive_bounds.visibly_contains(&event.position, cx)
 937                    && pending_mouse_down.borrow().is_none();
 938                if !is_hovered {
 939                    active_tooltip.borrow_mut().take();
 940                    return;
 941                }
 942
 943                if phase != DispatchPhase::Bubble {
 944                    return;
 945                }
 946
 947                if active_tooltip.borrow().is_none() {
 948                    let task = cx.spawn({
 949                        let active_tooltip = active_tooltip.clone();
 950                        let tooltip_builder = tooltip_builder.clone();
 951
 952                        move |mut cx| async move {
 953                            cx.background_executor().timer(TOOLTIP_DELAY).await;
 954                            cx.update(|_, cx| {
 955                                active_tooltip.borrow_mut().replace(ActiveTooltip {
 956                                    tooltip: Some(AnyTooltip {
 957                                        view: tooltip_builder(cx),
 958                                        cursor_offset: cx.mouse_position(),
 959                                    }),
 960                                    _task: None,
 961                                });
 962                                cx.notify();
 963                            })
 964                            .ok();
 965                        }
 966                    });
 967                    active_tooltip.borrow_mut().replace(ActiveTooltip {
 968                        tooltip: None,
 969                        _task: Some(task),
 970                    });
 971                }
 972            });
 973
 974            let active_tooltip = element_state.active_tooltip.clone();
 975            cx.on_mouse_event(move |_: &MouseDownEvent, _, _| {
 976                active_tooltip.borrow_mut().take();
 977            });
 978
 979            if let Some(active_tooltip) = element_state.active_tooltip.borrow().as_ref() {
 980                if active_tooltip.tooltip.is_some() {
 981                    cx.active_tooltip = active_tooltip.tooltip.clone()
 982                }
 983            }
 984        }
 985
 986        let active_state = element_state.clicked_state.clone();
 987        if active_state.borrow().is_clicked() {
 988            cx.on_mouse_event(move |_: &MouseUpEvent, phase, cx| {
 989                if phase == DispatchPhase::Capture {
 990                    *active_state.borrow_mut() = ElementClickedState::default();
 991                    cx.notify();
 992                }
 993            });
 994        } else {
 995            let active_group_bounds = self
 996                .group_active_style
 997                .as_ref()
 998                .and_then(|group_active| GroupBounds::get(&group_active.group, cx));
 999            let interactive_bounds = interactive_bounds.clone();
1000            cx.on_mouse_event(move |down: &MouseDownEvent, phase, cx| {
1001                if phase == DispatchPhase::Bubble {
1002                    let group =
1003                        active_group_bounds.map_or(false, |bounds| bounds.contains(&down.position));
1004                    let element = interactive_bounds.visibly_contains(&down.position, cx);
1005                    if group || element {
1006                        *active_state.borrow_mut() = ElementClickedState { group, element };
1007                        cx.notify();
1008                    }
1009                }
1010            });
1011        }
1012
1013        let overflow = style.overflow;
1014        if overflow.x == Overflow::Scroll || overflow.y == Overflow::Scroll {
1015            let scroll_offset = element_state
1016                .scroll_offset
1017                .get_or_insert_with(Rc::default)
1018                .clone();
1019            let line_height = cx.line_height();
1020            let scroll_max = (content_size - bounds.size).max(&Size::default());
1021            let interactive_bounds = interactive_bounds.clone();
1022
1023            cx.on_mouse_event(move |event: &ScrollWheelEvent, phase, cx| {
1024                if phase == DispatchPhase::Bubble
1025                    && interactive_bounds.visibly_contains(&event.position, cx)
1026                {
1027                    let mut scroll_offset = scroll_offset.borrow_mut();
1028                    let old_scroll_offset = *scroll_offset;
1029                    let delta = event.delta.pixel_delta(line_height);
1030
1031                    if overflow.x == Overflow::Scroll {
1032                        scroll_offset.x =
1033                            (scroll_offset.x + delta.x).clamp(-scroll_max.width, px(0.));
1034                    }
1035
1036                    if overflow.y == Overflow::Scroll {
1037                        scroll_offset.y =
1038                            (scroll_offset.y + delta.y).clamp(-scroll_max.height, px(0.));
1039                    }
1040
1041                    if *scroll_offset != old_scroll_offset {
1042                        cx.notify();
1043                        cx.stop_propagation();
1044                    }
1045                }
1046            });
1047        }
1048
1049        if let Some(group) = self.group.clone() {
1050            GroupBounds::push(group, bounds, cx);
1051        }
1052
1053        let scroll_offset = element_state
1054            .scroll_offset
1055            .as_ref()
1056            .map(|scroll_offset| *scroll_offset.borrow());
1057
1058        cx.with_key_dispatch(
1059            self.key_context.clone(),
1060            element_state.focus_handle.clone(),
1061            |_, cx| {
1062                for listener in self.key_down_listeners.drain(..) {
1063                    cx.on_key_event(move |event: &KeyDownEvent, phase, cx| {
1064                        listener(event, phase, cx);
1065                    })
1066                }
1067
1068                for listener in self.key_up_listeners.drain(..) {
1069                    cx.on_key_event(move |event: &KeyUpEvent, phase, cx| {
1070                        listener(event, phase, cx);
1071                    })
1072                }
1073
1074                for (action_type, listener) in self.action_listeners {
1075                    cx.on_action(action_type, listener)
1076                }
1077
1078                f(style, scroll_offset.unwrap_or_default(), cx)
1079            },
1080        );
1081
1082        if let Some(group) = self.group.as_ref() {
1083            GroupBounds::pop(group, cx);
1084        }
1085    }
1086
1087    pub fn compute_style(
1088        &self,
1089        bounds: Option<Bounds<Pixels>>,
1090        element_state: &mut InteractiveElementState,
1091        cx: &mut WindowContext,
1092    ) -> Style {
1093        let mut style = Style::default();
1094        style.refine(&self.base_style);
1095
1096        if let Some(focus_handle) = self.tracked_focus_handle.as_ref() {
1097            if let Some(in_focus_style) = self.in_focus_style.as_ref() {
1098                if focus_handle.within_focused(cx) {
1099                    style.refine(in_focus_style);
1100                }
1101            }
1102
1103            if let Some(focus_style) = self.focus_style.as_ref() {
1104                if focus_handle.is_focused(cx) {
1105                    style.refine(focus_style);
1106                }
1107            }
1108        }
1109
1110        if let Some(bounds) = bounds {
1111            let mouse_position = cx.mouse_position();
1112            if let Some(group_hover) = self.group_hover_style.as_ref() {
1113                if let Some(group_bounds) = GroupBounds::get(&group_hover.group, cx) {
1114                    if group_bounds.contains(&mouse_position)
1115                        && cx.was_top_layer(&mouse_position, cx.stacking_order())
1116                    {
1117                        style.refine(&group_hover.style);
1118                    }
1119                }
1120            }
1121            if let Some(hover_style) = self.hover_style.as_ref() {
1122                if bounds
1123                    .intersect(&cx.content_mask().bounds)
1124                    .contains(&mouse_position)
1125                    && cx.was_top_layer(&mouse_position, cx.stacking_order())
1126                {
1127                    style.refine(hover_style);
1128                }
1129            }
1130
1131            if let Some(drag) = cx.active_drag.take() {
1132                for (state_type, group_drag_style) in &self.group_drag_over_styles {
1133                    if let Some(group_bounds) = GroupBounds::get(&group_drag_style.group, cx) {
1134                        if *state_type == drag.view.entity_type()
1135                            && group_bounds.contains(&mouse_position)
1136                        {
1137                            style.refine(&group_drag_style.style);
1138                        }
1139                    }
1140                }
1141
1142                for (state_type, drag_over_style) in &self.drag_over_styles {
1143                    if *state_type == drag.view.entity_type()
1144                        && bounds
1145                            .intersect(&cx.content_mask().bounds)
1146                            .contains(&mouse_position)
1147                    {
1148                        style.refine(drag_over_style);
1149                    }
1150                }
1151
1152                cx.active_drag = Some(drag);
1153            }
1154        }
1155
1156        let clicked_state = element_state.clicked_state.borrow();
1157        if clicked_state.group {
1158            if let Some(group) = self.group_active_style.as_ref() {
1159                style.refine(&group.style)
1160            }
1161        }
1162
1163        if let Some(active_style) = self.active_style.as_ref() {
1164            if clicked_state.element {
1165                style.refine(active_style)
1166            }
1167        }
1168
1169        style
1170    }
1171}
1172
1173impl Default for Interactivity {
1174    fn default() -> Self {
1175        Self {
1176            element_id: None,
1177            key_context: None,
1178            focusable: false,
1179            tracked_focus_handle: None,
1180            scroll_handle: None,
1181            // scroll_offset: Point::default(),
1182            group: None,
1183            base_style: Box::new(StyleRefinement::default()),
1184            focus_style: None,
1185            in_focus_style: None,
1186            hover_style: None,
1187            group_hover_style: None,
1188            active_style: None,
1189            group_active_style: None,
1190            drag_over_styles: Vec::new(),
1191            group_drag_over_styles: Vec::new(),
1192            mouse_down_listeners: Vec::new(),
1193            mouse_up_listeners: Vec::new(),
1194            mouse_move_listeners: Vec::new(),
1195            scroll_wheel_listeners: Vec::new(),
1196            key_down_listeners: Vec::new(),
1197            key_up_listeners: Vec::new(),
1198            action_listeners: Vec::new(),
1199            drop_listeners: Vec::new(),
1200            click_listeners: Vec::new(),
1201            drag_listener: None,
1202            hover_listener: None,
1203            tooltip_builder: None,
1204        }
1205    }
1206}
1207
1208#[derive(Default)]
1209pub struct InteractiveElementState {
1210    pub focus_handle: Option<FocusHandle>,
1211    pub clicked_state: Rc<RefCell<ElementClickedState>>,
1212    pub hover_state: Rc<RefCell<bool>>,
1213    pub pending_mouse_down: Rc<RefCell<Option<MouseDownEvent>>>,
1214    pub scroll_offset: Option<Rc<RefCell<Point<Pixels>>>>,
1215    pub active_tooltip: Rc<RefCell<Option<ActiveTooltip>>>,
1216}
1217
1218pub struct ActiveTooltip {
1219    tooltip: Option<AnyTooltip>,
1220    _task: Option<Task<()>>,
1221}
1222
1223/// Whether or not the element or a group that contains it is clicked by the mouse.
1224#[derive(Copy, Clone, Default, Eq, PartialEq)]
1225pub struct ElementClickedState {
1226    pub group: bool,
1227    pub element: bool,
1228}
1229
1230impl ElementClickedState {
1231    fn is_clicked(&self) -> bool {
1232        self.group || self.element
1233    }
1234}
1235
1236#[derive(Default)]
1237pub struct GroupBounds(HashMap<SharedString, SmallVec<[Bounds<Pixels>; 1]>>);
1238
1239impl GroupBounds {
1240    pub fn get(name: &SharedString, cx: &mut AppContext) -> Option<Bounds<Pixels>> {
1241        cx.default_global::<Self>()
1242            .0
1243            .get(name)
1244            .and_then(|bounds_stack| bounds_stack.last())
1245            .cloned()
1246    }
1247
1248    pub fn push(name: SharedString, bounds: Bounds<Pixels>, cx: &mut AppContext) {
1249        cx.default_global::<Self>()
1250            .0
1251            .entry(name)
1252            .or_default()
1253            .push(bounds);
1254    }
1255
1256    pub fn pop(name: &SharedString, cx: &mut AppContext) {
1257        cx.default_global::<Self>().0.get_mut(name).unwrap().pop();
1258    }
1259}
1260
1261pub struct Focusable<E> {
1262    pub element: E,
1263}
1264
1265impl<E: InteractiveElement> FocusableElement for Focusable<E> {}
1266
1267impl<E> InteractiveElement for Focusable<E>
1268where
1269    E: InteractiveElement,
1270{
1271    fn interactivity(&mut self) -> &mut Interactivity {
1272        self.element.interactivity()
1273    }
1274}
1275
1276impl<E: StatefulInteractiveElement> StatefulInteractiveElement for Focusable<E> {}
1277
1278impl<E> Styled for Focusable<E>
1279where
1280    E: Styled,
1281{
1282    fn style(&mut self) -> &mut StyleRefinement {
1283        self.element.style()
1284    }
1285}
1286
1287impl<E> Element for Focusable<E>
1288where
1289    E: Element,
1290{
1291    type State = E::State;
1292
1293    fn layout(
1294        &mut self,
1295        state: Option<Self::State>,
1296        cx: &mut WindowContext,
1297    ) -> (LayoutId, Self::State) {
1298        self.element.layout(state, cx)
1299    }
1300
1301    fn paint(self, bounds: Bounds<Pixels>, state: &mut Self::State, cx: &mut WindowContext) {
1302        self.element.paint(bounds, state, cx)
1303    }
1304}
1305
1306impl<E> IntoElement for Focusable<E>
1307where
1308    E: Element,
1309{
1310    type Element = E;
1311
1312    fn element_id(&self) -> Option<ElementId> {
1313        self.element.element_id()
1314    }
1315
1316    fn into_element(self) -> Self::Element {
1317        self.element
1318    }
1319}
1320
1321impl<E> ParentElement for Focusable<E>
1322where
1323    E: ParentElement,
1324{
1325    fn children_mut(&mut self) -> &mut SmallVec<[AnyElement; 2]> {
1326        self.element.children_mut()
1327    }
1328}
1329
1330pub struct Stateful<E> {
1331    element: E,
1332}
1333
1334impl<E> Styled for Stateful<E>
1335where
1336    E: Styled,
1337{
1338    fn style(&mut self) -> &mut StyleRefinement {
1339        self.element.style()
1340    }
1341}
1342
1343impl<E> StatefulInteractiveElement for Stateful<E>
1344where
1345    E: Element,
1346    Self: InteractiveElement,
1347{
1348}
1349
1350impl<E> InteractiveElement for Stateful<E>
1351where
1352    E: InteractiveElement,
1353{
1354    fn interactivity(&mut self) -> &mut Interactivity {
1355        self.element.interactivity()
1356    }
1357}
1358
1359impl<E: FocusableElement> FocusableElement for Stateful<E> {}
1360
1361impl<E> Element for Stateful<E>
1362where
1363    E: Element,
1364{
1365    type State = E::State;
1366
1367    fn layout(
1368        &mut self,
1369        state: Option<Self::State>,
1370        cx: &mut WindowContext,
1371    ) -> (LayoutId, Self::State) {
1372        self.element.layout(state, cx)
1373    }
1374
1375    fn paint(self, bounds: Bounds<Pixels>, state: &mut Self::State, cx: &mut WindowContext) {
1376        self.element.paint(bounds, state, cx)
1377    }
1378}
1379
1380impl<E> IntoElement for Stateful<E>
1381where
1382    E: Element,
1383{
1384    type Element = Self;
1385
1386    fn element_id(&self) -> Option<ElementId> {
1387        self.element.element_id()
1388    }
1389
1390    fn into_element(self) -> Self::Element {
1391        self
1392    }
1393}
1394
1395impl<E> ParentElement for Stateful<E>
1396where
1397    E: ParentElement,
1398{
1399    fn children_mut(&mut self) -> &mut SmallVec<[AnyElement; 2]> {
1400        self.element.children_mut()
1401    }
1402}
1403
1404#[derive(Default)]
1405struct ScrollHandleState {
1406    // not great to have the nested rc's...
1407    offset: Rc<RefCell<Point<Pixels>>>,
1408    bounds: Bounds<Pixels>,
1409    child_bounds: Vec<Bounds<Pixels>>,
1410    requested_scroll_top: Option<(usize, Pixels)>,
1411}
1412
1413#[derive(Clone)]
1414pub struct ScrollHandle(Rc<RefCell<ScrollHandleState>>);
1415
1416impl ScrollHandle {
1417    pub fn new() -> Self {
1418        Self(Rc::default())
1419    }
1420
1421    pub fn offset(&self) -> Point<Pixels> {
1422        self.0.borrow().offset.borrow().clone()
1423    }
1424
1425    pub fn top_item(&self) -> usize {
1426        let state = self.0.borrow();
1427        let top = state.bounds.top() - state.offset.borrow().y;
1428
1429        match state.child_bounds.binary_search_by(|bounds| {
1430            if top < bounds.top() {
1431                Ordering::Greater
1432            } else if top > bounds.bottom() {
1433                Ordering::Less
1434            } else {
1435                Ordering::Equal
1436            }
1437        }) {
1438            Ok(ix) => ix,
1439            Err(ix) => ix.min(state.child_bounds.len().saturating_sub(1)),
1440        }
1441    }
1442
1443    pub fn bounds_for_item(&self, ix: usize) -> Option<Bounds<Pixels>> {
1444        self.0.borrow().child_bounds.get(ix).cloned()
1445    }
1446
1447    /// scroll_to_item scrolls the minimal amount to ensure that the item is
1448    /// fully visible
1449    pub fn scroll_to_item(&self, ix: usize) {
1450        let state = self.0.borrow();
1451
1452        let Some(bounds) = state.child_bounds.get(ix) else {
1453            return;
1454        };
1455
1456        let scroll_offset = state.offset.borrow().y;
1457
1458        if bounds.top() + scroll_offset < state.bounds.top() {
1459            state.offset.borrow_mut().y = state.bounds.top() - bounds.top();
1460        } else if bounds.bottom() + scroll_offset > state.bounds.bottom() {
1461            state.offset.borrow_mut().y = state.bounds.bottom() - bounds.bottom();
1462        }
1463    }
1464
1465    pub fn logical_scroll_top(&self) -> (usize, Pixels) {
1466        let ix = self.top_item();
1467        let state = self.0.borrow();
1468
1469        if let Some(child_bounds) = state.child_bounds.get(ix) {
1470            (
1471                ix,
1472                child_bounds.top() + state.offset.borrow().y - state.bounds.top(),
1473            )
1474        } else {
1475            (ix, px(0.))
1476        }
1477    }
1478
1479    pub fn set_logical_scroll_top(&self, ix: usize, px: Pixels) {
1480        self.0.borrow_mut().requested_scroll_top = Some((ix, px));
1481    }
1482}