zeptoclaw 0.7.4

//! Status and auth command handlers.

use std::io::{self, Write};

use anyhow::{Context, Result};

use zeptoclaw::auth;
use zeptoclaw::auth::codex_import;
use zeptoclaw::config::{Config, ContainerAgentBackend, ProviderConfig};
use zeptoclaw::providers::{
    configured_unsupported_provider_names, resolve_runtime_provider, RUNTIME_SUPPORTED_PROVIDERS,
};
use zeptoclaw::runtime::available_runtimes;

use super::common::{memory_backend_label, memory_citations_label, skills_loader_from_config};
use super::heartbeat::heartbeat_file_path;
use super::AuthAction;

/// Manage authentication.
pub(crate) async fn cmd_auth(action: AuthAction) -> Result<()> {
    match action {
        AuthAction::Login { provider } => {
            cmd_auth_login(provider).await?;
        }
        AuthAction::Logout { provider } => {
            cmd_auth_logout(provider)?;
        }
        AuthAction::Status => {
            cmd_auth_status().await?;
        }
        AuthAction::Refresh { provider } => {
            cmd_auth_refresh(&provider).await?;
        }
        AuthAction::SetupToken => {
            cmd_auth_setup_token().await?;
        }
    }
    Ok(())
}

/// OAuth login flow.
async fn cmd_auth_login(provider: Option<String>) -> Result<()> {
    let provider = provider.unwrap_or_else(|| {
        println!(
            "OAuth-supported providers: {}",
            auth::oauth_supported_providers().join(", ")
        );
        println!();
        "anthropic".to_string()
    });

    // OpenAI: import-first with hardcoded public client ID
    if provider == "openai" {
        return cmd_auth_login_openai().await;
    }

    // All other providers: existing flow (env-var client_id + ephemeral port)
    let oauth_config = auth::provider_oauth_config(&provider).ok_or_else(|| {
        anyhow::anyhow!(
            "Provider '{}' does not support OAuth authentication.\n\
             Supported providers: {}\n\n\
             To configure API keys instead:\n  \
             export ZEPTOCLAW_PROVIDERS_{}_API_KEY=your-key-here",
            provider,
            auth::oauth_supported_providers().join(", "),
            provider.to_uppercase()
        )
    })?;

    println!("WARNING: Using OAuth subscription tokens for API access may violate");
    println!("the provider's Terms of Service. The provider may block these tokens");
    println!("at any time. If blocked, ZeptoClaw will fall back to your API key.");
    println!();

    let provider_env = format!(
        "ZEPTOCLAW_PROVIDERS_{}_OAUTH_CLIENT_ID",
        provider.to_uppercase()
    );
    let client_id = std::env::var(&provider_env)
        .ok()
        .filter(|v| !v.trim().is_empty())
        .or_else(|| {
            std::env::var("ZEPTOCLAW_OAUTH_CLIENT_ID")
                .ok()
                .filter(|v| !v.trim().is_empty())
        })
        .ok_or_else(|| {
            anyhow::anyhow!(
                "OAuth client_id is required for '{}'.\n\
                 Set a registered client id via:\n\
                   export {}=your-client-id\n\
                 or:\n\
                   export ZEPTOCLAW_OAUTH_CLIENT_ID=your-client-id\n\n\
                 Note: Anthropic OAuth endpoints/flow may not be publicly available yet.",
                provider,
                provider_env
            )
        })?;

    let tokens = auth::oauth::run_oauth_flow(&oauth_config, &client_id).await?;

    save_and_print_tokens(&provider, tokens)?;

    Ok(())
}

/// OpenAI-specific login: try Codex CLI import first, then browser OAuth.
async fn cmd_auth_login_openai() -> Result<()> {
    const OPENAI_CLIENT_ID: &str = "app_EMoamEEZ73f0CkXaXp7hrann";
    const OPENAI_REDIRECT_PORT: u16 = 1455;

    println!("WARNING: OAuth subscription tokens for API access may violate");
    println!("OpenAI's Terms of Service. If tokens are rejected, set an API key:");
    println!("  export ZEPTOCLAW_PROVIDERS_OPENAI_API_KEY=your-key-here");
    println!();

    // 1. Try Codex CLI import
    println!("Checking for Codex CLI credentials...");
    if let Some(tokens) = codex_import::read_codex_credentials() {
        println!("Imported from Codex CLI (~/.codex/auth.json or Keychain)");
        save_and_print_tokens("openai", tokens)?;
        return Ok(());
    }

    // 2. Fall back to native browser OAuth
    println!("No Codex CLI credentials found.");
    println!("Starting browser-based OpenAI OAuth flow...");
    println!();

    let oauth_config = auth::provider_oauth_config("openai").ok_or_else(|| {
        anyhow::anyhow!("OpenAI OAuth config not registered in provider_oauth_config()")
    })?;

    let tokens = auth::oauth::run_oauth_flow_with_port(
        &oauth_config,
        OPENAI_CLIENT_ID,
        OPENAI_REDIRECT_PORT,
    )
    .await?;

    save_and_print_tokens("openai", tokens)?;
    Ok(())
}

