92 lines
2.9 KiB
Rust
92 lines
2.9 KiB
Rust
use collections::HashMap;
|
|
use component::ComponentId;
|
|
use gpui::{App, Entity, WeakEntity};
|
|
use linkme::distributed_slice;
|
|
use std::sync::OnceLock;
|
|
use ui::{AnyElement, Component, ComponentScope, Window};
|
|
use workspace::Workspace;
|
|
|
|
use crate::{ActiveThread, ThreadStore};
|
|
|
|
/// Function type for creating agent component previews
|
|
pub type PreviewFn = fn(
|
|
WeakEntity<Workspace>,
|
|
Entity<ActiveThread>,
|
|
WeakEntity<ThreadStore>,
|
|
&mut Window,
|
|
&mut App,
|
|
) -> Option<AnyElement>;
|
|
|
|
/// Distributed slice for preview registration functions
|
|
#[distributed_slice]
|
|
pub static __ALL_AGENT_PREVIEWS: [fn() -> (ComponentId, PreviewFn)] = [..];
|
|
|
|
/// Trait that must be implemented by components that provide agent previews.
|
|
pub trait AgentPreview: Component + Sized {
|
|
#[allow(unused)] // We can't know this is used due to the distributed slice
|
|
fn scope(&self) -> ComponentScope {
|
|
ComponentScope::Agent
|
|
}
|
|
|
|
/// Static method to create a preview for this component type
|
|
fn agent_preview(
|
|
workspace: WeakEntity<Workspace>,
|
|
active_thread: Entity<ActiveThread>,
|
|
thread_store: WeakEntity<ThreadStore>,
|
|
window: &mut Window,
|
|
cx: &mut App,
|
|
) -> Option<AnyElement>;
|
|
}
|
|
|
|
/// Register an agent preview for the given component type
|
|
#[macro_export]
|
|
macro_rules! register_agent_preview {
|
|
($type:ty) => {
|
|
#[linkme::distributed_slice($crate::ui::preview::__ALL_AGENT_PREVIEWS)]
|
|
static __REGISTER_AGENT_PREVIEW: fn() -> (
|
|
component::ComponentId,
|
|
$crate::ui::preview::PreviewFn,
|
|
) = || {
|
|
(
|
|
<$type as component::Component>::id(),
|
|
<$type as $crate::ui::preview::AgentPreview>::agent_preview,
|
|
)
|
|
};
|
|
};
|
|
}
|
|
|
|
/// Lazy initialized registry of preview functions
|
|
static AGENT_PREVIEW_REGISTRY: OnceLock<HashMap<ComponentId, PreviewFn>> = OnceLock::new();
|
|
|
|
/// Initialize the agent preview registry if needed
|
|
fn get_or_init_registry() -> &'static HashMap<ComponentId, PreviewFn> {
|
|
AGENT_PREVIEW_REGISTRY.get_or_init(|| {
|
|
let mut map = HashMap::default();
|
|
for register_fn in __ALL_AGENT_PREVIEWS.iter() {
|
|
let (id, preview_fn) = register_fn();
|
|
map.insert(id, preview_fn);
|
|
}
|
|
map
|
|
})
|
|
}
|
|
|
|
/// Get a specific agent preview by component ID.
|
|
pub fn get_agent_preview(
|
|
id: &ComponentId,
|
|
workspace: WeakEntity<Workspace>,
|
|
active_thread: Entity<ActiveThread>,
|
|
thread_store: WeakEntity<ThreadStore>,
|
|
window: &mut Window,
|
|
cx: &mut App,
|
|
) -> Option<AnyElement> {
|
|
let registry = get_or_init_registry();
|
|
registry
|
|
.get(id)
|
|
.and_then(|preview_fn| preview_fn(workspace, active_thread, thread_store, window, cx))
|
|
}
|
|
|
|
/// Get all registered agent previews.
|
|
pub fn all_agent_previews() -> Vec<ComponentId> {
|
|
let registry = get_or_init_registry();
|
|
registry.keys().cloned().collect()
|
|
}
|