bamboo-agent 2026.4.2

//! CLI module for agent interaction
//!
//! This module contains CLI utilities for interacting with the agent server.
//! Currently used for testing and development purposes.

#![allow(dead_code)]

use clap::{Parser, Subcommand};
use colored::Colorize;
use eventsource_client::Client;
use futures::StreamExt;
use serde::{Deserialize, Serialize};
use std::io::{self, Write};
use std::time::Instant;

#[derive(Parser)]
#[command(name = "copilot-agent-cli")]
#[command(about = "CLI tool for copilot-agent")]
#[command(version)]
struct Cli {
    #[arg(long, default_value = "http://localhost:9562")]
    server_url: String,

    #[arg(long)]
    session_id: Option<String>,

    /// Enable debug mode
    #[arg(long, short, default_value = "false")]
    debug: bool,

    #[command(subcommand)]
    command: Commands,
}

#[derive(Subcommand)]
enum Commands {
    /// Start interactive chat
    Chat,
    /// Send a single message
    Send {
        /// Message content
        message: String,
    },
    /// Test SSE streaming output
    Stream {
        /// Message content
        message: String,
    },
    /// View session history
    History,
}

#[derive(Serialize)]
struct ChatRequest {
    message: String,
    session_id: Option<String>,
    model: Option<String>,
}

#[derive(Deserialize, Debug)]
struct ChatResponse {
    session_id: String,
    stream_url: String,
    #[allow(dead_code)]
    status: String,
}

#[derive(Deserialize, Debug)]
#[serde(tag = "type", rename_all = "snake_case")]
enum AgentEvent {
    Token {
        content: String,
    },
    ReasoningToken {
        content: String,
    },
    ToolToken {
        #[allow(dead_code)]
        tool_call_id: String,
        content: String,
    },
    ToolStart {
        #[allow(dead_code)]
        tool_call_id: String,
        tool_name: String,
        arguments: serde_json::Value,
    },
    ToolComplete {
        #[allow(dead_code)]
        tool_call_id: String,
        result: ToolResult,
    },
    ToolError {
        #[allow(dead_code)]
        tool_call_id: String,
        error: String,
    },
    NeedClarification {
        question: String,
        options: Option<Vec<String>>,
    },
    ToolLifecycle {
        #[allow(dead_code)]
        tool_call_id: String,
        tool_name: String,
        phase: String,
        #[allow(dead_code)]
        elapsed_ms: Option<u64>,
        is_mutating: bool,
        #[allow(dead_code)]
        auto_approved: bool,
        #[allow(dead_code)]
        summary: Option<String>,
        #[allow(dead_code)]
        error: Option<String>,
    },
    ContextCompressionStatus {
        phase: String,
        status: String,
    },
    Complete {
        usage: TokenUsage,
    },
    Error {
        message: String,
    },
}

#[derive(Deserialize, Debug)]
struct ToolResult {
    #[allow(dead_code)]
    success: bool,
    result: String,
}

#[derive(Deserialize, Debug)]
struct TokenUsage {
    prompt_tokens: u32,
    completion_tokens: u32,
    total_tokens: u32,
}

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let cli = Cli::parse();

    if cli.debug {
        eprintln!("{}", "[DEBUG] Debug mode enabled".dimmed());
        eprintln!(
            "{}",
            format!("[DEBUG] Server URL: {}", cli.server_url).dimmed()
        );
    }

    match cli.command {
        Commands::Chat => run_interactive_chat(&cli.server_url, cli.session_id, cli.debug).await,
        Commands::Send { message } => {
            send_message(&cli.server_url, cli.session_id, &message, cli.debug).await
        }
        Commands::Stream { message } => {
            stream_message(&cli.server_url, cli.session_id, &message, cli.debug).await
        }
        Commands::History => get_history(&cli.server_url, cli.session_id, cli.debug).await,
    }
}

async fn send_message(
    server_url: &str,
    session_id: Option<String>,
    message: &str,
    debug: bool,
) -> anyhow::Result<()> {
    let client = reqwest::Client::new();
    let request = ChatRequest {
        message: message.to_string(),
        session_id: session_id.clone(),
        model: None,
    };

    let url = format!("{}/api/v1/chat", server_url);

    if debug {
        eprintln!("{}", format!("[DEBUG] POST {}", url).dimmed());
        eprintln!(
            "{}",
            format!("[DEBUG] Request body: {}", serde_json::to_string(&request)?).dimmed()
        );
    }

    println!("{}", format!("🚀 Sending message: {}", message).cyan());

    let start = Instant::now();
    let response = client.post(&url).json(&request).send().await?;
    let elapsed = start.elapsed();

    if debug {
        eprintln!(
            "{}",
            format!("[DEBUG] Response: {} in {:?}", response.status(), elapsed).dimmed()
        );
        eprintln!(
            "{}",
            format!("[DEBUG] Response headers: {:?}", response.headers()).dimmed()
        );
    }

    if response.status().is_success() {
        let chat_response: ChatResponse = response.json().await?;
        println!(
            "{}",
            format!("✅ Session ID: {}", chat_response.session_id).green()
        );
        println!(
            "{}",
            format!("📡 Stream URL: {}", chat_response.stream_url).green()
        );

        if debug {
            eprintln!(
                "{}",
                format!("[DEBUG] Full response: {:?}", chat_response).dimmed()
            );
        }

        // Try to read stream
        let stream_url = format!("{}{}", server_url, chat_response.stream_url);
        if debug {
            eprintln!(
                "{}",
                format!("[DEBUG] Connecting to stream: {}", stream_url).dimmed()
            );
        }

        let stream_response = client.get(&stream_url).send().await?;

        if debug {
            eprintln!(
                "{}",
                format!("[DEBUG] Stream response: {}", stream_response.status()).dimmed()
            );
        }

        if stream_response.status().is_success() {
            let body = stream_response.text().await?;
            println!("{}", format!("📦 Response: {}", body).yellow());
        }
    } else {
        println!("{}", format!("❌ Error: {}", response.status()).red());
        let text = response.text().await?;
        if debug {
            eprintln!("{}", format!("[DEBUG] Error body: {}", text).dimmed());
        }
        println!("{}", text.red());
    }

    Ok(())
}

