a2a-agents 0.5.0

use a2a_agents::agents::reimbursement::handler::ReimbursementHandler;
use a2a_rs::InMemoryStreamingHandler;
use a2a_rs::adapter::storage::InMemoryTaskStorage;
use a2a_rs::domain::{Message, Part, Role};
use a2a_rs::port::message_handler::AsyncMessageHandler;
use serde_json::json;
use uuid::Uuid;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    // Initialize logging
    tracing_subscriber::fmt().with_env_filter("debug").init();

    // Create handler with in-memory task storage and a dedicated streaming
    // handler (streaming and push are separate ports).
    let task_storage = InMemoryTaskStorage::new();
    let push = task_storage.push_notifier();
    let handler = ReimbursementHandler::new(task_storage, InMemoryStreamingHandler::new(), push);

    println!("=== Testing Reimbursement Handler ===\n");

    // Test 1: Text-based request
    println!("1. Testing text-based request:");
    let text_message = Message::builder()
        .role(Role::User)
        .parts(vec![Part::text(
            "I need to get reimbursed $150.50 for client lunch at downtown restaurant".to_string(),
        )])
        .message_id(Uuid::new_v4().to_string())
        .context_id("conv-123".to_string())
        .build();

    let task_id = format!("task_{}", Uuid::new_v4().simple());
    let result = handler
        .process_message(&task_id, &text_message, None)
        .await?;
    println!("Response: {:?}\n", result.status.message);

    // Test 2: Structured data request
    println!("2. Testing structured data request:");
    let data_message = Message::builder()
        .role(Role::User)
        .parts(vec![Part::data(
            serde_json::from_value(json!({
                "date": "2024-01-15",
                "amount": 250.00,
                "purpose": "Team building dinner for Q1 planning",
                "category": "meals"
            }))
            .unwrap(),
        )])
        .message_id(Uuid::new_v4().to_string())
        .context_id("conv-456".to_string())
        .build();

    let task_id = format!("task_{}", Uuid::new_v4().simple());
    let result = handler
        .process_message(&task_id, &data_message, None)
        .await?;
    println!("Response: {:?}\n", result.status.message);

    // Test 3: Form submission
    println!("3. Testing form submission:");
    let form_message = Message::builder()
        .role(Role::User)
        .parts(vec![Part::data(
            serde_json::from_value(json!({
                "request_id": "req_12345",
                "date": "2024-01-20",
                "amount": {"amount": 500.00, "currency": "USD"},
                "purpose": "Conference registration and travel expenses",
                "category": "travel",
                "notes": "Annual tech conference in SF"
            }))
            .unwrap(),
        )])
        .message_id(Uuid::new_v4().to_string())
        .context_id("conv-789".to_string())
        .build();

    let task_id = format!("task_{}", Uuid::new_v4().simple());
    let result = handler
        .process_message(&task_id, &form_message, None)
        .await?;
    println!("Response: {:?}\n", result.status.message);

    // Test 4: Mixed content (text + file reference)
    println!("4. Testing mixed content with file:");
    let file_part = Part::file_builder()
        .name("receipt_20240115.pdf".to_string())
        .mime_type("application/pdf".to_string())
        .bytes(b"Hello World!".to_vec())
        .with_metadata(serde_json::from_value(json!({"extracted_amount": "$75.50"})).unwrap())
        .build()
        .unwrap();

    let mixed_message = Message::builder()
        .role(Role::User)
        .parts(vec![
            Part::text("Here's my receipt for the office supplies".to_string()),
            file_part,
        ])
        .message_id(Uuid::new_v4().to_string())
        .build();

    let task_id = format!("task_{}", Uuid::new_v4().simple());
    let result = handler
        .process_message(&task_id, &mixed_message, None)
        .await?;
    println!("Response: {:?}\n", result.status.message);

    // Test 5: Status query
    println!("5. Testing status query:");
    let status_message = Message::builder()
        .role(Role::User)
        .parts(vec![Part::text(
            "What's the status of req_12345?".to_string(),
        )])
        .message_id(Uuid::new_v4().to_string())
        .build();

    let task_id = format!("task_{}", Uuid::new_v4().simple());
    let result = handler
        .process_message(&task_id, &status_message, None)
        .await?;
    println!("Response: {:?}\n", result.status.message);

    println!("All tests completed!");
    Ok(())
}