systemprompt-cli 0.19.0

use anyhow::{Context, Result, bail};
use clap::Args;
use systemprompt_agent::repository::context::ContextRepository;
use systemprompt_logging::CliService;

use super::resolve::resolve_context;
use super::types::ContextDeletedOutput;
use crate::context::CommandContext;
use crate::session::get_or_create_session;
use crate::shared::CommandOutput;

#[derive(Debug, Args)]
pub struct DeleteArgs {
    #[arg(help = "Context ID (full, partial prefix, or name)")]
    pub context: String,

    #[arg(short = 'y', long, help = "Skip confirmation prompt")]
    pub yes: bool,
}

pub(super) async fn execute(args: DeleteArgs, ctx: &CommandContext) -> Result<CommandOutput> {
    let session_ctx = get_or_create_session(ctx).await?;
    let pool = ctx.db_pool().await?;

    let repo = ContextRepository::new(&pool)?;

    let context_id = resolve_context(&args.context, &session_ctx.session.user_id, &repo).await?;

    if context_id == session_ctx.session.context_id {
        bail!(
            "Cannot delete the active context. Switch to a different context first with 'contexts \
             use <id>'."
        );
    }

    let context = repo
        .get_context(&context_id, &session_ctx.session.user_id)
        .await
        .context("Failed to fetch context details")?;

    if !args.yes && ctx.cli.is_interactive() {
        CliService::warning(&format!(
            "You are about to delete context '{}' ({})",
            context.name,
            context_id.as_str()
        ));

        if !ctx.prompter().confirm("Are you sure?", false)? {
            CliService::info("Deletion cancelled");
            let cancelled = ContextDeletedOutput {
                id: context_id,
                message: "Deletion cancelled".to_owned(),
            };
            return Ok(CommandOutput::card_value(
                "Context Delete Cancelled",
                &cancelled,
            ));
        }
    }

    repo.delete_context(&context_id, &session_ctx.session.user_id)
        .await
        .context("Failed to delete context")?;

    let output = ContextDeletedOutput {
        id: context_id.clone(),
        message: format!("Context '{}' deleted successfully", context.name),
    };

    if !ctx.cli.is_json_output() {
        CliService::success(&output.message);
    }

    Ok(CommandOutput::card_value("Context Deleted", &output))
}