systemprompt-cli 0.14.5

use anyhow::Result;
use clap::Args;
use schemars::JsonSchema;
use serde::{Deserialize, Serialize};
use std::path::PathBuf;
use systemprompt_analytics::RequestAnalyticsRepository;
use systemprompt_identifiers::{AiRequestId, UserId};
use systemprompt_logging::CliService;
use systemprompt_runtime::{AppContext, DatabaseContext};

use crate::CliConfig;
use crate::commands::analytics::shared::{export_to_csv, parse_time_range, resolve_export_path};
use crate::shared::CommandOutput;

#[derive(Debug, Args)]
pub struct ListArgs {
    #[arg(
        long,
        alias = "from",
        default_value = "24h",
        help = "Time range (e.g., '1h', '24h', '7d')"
    )]
    pub since: Option<String>,

    #[arg(long, alias = "to", help = "End time for range")]
    pub until: Option<String>,

    #[arg(
        long,
        short = 'n',
        default_value = "20",
        help = "Maximum number of requests"
    )]
    pub limit: i64,

    #[arg(long, help = "Filter by model name")]
    pub model: Option<String>,

    #[arg(long, help = "Export results to CSV file")]
    pub export: Option<PathBuf>,
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
pub(super) struct RequestListRowOutput {
    pub id: AiRequestId,
    pub user_id: UserId,
    pub provider: String,
    pub model: String,
    pub status: String,
    pub error_message: Option<String>,
    pub input_tokens: i32,
    pub output_tokens: i32,
    pub cost_microdollars: i64,
    pub latency_ms: i32,
    pub cache_hit: bool,
    pub created_at: String,
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
pub(super) struct RequestListOutput {
    pub total: i64,
    pub requests: Vec<RequestListRowOutput>,
}

pub(super) async fn execute(args: ListArgs, _config: &CliConfig) -> Result<CommandOutput> {
    let ctx = AppContext::new().await?;
    let repo = RequestAnalyticsRepository::new(ctx.db_pool())?;
    execute_internal(args, &repo).await
}

pub(super) async fn execute_with_pool(
    args: ListArgs,
    db_ctx: &DatabaseContext,
    _config: &CliConfig,
) -> Result<CommandOutput> {
    let repo = RequestAnalyticsRepository::new(db_ctx.db_pool())?;
    execute_internal(args, &repo).await
}

async fn execute_internal(
    args: ListArgs,
    repo: &RequestAnalyticsRepository,
) -> Result<CommandOutput> {
    let (start, end) = parse_time_range(args.since.as_ref(), args.until.as_ref())?;
    let rows = repo
        .list_requests(start, end, args.limit, args.model.as_deref())
        .await?;

    let requests: Vec<RequestListRowOutput> = rows
        .into_iter()
        .map(|row| RequestListRowOutput {
            id: AiRequestId::new(row.id),
            user_id: row.user_id,
            provider: row.provider,
            model: row.model,
            status: row.status,
            error_message: row.error_message,
            input_tokens: row.input_tokens.unwrap_or(0),
            output_tokens: row.output_tokens.unwrap_or(0),
            cost_microdollars: row.cost_microdollars.unwrap_or(0),
            latency_ms: row.latency_ms.unwrap_or(0),
            cache_hit: row.cache_hit.unwrap_or(false),
            created_at: row.created_at.format("%Y-%m-%d %H:%M:%S").to_string(),
        })
        .collect();

    let output = RequestListOutput {
        total: requests.len() as i64,
        requests,
    };

    if let Some(ref path) = args.export {
        let resolved_path = resolve_export_path(path)?;
        export_to_csv(&output.requests, &resolved_path)?;
        CliService::success(&format!("Exported to {}", resolved_path.display()));
        return Ok(CommandOutput::table_of(
            vec![
                "created_at",
                "status",
                "user_id",
                "provider",
                "model",
                "input_tokens",
                "output_tokens",
                "latency_ms",
                "error_message",
            ],
            &output.requests,
        )
        .with_skip_render());
    }

    if output.requests.is_empty() {
        CliService::warning("No requests found in the specified time range");
        return Ok(CommandOutput::table_of(
            vec![
                "created_at",
                "status",
                "user_id",
                "provider",
                "model",
                "input_tokens",
                "output_tokens",
                "latency_ms",
                "error_message",
            ],
            &output.requests,
        )
        .with_skip_render());
    }

    Ok(CommandOutput::table_of(
        vec![
            "created_at",
            "status",
            "user_id",
            "provider",
            "model",
            "input_tokens",
            "output_tokens",
            "latency_ms",
            "error_message",
        ],
        &output.requests,
    )
    .with_title("AI Requests"))
}