c5b22eee2d
This could still use some improvement UI-wise but the user experience
should be a lot better.
- [x] Show in "Window" application menu
- [x] Load prompt as it's selected in the picker
- [x] Refocus picker on `esc`
- [x] When creating a new prompt, if a new prompt already exists and is
unedited, activate it instead
- [x] Add `/default` command
- [x] Evaluate /commands on prompt insertion
- [x] Autocomplete /commands (but don't evaluate) during prompt editing
- [x] Show token count using the settings model, right-aligned in the
editor
- [x] Picker
- [x] Sorted alpha
- [x] 2 sublists
- Default
- Empty state: Star a prompt to add it to your default prompt
- Otherwise show prompts with star on hover
- All
- Move prompts with star on hover
Release Notes:
- N/A
105 lines
3.6 KiB
Rust
105 lines
3.6 KiB
Rust
use super::{file_command::FilePlaceholder, SlashCommand, SlashCommandOutput};
|
|
use anyhow::{anyhow, Result};
|
|
use assistant_slash_command::SlashCommandOutputSection;
|
|
use editor::Editor;
|
|
use gpui::{AppContext, Task, WeakView};
|
|
use language::LspAdapterDelegate;
|
|
use std::{borrow::Cow, sync::Arc};
|
|
use ui::{IntoElement, WindowContext};
|
|
use workspace::Workspace;
|
|
|
|
pub(crate) struct ActiveSlashCommand;
|
|
|
|
impl SlashCommand for ActiveSlashCommand {
|
|
fn name(&self) -> String {
|
|
"active".into()
|
|
}
|
|
|
|
fn description(&self) -> String {
|
|
"insert active tab".into()
|
|
}
|
|
|
|
fn menu_text(&self) -> String {
|
|
"Insert Active Tab".into()
|
|
}
|
|
|
|
fn complete_argument(
|
|
&self,
|
|
_query: String,
|
|
_cancel: std::sync::Arc<std::sync::atomic::AtomicBool>,
|
|
_workspace: WeakView<Workspace>,
|
|
_cx: &mut AppContext,
|
|
) -> Task<Result<Vec<String>>> {
|
|
Task::ready(Err(anyhow!("this command does not require argument")))
|
|
}
|
|
|
|
fn requires_argument(&self) -> bool {
|
|
false
|
|
}
|
|
|
|
fn run(
|
|
self: Arc<Self>,
|
|
_argument: Option<&str>,
|
|
workspace: WeakView<Workspace>,
|
|
_delegate: Arc<dyn LspAdapterDelegate>,
|
|
cx: &mut WindowContext,
|
|
) -> Task<Result<SlashCommandOutput>> {
|
|
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);
|
|
let text = cx.background_executor().spawn({
|
|
let path = path.clone();
|
|
async move {
|
|
let path = path
|
|
.as_ref()
|
|
.map(|path| path.to_string_lossy())
|
|
.unwrap_or_else(|| Cow::Borrowed("untitled"));
|
|
|
|
let mut output = String::with_capacity(path.len() + snapshot.len() + 9);
|
|
output.push_str("```");
|
|
output.push_str(&path);
|
|
output.push('\n');
|
|
for chunk in snapshot.as_rope().chunks() {
|
|
output.push_str(chunk);
|
|
}
|
|
if !output.ends_with('\n') {
|
|
output.push('\n');
|
|
}
|
|
output.push_str("```");
|
|
output
|
|
}
|
|
});
|
|
cx.foreground_executor().spawn(async move {
|
|
let text = text.await;
|
|
let range = 0..text.len();
|
|
Ok(SlashCommandOutput {
|
|
text,
|
|
sections: vec![SlashCommandOutputSection {
|
|
range,
|
|
render_placeholder: Arc::new(move |id, unfold, _| {
|
|
FilePlaceholder {
|
|
id,
|
|
path: path.clone(),
|
|
line_range: None,
|
|
unfold,
|
|
}
|
|
.into_any_element()
|
|
}),
|
|
}],
|
|
run_commands_in_text: false,
|
|
})
|
|
})
|
|
});
|
|
output.unwrap_or_else(|error| Task::ready(Err(error)))
|
|
}
|
|
}
|