/// Save tokens to the encrypted store and print success information.
fn save_and_print_tokens(provider: &str, tokens: auth::OAuthTokenSet) -> Result<()> {
    let encryption = zeptoclaw::security::encryption::resolve_master_key(true)
        .map_err(|e| anyhow::anyhow!("Cannot store tokens without encryption key: {}", e))?;

    let store = auth::store::TokenStore::new(encryption);
    store
        .save(&tokens)
        .map_err(|e| anyhow::anyhow!("Failed to save tokens: {}", e))?;

    println!();
    println!("Authenticated with {} successfully!", provider);
    if tokens.expires_at.is_some() {
        println!("Token expires in: {}", tokens.expires_in_human());
    }
    if tokens.refresh_token.is_some() {
        println!("Refresh token: stored (will auto-refresh before expiry)");
    }

    // Suggest config update
    println!();
    println!("To use OAuth tokens automatically, update your config:");
    println!(
        r#"  "providers": {{ "{}": {{ "auth_method": "auto" }} }}"#,
        provider
    );

    Ok(())
}

/// OAuth logout.
fn cmd_auth_logout(provider: Option<String>) -> Result<()> {
    let encryption = match zeptoclaw::security::encryption::resolve_master_key(true) {
        Ok(enc) => enc,
        Err(_) => {
            println!("No encryption key available. If you have stored tokens,");
            let path = Config::dir().join("auth").join("tokens.json.enc");
            #[cfg(windows)]
            println!("delete them manually: del {}", path.display());
            #[cfg(not(windows))]
            println!("delete them manually: rm {}", path.display());
            return Ok(());
        }
    };

    let store = auth::store::TokenStore::new(encryption);

    if let Some(provider) = provider {
        match store.delete(&provider) {
            Ok(true) => println!("Logged out from {} (OAuth tokens removed).", provider),
            Ok(false) => println!("No OAuth tokens stored for '{}'.", provider),
            Err(e) => println!("Failed to remove tokens: {}", e),
        }
    } else {
        // Show what's stored and ask which to remove
        match store.list() {
            Ok(entries) if entries.is_empty() => {
                println!("No OAuth tokens stored.");
            }
            Ok(entries) => {
                println!("Stored OAuth tokens:");
                for (name, summary) in &entries {
                    println!(
                        "  {}: {} (refresh: {})",
                        name,
                        if summary.is_expired {
                            "expired"
                        } else {
                            &summary.expires_in
                        },
                        if summary.has_refresh_token {
                            "yes"
                        } else {
                            "no"
                        },
                    );
                }
                println!();
                println!("Specify a provider to log out: zeptoclaw auth logout <provider>");
            }
            Err(e) => println!("Failed to read token store: {}", e),
        }
    }

    Ok(())
}

/// Force refresh OAuth tokens.
async fn cmd_auth_refresh(provider: &str) -> Result<()> {
    let encryption = zeptoclaw::security::encryption::resolve_master_key(true)
        .map_err(|e| anyhow::anyhow!("Cannot access tokens without encryption key: {}", e))?;

    let store = auth::store::TokenStore::new(encryption);

    match auth::refresh::ensure_fresh_token(&store, provider).await {
        Ok(_) => {
            println!("Token checked for {}.", provider);
            println!("If a refresh was needed and succeeded, the new token is stored securely.");
        }
        Err(e) => {
            println!("Failed to refresh token for {}: {}", provider, e);
            println!();
            println!("Try logging in again: zeptoclaw auth login {}", provider);
        }
    }

    Ok(())
}

