use std::sync::Arc; use collections::{BTreeSet, HashMap};use derive_more::{Deref, DerefMut};use gpui::Global;use gpui::{AppContext, ReadGlobal};use parking_lot::RwLock; use crate::SlashCommand; #[derive(Default, Deref, DerefMut)]struct GlobalSlashCommandRegistry(Arc<SlashCommandRegistry>); impl Global for GlobalSlashCommandRegistry {} #[derive(Default)]struct SlashCommandRegistryState { commands: HashMap<Arc<str>, Arc<dyn SlashCommand>>, featured_commands: BTreeSet<Arc<str>>,} #[derive(Default)]pub struct SlashCommandRegistry { state: RwLock<SlashCommandRegistryState>,} impl SlashCommandRegistry { /// Returns the global [`SlashCommandRegistry`]. pub fn global(cx: &AppContext) -> Arc<Self> { GlobalSlashCommandRegistry::global(cx).0.clone() } /// Returns the global [`SlashCommandRegistry`]. /// /// Inserts a default [`SlashCommandRegistry`] if one does not yet exist. pub fn default_global(cx: &mut AppContext) -> Arc<Self> { cx.default_global::<GlobalSlashCommandRegistry>().0.clone() } pub fn new() -> Arc<Self> { Arc::new(Self { state: RwLock::new(SlashCommandRegistryState { commands: HashMap::default(), featured_commands: BTreeSet::default(), }), }) } /// Registers the provided [`SlashCommand`]. pub fn register_command(&self, command: impl SlashCommand, is_featured: bool) { let mut state = self.state.write(); let command_name: Arc<str> = command.name().into(); if is_featured { state.featured_commands.insert(command_name.clone()); } state.commands.insert(command_name, Arc::new(command)); } /// Unregisters the provided [`SlashCommand`]. pub fn unregister_command(&self, command: impl SlashCommand) { self.unregister_command_by_name(command.name().as_str()) } /// Unregisters the command with the given name. pub fn unregister_command_by_name(&self, command_name: &str) { let mut state = self.state.write(); state.featured_commands.remove(command_name); state.commands.remove(command_name); } /// Returns the names of registered [`SlashCommand`]s. pub fn command_names(&self) -> Vec<Arc<str>> { self.state.read().commands.keys().cloned().collect() } /// Returns the names of registered, featured [`SlashCommand`]s. pub fn featured_command_names(&self) -> Vec<Arc<str>> { self.state .read() .featured_commands .iter() .cloned() .collect() } /// Returns the [`SlashCommand`] with the given name. pub fn command(&self, name: &str) -> Option<Arc<dyn SlashCommand>> { self.state.read().commands.get(name).cloned() }}
use std::sync::Arc;
use collections::{BTreeSet, HashMap};
use derive_more::{Deref, DerefMut};
use gpui::Global;
use gpui::{AppContext, ReadGlobal};
use parking_lot::RwLock;
use crate::SlashCommand;
#[derive(Default, Deref, DerefMut)]
struct GlobalSlashCommandRegistry(Arc<SlashCommandRegistry>);
impl Global for GlobalSlashCommandRegistry {}
#[derive(Default)]
struct SlashCommandRegistryState {
commands: HashMap<Arc<str>, Arc<dyn SlashCommand>>,
featured_commands: BTreeSet<Arc<str>>,
}
#[derive(Default)]
pub struct SlashCommandRegistry {
state: RwLock<SlashCommandRegistryState>,
}
impl SlashCommandRegistry {
/// Returns the global [`SlashCommandRegistry`].
pub fn global(cx: &AppContext) -> Arc<Self> {
GlobalSlashCommandRegistry::global(cx).0.clone()
}
/// Returns the global [`SlashCommandRegistry`].
///
/// Inserts a default [`SlashCommandRegistry`] if one does not yet exist.
pub fn default_global(cx: &mut AppContext) -> Arc<Self> {
cx.default_global::<GlobalSlashCommandRegistry>().0.clone()
}
pub fn new() -> Arc<Self> {
Arc::new(Self {
state: RwLock::new(SlashCommandRegistryState {
commands: HashMap::default(),
featured_commands: BTreeSet::default(),
}),
})
}
/// Registers the provided [`SlashCommand`].
pub fn register_command(&self, command: impl SlashCommand, is_featured: bool) {
let mut state = self.state.write();
let command_name: Arc<str> = command.name().into();
if is_featured {
state.featured_commands.insert(command_name.clone());
}
state.commands.insert(command_name, Arc::new(command));
}
/// Unregisters the provided [`SlashCommand`].
pub fn unregister_command(&self, command: impl SlashCommand) {
self.unregister_command_by_name(command.name().as_str())
}
/// Unregisters the command with the given name.
pub fn unregister_command_by_name(&self, command_name: &str) {
let mut state = self.state.write();
state.featured_commands.remove(command_name);
state.commands.remove(command_name);
}
/// Returns the names of registered [`SlashCommand`]s.
pub fn command_names(&self) -> Vec<Arc<str>> {
self.state.read().commands.keys().cloned().collect()
}
/// Returns the names of registered, featured [`SlashCommand`]s.
pub fn featured_command_names(&self) -> Vec<Arc<str>> {
self.state
.read()
.featured_commands
.iter()
.cloned()
.collect()
}
/// Returns the [`SlashCommand`] with the given name.
pub fn command(&self, name: &str) -> Option<Arc<dyn SlashCommand>> {
self.state.read().commands.get(name).cloned()
}
}