68accaeb00
This PR improves the completions for arguments in the `/docs` slash command. We achieved this by extending the `complete_argument` method on the `SlashCommand` trait to return a `Vec<ArgumentCompletion>` instead of a `Vec<String>`. In addition to the completion `label`, `ArgumentCompletion` has two new fields that are can be used to customize the completion behavior: - `new_text`: The actual text that will be inserted when the completion is accepted, which may be different from what is shown by the completion label. - `run_command`: Whether the command is run when the completion is accepted. This can be set to `false` to allow accepting a completion without running the command. Release Notes: - N/A --------- Co-authored-by: Antonio <antonio@zed.dev>
79 lines
2.2 KiB
Rust
79 lines
2.2 KiB
Rust
use super::{SlashCommand, SlashCommandOutput};
|
|
use crate::prompt_library::PromptStore;
|
|
use anyhow::{anyhow, Result};
|
|
use assistant_slash_command::{ArgumentCompletion, SlashCommandOutputSection};
|
|
use gpui::{AppContext, Task, WeakView};
|
|
use language::LspAdapterDelegate;
|
|
use std::{
|
|
fmt::Write,
|
|
sync::{atomic::AtomicBool, Arc},
|
|
};
|
|
use ui::prelude::*;
|
|
use workspace::Workspace;
|
|
|
|
pub(crate) struct DefaultSlashCommand;
|
|
|
|
impl SlashCommand for DefaultSlashCommand {
|
|
fn name(&self) -> String {
|
|
"default".into()
|
|
}
|
|
|
|
fn description(&self) -> String {
|
|
"insert default prompt".into()
|
|
}
|
|
|
|
fn menu_text(&self) -> String {
|
|
"Insert Default Prompt".into()
|
|
}
|
|
|
|
fn requires_argument(&self) -> bool {
|
|
false
|
|
}
|
|
|
|
fn complete_argument(
|
|
self: Arc<Self>,
|
|
_query: String,
|
|
_cancellation_flag: Arc<AtomicBool>,
|
|
_workspace: Option<WeakView<Workspace>>,
|
|
_cx: &mut AppContext,
|
|
) -> Task<Result<Vec<ArgumentCompletion>>> {
|
|
Task::ready(Err(anyhow!("this command does not require argument")))
|
|
}
|
|
|
|
fn run(
|
|
self: Arc<Self>,
|
|
_argument: Option<&str>,
|
|
_workspace: WeakView<Workspace>,
|
|
_delegate: Arc<dyn LspAdapterDelegate>,
|
|
cx: &mut WindowContext,
|
|
) -> Task<Result<SlashCommandOutput>> {
|
|
let store = PromptStore::global(cx);
|
|
cx.background_executor().spawn(async move {
|
|
let store = store.await?;
|
|
let prompts = store.default_prompt_metadata();
|
|
|
|
let mut text = String::new();
|
|
text.push('\n');
|
|
for prompt in prompts {
|
|
if let Some(title) = prompt.title {
|
|
writeln!(text, "/prompt {}", title).unwrap();
|
|
}
|
|
}
|
|
text.pop();
|
|
|
|
if text.is_empty() {
|
|
text.push('\n');
|
|
}
|
|
|
|
Ok(SlashCommandOutput {
|
|
sections: vec![SlashCommandOutputSection {
|
|
range: 0..text.len(),
|
|
icon: IconName::Library,
|
|
label: "Default".into(),
|
|
}],
|
|
text,
|
|
run_commands_in_text: true,
|
|
})
|
|
})
|
|
}
|
|
}
|