1use crate::prompts::PromptBuilder;
 2use std::sync::Arc;
 3
 4use std::sync::atomic::AtomicBool;
 5
 6use anyhow::Result;
 7use assistant_slash_command::{
 8    ArgumentCompletion, SlashCommand, SlashCommandOutput, SlashCommandOutputSection,
 9};
10use gpui::{Task, WeakView};
11use language::{BufferSnapshot, LspAdapterDelegate};
12use ui::prelude::*;
13
14use workspace::Workspace;
15
16pub(crate) struct WorkflowSlashCommand {
17    prompt_builder: Arc<PromptBuilder>,
18}
19
20impl WorkflowSlashCommand {
21    pub fn new(prompt_builder: Arc<PromptBuilder>) -> Self {
22        Self { prompt_builder }
23    }
24}
25
26impl SlashCommand for WorkflowSlashCommand {
27    fn name(&self) -> String {
28        "workflow".into()
29    }
30
31    fn description(&self) -> String {
32        "insert a prompt that opts into the edit workflow".into()
33    }
34
35    fn menu_text(&self) -> String {
36        "Insert Workflow Prompt".into()
37    }
38
39    fn requires_argument(&self) -> bool {
40        false
41    }
42
43    fn complete_argument(
44        self: Arc<Self>,
45        _arguments: &[String],
46        _cancel: Arc<AtomicBool>,
47        _workspace: Option<WeakView<Workspace>>,
48        _cx: &mut WindowContext,
49    ) -> Task<Result<Vec<ArgumentCompletion>>> {
50        Task::ready(Ok(Vec::new()))
51    }
52
53    fn run(
54        self: Arc<Self>,
55        _arguments: &[String],
56        _context_slash_command_output_sections: &[SlashCommandOutputSection<language::Anchor>],
57        _context_buffer: BufferSnapshot,
58        _workspace: WeakView<Workspace>,
59        _delegate: Option<Arc<dyn LspAdapterDelegate>>,
60        cx: &mut WindowContext,
61    ) -> Task<Result<SlashCommandOutput>> {
62        let prompt_builder = self.prompt_builder.clone();
63        cx.spawn(|_cx| async move {
64            let text = prompt_builder.generate_workflow_prompt()?;
65            let range = 0..text.len();
66
67            Ok(SlashCommandOutput {
68                text,
69                sections: vec![SlashCommandOutputSection {
70                    range,
71                    icon: IconName::Route,
72                    label: "Workflow".into(),
73                    metadata: None,
74                }],
75                run_commands_in_text: false,
76            })
77        })
78    }
79}