dirge-agent 0.18.1

//! /prompt command dispatch.

pub(crate) mod default;
pub(crate) mod list;
pub(crate) mod switch;

use crate::ui::slash::{SlashCtx, SlashOutcome};

pub(crate) async fn cmd_prompt(
    ctx: &mut SlashCtx<'_>,
    parts: &[&str],
) -> anyhow::Result<SlashOutcome> {
    if parts.len() < 2 {
        return list::cmd_prompt_list(ctx)
            .await
            .map(|_| SlashOutcome::Handled);
    }
    if parts[1] == "default" && !ctx.context.prompts.contains_key("default") {
        return default::cmd_prompt_default(ctx)
            .await
            .map(|_| SlashOutcome::Handled);
    }
    let name = parts[1].trim();
    switch::cmd_prompt_switch(ctx, name).await?;
    // `/prompt <name> <text...>`: switch then run a streamed turn on the
    // trailing text, mirroring `/plan`. Only when the named prompt actually
    // resolved — switch renders an "unknown prompt" notice and returns Ok for a
    // bad name, so gate the defer on the prompt existing. Slash handlers can't
    // touch the loop's run slots, so the DeferPromptRun outcome hands the text
    // back to the UI loop, which launches the turn (see the match on
    // handle_slash's result in ui/mod.rs).
    if ctx.context.prompts.contains_key(name)
        && let Some(prompt) = defer_prompt_run_prompt(parts)
    {
        return Ok(SlashOutcome::DeferPromptRun { prompt });
    }
    Ok(SlashOutcome::Handled)
}

/// `/prompt <name> <text...>`: when the user typed trailing words after the
/// prompt name, this is the prompt text `cmd_prompt` defers so the UI loop
/// can launch a streamed turn on it (mirroring `/plan`). Returns `None` for a
/// bare `/prompt <name>`.
fn defer_prompt_run_prompt(parts: &[&str]) -> Option<String> {
    (parts.len() > 2).then(|| parts[2..].join(" "))
}

#[cfg(test)]
mod tests {
    use super::defer_prompt_run_prompt;

    #[test]
    fn defer_prompt_joins_trailing_text_after_the_prompt_name() {
        let parts = ["/prompt", "review", "please", "review", "my", "changes"];
        assert_eq!(
            defer_prompt_run_prompt(&parts).as_deref(),
            Some("please review my changes")
        );
    }

    #[test]
    fn defer_prompt_is_none_for_bare_prompt_switch() {
        let parts = ["/prompt", "review"];
        assert_eq!(defer_prompt_run_prompt(&parts), None);
    }

    #[test]
    fn defer_prompt_preserves_a_single_trailing_word() {
        let parts = ["/prompt", "plan", "go"];
        assert_eq!(defer_prompt_run_prompt(&parts).as_deref(), Some("go"));
    }
}