/// Set up a Claude Code subscription token for API access.
///
/// Prompts the user to paste access and refresh tokens from `claude auth token`,
/// validates the prefix, stores them encrypted, and sets auth_method to auto.
pub(crate) async fn cmd_auth_setup_token() -> Result<()> {
    println!("Claude Code Subscription Token Setup");
    println!("====================================");
    println!();
    println!("This imports tokens from your Claude Pro/Max subscription.");
    println!("In Claude Code CLI, run: claude auth token");
    println!("Then paste the tokens below.");
    println!();
    println!("WARNING: Using subscription tokens for API access may violate");
    println!("Anthropic's Terms of Service. Tokens may be revoked at any time.");
    println!("If revoked, ZeptoClaw will fall back to your API key.");
    println!();

    // Read access token
    print!("Access token: ");
    io::stdout().flush()?;
    let access_token = super::common::read_secret()?;
    if access_token.is_empty() {
        println!("No access token provided. Aborting.");
        return Ok(());
    }

    // Read refresh token
    print!("Refresh token (optional, press Enter to skip): ");
    io::stdout().flush()?;
    let refresh_token = super::common::read_secret()?;
    let refresh_token = if refresh_token.is_empty() {
        None
    } else {
        Some(refresh_token)
    };

    // Build the token set
    let now = chrono::Utc::now().timestamp();
    let tokens = auth::OAuthTokenSet {
        provider: "anthropic".to_string(),
        access_token,
        refresh_token,
        expires_at: None, // subscription tokens have no fixed expiry
        token_type: "Bearer".to_string(),
        scope: None,
        obtained_at: now,
        client_id: Some(auth::CLAUDE_CODE_CLIENT_ID.to_string()),
    };

    // Store encrypted
    let encryption = zeptoclaw::security::encryption::resolve_master_key(true)
        .map_err(|e| anyhow::anyhow!("Cannot store tokens without encryption key: {}", e))?;
    let store = auth::store::TokenStore::new(encryption);
    store
        .save(&tokens)
        .map_err(|e| anyhow::anyhow!("Failed to save tokens: {}", e))?;

    // Set auth_method to auto in config
    let mut config = Config::load().unwrap_or_default();
    let provider_config = config
        .providers
        .anthropic
        .get_or_insert_with(Default::default);
    provider_config.auth_method = Some("auto".to_string());
    config.save().with_context(|| "Failed to save config")?;

    println!();
    println!("Subscription token stored and encrypted.");
    println!("Auth method set to \"auto\" (OAuth first, API key fallback).");
    if tokens.refresh_token.is_some() {
        println!("Refresh token: stored (will auto-refresh before expiry).");
    }
    println!();
    println!("Test with: zeptoclaw agent -m \"Hello\"");

    Ok(())
}

