1use super::{SlashCommand, SlashCommandOutput};
 2use crate::prompt_library::PromptStore;
 3use anyhow::{anyhow, Result};
 4use assistant_slash_command::{ArgumentCompletion, SlashCommandOutputSection};
 5use gpui::{AppContext, Task, WeakView};
 6use language::LspAdapterDelegate;
 7use std::{
 8    fmt::Write,
 9    sync::{atomic::AtomicBool, Arc},
10};
11use ui::prelude::*;
12use workspace::Workspace;
13
14pub(crate) struct DefaultSlashCommand;
15
16impl SlashCommand for DefaultSlashCommand {
17    fn name(&self) -> String {
18        "default".into()
19    }
20
21    fn description(&self) -> String {
22        "insert default prompt".into()
23    }
24
25    fn menu_text(&self) -> String {
26        "Insert Default Prompt".into()
27    }
28
29    fn requires_argument(&self) -> bool {
30        false
31    }
32
33    fn complete_argument(
34        self: Arc<Self>,
35        _query: String,
36        _cancellation_flag: Arc<AtomicBool>,
37        _workspace: Option<WeakView<Workspace>>,
38        _cx: &mut AppContext,
39    ) -> Task<Result<Vec<ArgumentCompletion>>> {
40        Task::ready(Err(anyhow!("this command does not require argument")))
41    }
42
43    fn run(
44        self: Arc<Self>,
45        _argument: Option<&str>,
46        _workspace: WeakView<Workspace>,
47        _delegate: Arc<dyn LspAdapterDelegate>,
48        cx: &mut WindowContext,
49    ) -> Task<Result<SlashCommandOutput>> {
50        let store = PromptStore::global(cx);
51        cx.background_executor().spawn(async move {
52            let store = store.await?;
53            let prompts = store.default_prompt_metadata();
54
55            let mut text = String::new();
56            text.push('\n');
57            for prompt in prompts {
58                if let Some(title) = prompt.title {
59                    writeln!(text, "/prompt {}", title).unwrap();
60                }
61            }
62            text.pop();
63
64            if text.is_empty() {
65                text.push('\n');
66            }
67
68            Ok(SlashCommandOutput {
69                sections: vec![SlashCommandOutputSection {
70                    range: 0..text.len(),
71                    icon: IconName::Library,
72                    label: "Default".into(),
73                }],
74                text,
75                run_commands_in_text: true,
76            })
77        })
78    }
79}