tarzi 0.2.0

use tarzi::{FetchMode, Format, Result, SearchEngine, WebFetcher, config::Config};

#[tokio::main]
async fn main() -> Result<()> {
    // Initialize tracing
    tracing_subscriber::fmt::init();

    println!("=== Tarzi Modular Example ===\n");

    // Load configuration and configure Chrome driver
    let mut config = Config::load().unwrap_or_default();
    config.fetcher.web_driver = "chromedriver".to_string();

    // Example 1: Using the fetcher module directly
    println!("1. Fetching content with different modes:");

    let mut fetcher = WebFetcher::from_config(&config);
    let test_url = tarzi::constants::HTTPBIN_HTML_URL;

    // Plain request mode
    match fetcher
        .fetch(test_url, FetchMode::PlainRequest, Format::Html)
        .await
    {
        Ok(content) => println!("   Plain request: {} characters", content.len()),
        Err(e) => println!("   Plain request failed: {e}"),
    }

    // Browser headless mode
    match fetcher
        .fetch(test_url, FetchMode::BrowserHeadless, Format::Markdown)
        .await
    {
        Ok(content) => println!("   Browser headless: {} characters", content.len()),
        Err(e) => println!("   Browser headless failed: {e}"),
    }

    println!();

    // Example 2: Using the search module
    println!("2. Searching and fetching content:");

    let mut search_engine = SearchEngine::from_config(&config);
    let query = "agentic AI";

    // Search for results
    match search_engine.search(query, 3).await {
        Ok(results) => {
            println!("Found {} results:", results.len());
            for (i, result) in results.iter().enumerate() {
                println!("{}. {}", i + 1, result.title);
                println!("   URL: {}", result.url);
                println!("   Snippet: {}", result.snippet);
            }
        }
        Err(e) => {
            eprintln!("Search failed: {e}");
        }
    }

    println!();

    // Example 3: Search and fetch content for each result
    println!("3. Search and fetch content for each result:");

    let results = search_engine
        .search_with_content(query, 3, FetchMode::BrowserHeadless, Format::Markdown)
        .await?;

    println!("Found {} results with content:", results.len());
    for (i, (result, content)) in results.iter().enumerate() {
        println!("{}. {}", i + 1, result.title);
        println!("   URL: {}", result.url);
        println!("   Content length: {} characters", content.len());
    }

    println!();
    println!("=== Example completed ===");

    Ok(())
}