/// Show authentication status.
async fn cmd_auth_status() -> Result<()> {
    let config = Config::load().with_context(|| "Failed to load configuration")?;

    println!("Authentication Status");
    println!("=====================");
    println!();

    // Load OAuth token store (best-effort)
    let token_store = match zeptoclaw::security::encryption::resolve_master_key(false) {
        Ok(enc) => Ok(auth::store::TokenStore::new(enc)),
        Err(err) => {
            println!("OAuth token store unavailable: {}", err);
            println!();
            Err(err)
        }
    };

    let oauth_status = |name: &str| -> String {
        if let Ok(ref store) = token_store {
            match store.load(name) {
                Ok(Some(token)) => {
                    if token.is_expired() {
                        return format!(
                            "OAuth (expired{})",
                            if token.refresh_token.is_some() {
                                ", has refresh token"
                            } else {
                                ""
                            }
                        );
                    }
                    return format!("OAuth (expires in {})", token.expires_in_human());
                }
                Ok(None) => {}
                Err(err) => {
                    return format!("OAuth (error: {})", err);
                }
            }
        }
        String::new()
    };

    let provider_display = |name: &str, label: &str, provider: &Option<ProviderConfig>| {
        let api = provider_status(provider);
        let oauth = oauth_status(name);
        let auth_method = provider
            .as_ref()
            .and_then(|p| p.auth_method.as_deref())
            .unwrap_or("api_key");

        if !oauth.is_empty() {
            println!("  {}: {} | {} [method: {}]", label, api, oauth, auth_method);
        } else {
            println!("  {}: {} [method: {}]", label, api, auth_method);
        }
    };

    provider_display(
        "anthropic",
        "Anthropic (Claude)",
        &config.providers.anthropic,
    );
    provider_display("openai", "OpenAI            ", &config.providers.openai);
    provider_display(
        "openrouter",
        "OpenRouter        ",
        &config.providers.openrouter,
    );
    provider_display("groq", "Groq              ", &config.providers.groq);
    provider_display("gemini", "Gemini            ", &config.providers.gemini);
    provider_display("zhipu", "Zhipu             ", &config.providers.zhipu);

    println!();
    println!(
        "OAuth-supported providers: {}",
        auth::oauth_supported_providers().join(", ")
    );

    println!();
    println!("Runtime Provider Support");
    println!("------------------------");
    println!("  Supported: {}", RUNTIME_SUPPORTED_PROVIDERS.join(", "));
    let unsupported = configured_unsupported_provider_names(&config);
    if !unsupported.is_empty() {
        println!("  Configured but unsupported: {}", unsupported.join(", "));
    }

    println!();

    // Channel tokens
    println!("Channel Tokens");
    println!("--------------");

    // Telegram
    let telegram_status = config
        .channels
        .telegram
        .as_ref()
        .map(|c| {
            if c.token.is_empty() {
                "not set"
            } else if c.enabled {
                "configured (enabled)"
            } else {
                "configured (disabled)"
            }
        })
        .unwrap_or("not set");
    println!("  Telegram: {}", telegram_status);

    // Discord
    let discord_status = config
        .channels
        .discord
        .as_ref()
        .map(|c| {
            if c.token.is_empty() {
                "not set"
            } else if c.enabled {
                "configured (enabled)"
            } else {
                "configured (disabled)"
            }
        })
        .unwrap_or("not set");
    println!("  Discord:  {}", discord_status);

    // Slack
    let slack_status = config
        .channels
        .slack
        .as_ref()
        .map(|c| {
            if c.bot_token.is_empty() {
                "not set"
            } else if c.enabled {
                "configured (enabled)"
            } else {
                "configured (disabled)"
            }
        })
        .unwrap_or("not set");
    println!("  Slack:    {}", slack_status);

    println!();

    Ok(())
}

fn provider_status(provider: &Option<ProviderConfig>) -> &'static str {
    provider
        .as_ref()
        .and_then(|p| p.api_key.as_ref())
        .map(|k| {
            if k.is_empty() {
                "not set"
            } else {
                "configured"
            }
        })
        .unwrap_or("not set")
}

