51 lines
1.4 KiB
Rust
51 lines
1.4 KiB
Rust
use std::collections::HashMap;
|
|
|
|
use ely_gpui_component::primitives::Measure;
|
|
use gpui::{
|
|
AnyWindowHandle, App, Bounds, Div, Global, IntoElement, ParentElement, Pixels, Styled, div,
|
|
};
|
|
|
|
/// Window bounds of named demo elements, for scripted capture.
|
|
#[derive(Default)]
|
|
pub struct Probes(HashMap<&'static str, Bounds<Pixels>>);
|
|
|
|
impl Global for Probes {}
|
|
|
|
impl Probes {
|
|
pub fn get(key: &str, cx: &App) -> Option<Bounds<Pixels>> {
|
|
cx.try_global::<Probes>()
|
|
.and_then(|probes| probes.0.get(key).copied())
|
|
}
|
|
}
|
|
|
|
/// Wraps `element`, sized to it, so scripts can find it by `key`.
|
|
pub fn probe(key: &'static str, element: impl IntoElement) -> Div {
|
|
div().flex().child(
|
|
Measure::new(key, move |bounds, _, cx| {
|
|
cx.default_global::<Probes>().0.insert(key, bounds);
|
|
})
|
|
.child(element),
|
|
)
|
|
}
|
|
|
|
/// Windows the demos opened, by key, for scripted capture.
|
|
#[derive(Default)]
|
|
pub struct Opened(HashMap<&'static str, AnyWindowHandle>);
|
|
|
|
impl Global for Opened {}
|
|
|
|
impl Opened {
|
|
pub fn insert(key: &'static str, handle: AnyWindowHandle, cx: &mut App) {
|
|
cx.default_global::<Opened>().0.insert(key, handle);
|
|
}
|
|
|
|
pub fn take(key: &str, cx: &mut App) -> Option<AnyWindowHandle> {
|
|
cx.default_global::<Opened>().0.remove(key)
|
|
}
|
|
|
|
pub fn get(key: &str, cx: &App) -> Option<AnyWindowHandle> {
|
|
cx.try_global::<Opened>()
|
|
.and_then(|opened| opened.0.get(key).copied())
|
|
}
|
|
}
|