orch-cli 0.91.1

//! CLI handlers for `orch cooldown list` and `orch cooldown clear`.
//!
//! Cooldowns are persisted to the SQLite KV store so they survive restarts.
//! The `list` command opens the store, hydrates the in-memory map, then
//! prints all active entries — accurate even when the service is not running.

use crate::engine::cooldown;
use std::sync::Arc;

/// List all active agent and model cooldowns.
pub async fn list() -> anyhow::Result<()> {
    let store = Arc::new(crate::cli::init_store().await?);
    // Hydrate in-memory map from persisted KV so we see the full state.
    cooldown::init_cooldown_store(store).await;

    let entries = cooldown::list_all_cooldowns();
    if entries.is_empty() {
        println!("No active cooldowns.");
        return Ok(());
    }

    let now = chrono::Utc::now().timestamp();
    println!("{:<30} {:<12} REASON", "KEY", "REMAINING");
    println!("{}", "-".repeat(72_usize));
    for (key, until, reason) in entries {
        let remaining_secs = until - now;
        let remaining = format_remaining(remaining_secs);
        println!("{:<30} {:<12} {}", key, remaining, reason);
    }
    Ok(())
}

/// Clear a specific cooldown by key, or all cooldowns with `--all`.
pub async fn clear(key: Option<String>, all: bool) -> anyhow::Result<()> {
    let store = Arc::new(crate::cli::init_store().await?);
    // Hydrate first so clear operates on the current state.
    cooldown::init_cooldown_store(store.clone()).await;

    if all {
        cooldown::clear_cooldown("*", &store).await;
        println!("Cleared all cooldowns.");
    } else if let Some(k) = key {
        cooldown::clear_cooldown(&k, &store).await;
        println!("Cleared cooldown: {k}");
    } else {
        anyhow::bail!("specify a key (e.g. 'claude' or 'claude:sonnet') or use --all");
    }
    Ok(())
}

fn format_remaining(secs: i64) -> String {
    if secs <= 0 {
        return "expired".to_string();
    }
    if secs < 3600 {
        format!("{}m", secs / 60)
    } else if secs < 86400 {
        let h = secs / 3600;
        let m = (secs % 3600) / 60;
        if m == 0 {
            format!("{h}h")
        } else {
            format!("{h}h{m}m")
        }
    } else {
        let d = secs / 86400;
        let h = (secs % 86400) / 3600;
        if h == 0 {
            format!("{d}d")
        } else {
            format!("{d}d{h}h")
        }
    }
}