/// Show system status.
pub(crate) async fn cmd_status() -> Result<()> {
    let config = Config::load().unwrap_or_default();

    println!("ZeptoClaw Status");
    println!("================");
    println!();

    // Version
    println!("Version: {}", env!("CARGO_PKG_VERSION"));
    println!();

    // Configuration
    println!("Configuration");
    println!("-------------");
    println!("  Config directory: {:?}", Config::dir());
    println!("  Config file:      {:?}", Config::path());
    println!("  Config exists:    {}", Config::path().exists());
    println!();

    // Workspace
    println!("Workspace");
    println!("---------");
    let workspace_path = config.workspace_path();
    println!("  Path:   {:?}", workspace_path);
    println!("  Exists: {}", workspace_path.exists());
    println!();

    // Sessions
    println!("Sessions");
    println!("--------");
    let sessions_path = Config::dir().join("sessions");
    println!("  Path:   {:?}", sessions_path);
    println!("  Exists: {}", sessions_path.exists());
    if sessions_path.exists() {
        let session_count = std::fs::read_dir(&sessions_path)
            .map(|entries| entries.filter_map(|e| e.ok()).count())
            .unwrap_or(0);
        println!("  Count:  {}", session_count);
    }
    println!();

    // Agent defaults
    println!("Agent Defaults");
    println!("--------------");
    println!("  Model:              {}", config.agents.defaults.model);
    println!(
        "  Max tokens:         {}",
        config.agents.defaults.max_tokens
    );
    println!(
        "  Temperature:        {}",
        config.agents.defaults.temperature
    );
    println!(
        "  Max tool iterations: {}",
        config.agents.defaults.max_tool_iterations
    );
    println!();

    // Gateway
    println!("Gateway");
    println!("-------");
    println!("  Host: {}", config.gateway.host);
    println!("  Port: {}", config.gateway.port);
    println!();

    // Daemon
    println!("Daemon");
    println!("------");
    match super::daemon::read_state() {
        Some(state) => {
            println!("  Status:     {}", state.status);
            println!("  Started at: {}", state.started_at);
            println!("  Gateway:    {}", state.gateway);
            for comp in &state.components {
                let running = if comp.running { "running" } else { "stopped" };
                println!(
                    "  Component:  {} ({}, restarts: {})",
                    comp.name, running, comp.restart_count
                );
                if let Some(ref err) = comp.last_error {
                    println!("    Last error: {}", err);
                }
            }
        }
        None => {
            println!("  Not running (no daemon state file)");
        }
    }
    println!();

    // Container Agent
    println!("Container Agent");
    println!("---------------");
    let backend_label = match config.container_agent.backend {
        ContainerAgentBackend::Auto => "auto",
        ContainerAgentBackend::Docker => "docker",
        #[cfg(target_os = "macos")]
        ContainerAgentBackend::Apple => "apple",
    };
    println!("  Backend: {}", backend_label);
    println!("  Image: {}", config.container_agent.image);
    if let Some(binary) = config.container_agent.docker_binary.as_deref() {
        if !binary.trim().is_empty() {
            println!("  Docker binary override: {}", binary);
        }
    }
    println!("  Timeout: {}s", config.container_agent.timeout_secs);
    println!(
        "  Max concurrent: {}",
        config.container_agent.max_concurrent
    );
    println!();

    // Runtime info
    println!("Runtime");
    println!("-------");
    println!("  Type: {:?}", config.runtime.runtime_type);
    println!(
        "  Native fallback: {}",
        if config.runtime.allow_fallback_to_native {
            "enabled (opt-in)"
        } else {
            "disabled (fail-closed)"
        }
    );
    let available = available_runtimes().await;
    println!("  Available: {}", available.join(", "));
    println!();

    // Memory
    println!("Memory");
    println!("------");
    println!(
        "  Backend: {}",
        memory_backend_label(&config.memory.backend)
    );
    println!(
        "  Citations: {}",
        memory_citations_label(&config.memory.citations)
    );
    println!(
        "  Include default files: {}",
        if config.memory.include_default_memory {
            "yes"
        } else {
            "no"
        }
    );
    println!("  Max results: {}", config.memory.max_results);
    println!("  Min score: {}", config.memory.min_score);

    // Long-term memory stats
    let ltm_path = Config::dir().join("memory").join("longterm.json");
    if ltm_path.exists() {
        match zeptoclaw::memory::longterm::LongTermMemory::new() {
            Ok(mem) => {
                let count = mem.count();
                let categories = mem.categories();
                println!("  Long-term entries: {}", count);
                if !categories.is_empty() {
                    println!("  Categories: {}", categories.join(", "));
                }
            }
            Err(_) => {
                println!("  Long-term memory: error reading");
            }
        }
    } else {
        println!("  Long-term entries: 0 (no data file)");
    }
    println!();

    // Heartbeat
    println!("Heartbeat");
    println!("---------");
    println!(
        "  Enabled: {}",
        if config.heartbeat.enabled {
            "yes"
        } else {
            "no"
        }
    );
    println!("  Interval: {}s", config.heartbeat.interval_secs);
    println!("  File: {:?}", heartbeat_file_path(&config));
    println!();

    // Skills
    println!("Skills");
    println!("------");
    println!(
        "  Enabled: {}",
        if config.skills.enabled { "yes" } else { "no" }
    );
    println!(
        "  Workspace dir: {:?}",
        skills_loader_from_config(&config).workspace_dir()
    );
    if !config.skills.always_load.is_empty() {
        println!("  Always load: {}", config.skills.always_load.join(", "));
    }
    if !config.skills.disabled.is_empty() {
        println!("  Disabled: {}", config.skills.disabled.join(", "));
    }
    println!();

    // Provider status
    let runtime_provider_name = resolve_runtime_provider(&config).map(|provider| provider.name);
    println!(
        "Runtime provider: {}",
        runtime_provider_name.unwrap_or("not configured")
    );
    let unsupported = configured_unsupported_provider_names(&config);
    if !unsupported.is_empty() {
        println!("Configured but unsupported: {}", unsupported.join(", "));
    }
    println!(
        "Runtime supports: {}",
        RUNTIME_SUPPORTED_PROVIDERS.join(", ")
    );
    println!();

    // Available tools (dynamic)
    println!("Available Tools");
    println!("---------------");
    super::tools::print_tools_summary(&config);
    println!();

    Ok(())
}