async fn stream_message(
    server_url: &str,
    session_id: Option<String>,
    message: &str,
    debug: bool,
) -> anyhow::Result<()> {
    let client = reqwest::Client::new();
    let session_id = session_id.unwrap_or_else(|| uuid::Uuid::new_v4().to_string());

    let request = ChatRequest {
        message: message.to_string(),
        session_id: Some(session_id.clone()),
        model: None,
    };

    let url = format!("{}/api/v1/chat", server_url);

    if debug {
        eprintln!("{}", format!("[DEBUG] POST {}", url).dimmed());
        eprintln!("{}", format!("[DEBUG] Session ID: {}", session_id).dimmed());
        eprintln!("{}", format!("[DEBUG] Message: {}", message).dimmed());
    }

    println!(
        "{}",
        format!("🚀 Starting stream session: {}", session_id).cyan()
    );

    let start = Instant::now();
    let response = client.post(&url).json(&request).send().await?;

    if debug {
        eprintln!(
            "{}",
            format!(
                "[DEBUG] Chat response: {} in {:?}",
                response.status(),
                start.elapsed()
            )
            .dimmed()
        );
    }

    if !response.status().is_success() {
        println!("{}", format!("❌ Error: {}", response.status()).red());
        return Ok(());
    }

    let chat_response: ChatResponse = response.json().await?;

    if debug {
        eprintln!(
            "{}",
            format!("[DEBUG] Stream URL: {}", chat_response.stream_url).dimmed()
        );
    }

    println!("{}", "📝 Stream output:".cyan());
    println!("{}", "─".repeat(50).dimmed());

    // Use SSE client to read stream
    let stream_url = format!("{}{}", server_url, chat_response.stream_url);

    if debug {
        eprintln!(
            "{}",
            format!("[DEBUG] Connecting SSE: {}", stream_url).dimmed()
        );
    }

    let sse_client = eventsource_client::ClientBuilder::for_url(&stream_url)?.build();

    let mut stream = sse_client.stream();
    let mut content_buffer = String::new();
    let mut event_count = 0;
    let stream_start = Instant::now();

    while let Some(event) = stream.next().await {
        match event {
            Ok(eventsource_client::SSE::Event(event)) => {
                event_count += 1;

                if debug {
                    eprintln!(
                        "{}",
                        format!("[DEBUG] Raw event {}: {}", event_count, event.data).dimmed()
                    );
                }

                if let Ok(agent_event) = serde_json::from_str::<AgentEvent>(&event.data) {
                    match &agent_event {
                        AgentEvent::Token { content } => {
                            print!("{}", content.green());
                            io::stdout().flush()?;
                            content_buffer.push_str(content);
                        }
                        AgentEvent::ReasoningToken { content } => {
                            print!("{}", content.blue().dimmed());
                            io::stdout().flush()?;
                        }
                        AgentEvent::ToolToken { content, .. } => {
                            // Don't mix tool output into the assistant content buffer.
                            print!("{}", content.dimmed());
                            io::stdout().flush()?;
                        }
                        AgentEvent::ToolStart {
                            tool_name,
                            arguments,
                            ..
                        } => {
                            println!();
                            println!("{}", format!("🔧 Executing tool: {}", tool_name).yellow());
                            println!("{}", format!("   Args: {}", arguments).dimmed());
                        }
                        AgentEvent::ToolComplete { result, .. } => {
                            println!("{}", format!("✅ Tool result: {}", result.result).green());
                        }
                        AgentEvent::ToolError { error, .. } => {
                            println!("{}", format!("❌ Tool error: {}", error).red());
                        }
                        AgentEvent::NeedClarification { question, options } => {
                            println!();
                            println!(
                                "{}",
                                format!("❓ Clarification needed: {}", question).yellow()
                            );
                            if let Some(options) = options {
                                println!(
                                    "{}",
                                    format!("   Options: {}", options.join(" | ")).dimmed()
                                );
                            }
                        }
                        AgentEvent::ToolLifecycle {
                            tool_name,
                            phase,
                            is_mutating,
                            ..
                        } => {
                            let mutating_label = if *is_mutating { "⚡" } else { "📖" };
                            let msg = format!(
                                "{} Tool lifecycle: {} {} [{}]",
                                mutating_label,
                                tool_name,
                                phase,
                                if *is_mutating {
                                    "mutating"
                                } else {
                                    "read-only"
                                }
                            );
                            if phase == "error" || phase == "cancelled" {
                                println!("{}", msg.yellow());
                            } else {
                                println!("{}", msg.dimmed());
                            }
                        }
                        AgentEvent::ContextCompressionStatus { phase, status } => {
                            println!(
                                "{}",
                                format!("🗜️ Context compression {status} ({phase})").dimmed()
                            );
                        }
                        AgentEvent::Complete { usage } => {
                            println!();
                            println!(
                                "{}",
                                format!(
                                    "📊 Tokens: prompt={}, completion={}, total={}",
                                    usage.prompt_tokens,
                                    usage.completion_tokens,
                                    usage.total_tokens
                                )
                                .dimmed()
                            );
                        }
                        AgentEvent::Error { message } => {
                            println!();
                            println!("{}", format!("❌ Error: {}", message).red());
                        }
                    }
                } else if debug {
                    eprintln!(
                        "{}",
                        format!("[DEBUG] Failed to parse event: {}", event.data).dimmed()
                    );
                }
            }
            Ok(eventsource_client::SSE::Comment(comment)) => {
                if debug {
                    eprintln!("{}", format!("[DEBUG] SSE Comment: {}", comment).dimmed());
                }
            }
            Err(e) => {
                if debug {
                    eprintln!("{}", format!("[DEBUG] SSE Error: {:?}", e).dimmed());
                }
                eprintln!("\n{}: {:?}", "SSE Error".red(), e);
                break;
            }
        }
    }

    let stream_duration = stream_start.elapsed();

    if debug {
        eprintln!(
            "{}",
            format!(
                "[DEBUG] Stream completed: {} events in {:?}",
                event_count, stream_duration
            )
            .dimmed()
        );
    }

    println!();
    println!("{}", "─".repeat(50).dimmed());
    println!("{}", "✨ Stream complete".cyan());

    if !content_buffer.is_empty() {
        println!();
        println!("{}", "📝 Complete response:".cyan());
        println!("{}", content_buffer);
    }

    Ok(())
}

