use gpui::{App, Global, SharedString};
use ui::menu::{self, Item};
use markdown::Cursor;
#[derive(Clone, Debug, PartialEq, Eq)]
pub struct Mention {
pub label: SharedString,
pub description: Option<SharedString>,
pub url: String,
}
pub type MentionSource = fn(query: &str, cx: &App) -> Vec<Mention>;
pub(crate) struct Installed(pub MentionSource);
impl Global for Installed {}
pub struct MentionMenu {
pub at: Cursor,
query: String,
asked: Option<String>,
rows: Vec<Mention>,
pub cursor: menu::Cursor,
}
impl MentionMenu {
pub fn open(at: Cursor) -> Self {
Self {
at,
query: String::new(),
asked: None,
rows: Vec::new(),
cursor: menu::Cursor::default(),
}
}
pub fn set_query(&mut self, query: String) {
self.query = query;
}
pub fn refresh(&mut self, cx: &App) {
if self.asked.as_ref() == Some(&self.query) {
return;
}
self.rows = cx
.try_global::<Installed>()
.map(|Installed(source)| source(&self.query, cx))
.unwrap_or_default();
self.asked = Some(self.query.clone());
self.cursor.clear();
self.cursor.step(&self.menu(), 1);
}
pub fn menu(&self) -> Vec<Item> {
self.rows
.iter()
.map(|row| {
let item = Item::action(row.label.clone());
match &row.description {
Some(description) => item.with_description(description.clone()),
None => item,
}
})
.collect()
}
pub fn step(&mut self, delta: isize) {
let menu = self.menu();
self.cursor.step(&menu, delta);
}
pub fn row(&self, ix: Option<usize>) -> Option<&Mention> {
self.rows.get(ix.or(self.cursor.row())?)
}
}