101 lines
3.3 KiB
Rust
101 lines
3.3 KiB
Rust
use anyhow::{Context as _, Result, anyhow};
|
|
use assistant_slash_command::{
|
|
ArgumentCompletion, SlashCommand, SlashCommandOutput, SlashCommandOutputSection,
|
|
SlashCommandResult,
|
|
};
|
|
use editor::Editor;
|
|
use gpui::{AppContext as _, Task, WeakEntity};
|
|
use language::{BufferSnapshot, LspAdapterDelegate};
|
|
use std::sync::Arc;
|
|
use std::{path::Path, sync::atomic::AtomicBool};
|
|
use ui::{App, IconName, Window};
|
|
use workspace::Workspace;
|
|
|
|
pub struct OutlineSlashCommand;
|
|
|
|
impl SlashCommand for OutlineSlashCommand {
|
|
fn name(&self) -> String {
|
|
"symbols".into()
|
|
}
|
|
|
|
fn description(&self) -> String {
|
|
"Insert symbols for active tab".into()
|
|
}
|
|
|
|
fn icon(&self) -> IconName {
|
|
IconName::ListTree
|
|
}
|
|
|
|
fn menu_text(&self) -> String {
|
|
self.description()
|
|
}
|
|
|
|
fn complete_argument(
|
|
self: Arc<Self>,
|
|
_arguments: &[String],
|
|
_cancel: Arc<AtomicBool>,
|
|
_workspace: Option<WeakEntity<Workspace>>,
|
|
_window: &mut Window,
|
|
_cx: &mut App,
|
|
) -> Task<Result<Vec<ArgumentCompletion>>> {
|
|
Task::ready(Err(anyhow!("this command does not require argument")))
|
|
}
|
|
|
|
fn requires_argument(&self) -> bool {
|
|
false
|
|
}
|
|
|
|
fn run(
|
|
self: Arc<Self>,
|
|
_arguments: &[String],
|
|
_context_slash_command_output_sections: &[SlashCommandOutputSection<language::Anchor>],
|
|
_context_buffer: BufferSnapshot,
|
|
workspace: WeakEntity<Workspace>,
|
|
_delegate: Option<Arc<dyn LspAdapterDelegate>>,
|
|
_: &mut Window,
|
|
cx: &mut App,
|
|
) -> Task<SlashCommandResult> {
|
|
let output = workspace.update(cx, |workspace, cx| {
|
|
let Some(active_item) = workspace.active_item(cx) else {
|
|
return Task::ready(Err(anyhow!("no active tab")));
|
|
};
|
|
let Some(buffer) = active_item
|
|
.downcast::<Editor>()
|
|
.and_then(|editor| editor.read(cx).buffer().read(cx).as_singleton())
|
|
else {
|
|
return Task::ready(Err(anyhow!("active tab is not an editor")));
|
|
};
|
|
|
|
let snapshot = buffer.read(cx).snapshot();
|
|
let path = snapshot.resolve_file_path(cx, true);
|
|
|
|
cx.background_spawn(async move {
|
|
let outline = snapshot
|
|
.outline(None)
|
|
.context("no symbols for active tab")?;
|
|
|
|
let path = path.as_deref().unwrap_or(Path::new("untitled"));
|
|
let mut outline_text = format!("Symbols for {}:\n", path.display());
|
|
for item in &outline.path_candidates {
|
|
outline_text.push_str("- ");
|
|
outline_text.push_str(&item.string);
|
|
outline_text.push('\n');
|
|
}
|
|
|
|
Ok(SlashCommandOutput {
|
|
sections: vec![SlashCommandOutputSection {
|
|
range: 0..outline_text.len(),
|
|
icon: IconName::ListTree,
|
|
label: path.to_string_lossy().to_string().into(),
|
|
metadata: None,
|
|
}],
|
|
text: outline_text,
|
|
run_commands_in_text: false,
|
|
}
|
|
.to_event_stream())
|
|
})
|
|
});
|
|
|
|
output.unwrap_or_else(|error| Task::ready(Err(error)))
|
|
}
|
|
}
|