async fn run_interactive_chat(
    server_url: &str,
    session_id: Option<String>,
    debug: bool,
) -> anyhow::Result<()> {
    let session_id = session_id.unwrap_or_else(|| uuid::Uuid::new_v4().to_string());

    println!("{}", "🤖 Copilot Agent Interactive Chat".cyan().bold());
    println!("{}", format!("Session ID: {}", session_id).dimmed());
    println!("{}", "Type 'exit' or 'quit' to leave".dimmed());

    if debug {
        eprintln!("{}", format!("[DEBUG] Server URL: {}", server_url).dimmed());
        eprintln!("{}", "[DEBUG] Debug mode enabled".dimmed());
    }

    println!();

    loop {
        print!("{} ", "You:".cyan().bold());
        io::stdout().flush()?;

        let mut input = String::new();
        io::stdin().read_line(&mut input)?;
        let input = input.trim();

        if input.eq_ignore_ascii_case("exit") || input.eq_ignore_ascii_case("quit") {
            println!("{}", "👋 Goodbye!".cyan());
            break;
        }

        if input.is_empty() {
            continue;
        }

        println!("{}", "Assistant:".green().bold());

        if let Err(e) = stream_message(server_url, Some(session_id.clone()), input, debug).await {
            if debug {
                eprintln!("{}", format!("[DEBUG] Error: {:?}", e).dimmed());
            }
            println!("{}", format!("❌ Error: {}", e).red());
        }

        println!();
    }

    Ok(())
}

async fn get_history(
    server_url: &str,
    session_id: Option<String>,
    debug: bool,
) -> anyhow::Result<()> {
    let session_id = match session_id {
        Some(id) => id,
        None => {
            println!("{}", "❌ Please provide --session-id".red());
            return Ok(());
        }
    };

    let url = format!("{}/api/v1/history/{}", server_url, session_id);

    if debug {
        eprintln!("{}", format!("[DEBUG] GET {}", url).dimmed());
    }

    let client = reqwest::Client::new();
    let response = client.get(&url).send().await?;

    if debug {
        eprintln!(
            "{}",
            format!("[DEBUG] Response: {}", response.status()).dimmed()
        );
    }

    if response.status().is_success() {
        let history: serde_json::Value = response.json().await?;

        if debug {
            eprintln!("{}", "[DEBUG] Raw response:".dimmed());
        }

        println!("{}", serde_json::to_string_pretty(&history)?);
    } else {
        println!("{}", format!("❌ Error: {}", response.status()).red());
        let text = response.text().await?;
        if debug {
            eprintln!("{}", format!("[DEBUG] Error body: {}", text).dimmed());
        }
    }

    Ok(())
}